Patched test_line_endings.py

Author: patched.codes[bot]

test_line_endings.py

@@ -0,0 +1,147 @@
+import tempfile
+import os
+
+def analyze_line_endings(content):
+    """Analyze line endings in bytes content, handling the last line correctly"""
+    if not content:
+        return []
+        
+    # Split into lines but keep line endings
+    lines = []
+    current_pos = 0
+    content_len = len(content)
+    
+    while current_pos < content_len:
+        next_lf = content.find(b'\n', current_pos)
+        if next_lf == -1:
+            # Last line without line ending
+            lines.append(content[current_pos:])
+            break
+            
+        # Include the line ending in the line
+        lines.append(content[current_pos:next_lf + 1])
+        current_pos = next_lf + 1
+        
+    return lines
+
+def test_line_endings_case(content, expected_line_endings, test_name):
+    print(f"\n=== Testing {test_name} ===")
+    print("Input file contents (bytes):", repr(content))
+    
+    # Create a temporary file with given content
+    with tempfile.NamedTemporaryFile(mode='wb', delete=False) as f:
+        f.write(content)
+        temp_path = f.name
+
+    # Modify file using ModifyCode functionality
+    from patchwork.steps.ModifyCode.ModifyCode import replace_code_in_file
+    replace_code_in_file(temp_path, 1, 2, "new_line2")
+
+    # Check resulting content
+    with open(temp_path, 'rb') as f:
+        modified_content = f.read()
+        print("Modified contents (bytes):", repr(modified_content))
+    
+    # Analyze line endings
+    original_lines = analyze_line_endings(content)
+    modified_lines = analyze_line_endings(modified_content)
+    
+    print("\nLine ending analysis:")
+    print("Original:")
+    for i, line in enumerate(original_lines):
+        ending = "CRLF" if line.endswith(b'\r\n') else "LF" if line.endswith(b'\n') else "None"
+        print(f"  Line {i+1}: {ending} - {repr(line)}")
+    
+    print("Modified:")
+    for i, line in enumerate(modified_lines):
+        ending = "CRLF" if line.endswith(b'\r\n') else "LF" if line.endswith(b'\n') else "None"
+        print(f"  Line {i+1}: {ending} - {repr(line)}")
+    
+    # Verify against expected line endings
+    success = True
+    for i, (line, expected) in enumerate(zip(modified_lines, expected_line_endings)):
+        actual = "CRLF" if line.endswith(b'\r\n') else "LF" if line.endswith(b'\n') else "None"
+        if actual != expected:
+            print(f"ERROR: Line {i+1} has {actual} ending, expected {expected}")
+            success = False
+    
+    os.unlink(temp_path)
+    return success
+
+def test_new_file_case():
+    """Special test case for creating a new file (when file doesn't exist)"""
+    print("\n=== Testing New File Creation ===")
+    
+    temp_path = os.path.join(tempfile.gettempdir(), "newfile_test.txt")
+    if os.path.exists(temp_path):
+        os.unlink(temp_path)
+        
+    print("Creating new file...")
+    from patchwork.steps.ModifyCode.ModifyCode import replace_code_in_file
+    replace_code_in_file(temp_path, None, None, "line1\nline2\nline3\n")
+    
+    with open(temp_path, 'rb') as f:
+        content = f.read()
+        print("File contents (bytes):", repr(content))
+    
+    lines = analyze_line_endings(content)
+    print("\nLine ending analysis:")
+    success = True
+    for i, line in enumerate(lines):
+        ending = "CRLF" if line.endswith(b'\r\n') else "LF" if line.endswith(b'\n') else "None"
+        print(f"  Line {i+1}: {ending} - {repr(line)}")
+        # For new files, we expect LF endings by default
+        if ending != "LF":
+            print(f"ERROR: Line {i+1} has {ending} ending, expected LF")
+            success = False
+            
+    os.unlink(temp_path)
+    return success
+
+def run_all_tests():
+    test_cases = [
+        {
+            "name": "CRLF file",
+            "content": b"line1\r\nline2\r\nline3\r\n",
+            "expected_endings": ["CRLF", "CRLF", "CRLF"]
+        },
+        {
+            "name": "LF file",
+            "content": b"line1\nline2\nline3\n",
+            "expected_endings": ["LF", "LF", "LF"]
+        },
+        {
+            "name": "Mixed line endings",
+            "content": b"line1\r\nline2\nline3\r\n",
+            "expected_endings": ["CRLF", "CRLF", "CRLF"]  # Should normalize to the first line ending type
+        },
+        {
+            "name": "No trailing newline CRLF",
+            "content": b"line1\r\nline2\r\nline3",  # Note: no trailing newline
+            "expected_endings": ["CRLF", "CRLF", "None"]
+        }
+    ]  # Close test_cases list
+    
+    all_passed = True
+    
+    # Run modification tests
+    print("\n=== Running File Modification Tests ===")
+    for test in test_cases:
+        success = test_line_endings_case(
+            test["content"], 
+            test["expected_endings"],
+            test["name"]
+        )
+        all_passed = all_passed and success
+    
+    # Run new file creation test
+    print("\n=== Running New File Creation Test ===")
+    new_file_success = test_new_file_case()
+    all_passed = all_passed and new_file_success
+    
+    print("\n=== Final Test Summary ===")
+    print("All tests passed!" if all_passed else "Some tests failed!")
+    return all_passed
+
+if __name__ == "__main__":
+    run_all_tests()