Rewrite test to be dumber

{{cookiecutter.project_slug}}/merge_production_dotenvs_in_dotenv.py

@@ -30,30 +30,54 @@ def main():
     merge(DOTENV_FILE, PRODUCTION_DOTENV_FILES)
 
 
-@pytest.mark.parametrize("files_count", range(3))
-@pytest.mark.parametrize("append_linesep", [True, False])
-def test_merge(tmp_path: Path, files_count: int, append_linesep: bool):
+@pytest.mark.parametrize(
+    ("input_contents", "append_linesep", "expected_output"),
+    [
+        # No line separator
+        ([], False, ""),
+        ([""], False, ""),
+        (["\n"], False, "\n"),
+        (["TEST=1"], False, "TEST=1"),
+        (["THIS=2", "THAT='example'"], False, "THIS=2THAT='example'"),
+        (["ONE=1\n", "TWO=2"], False, "ONE=1\nTWO=2"),
+        (["A=0", "B=1", "C=2"], False, "A=0B=1C=2"),
+        # With line separator
+        (
+            [],
+            True,
+            "",
+        ),
+        (
+            [""],
+            True,
+            "\n",
+        ),
+        (
+            ["JANE=doe"],
+            True,
+            "JANE=doe\n",
+        ),
+        (["SEP=true", "AR=ator"], True, "SEP=true\nAR=ator\n"),
+        (["X=x\n", "Y=y", "Z=z\n"], True, "X=x\n\nY=y\nZ=z\n\n"),
+    ],
+)
+def test_merge(
+    tmp_path: Path,
+    input_contents: list[str],
+    append_linesep: bool,
+    expected_output: str,
+):
     output_file = tmp_path / ".env"
 
-    expected_output_file_content = ""
     files_to_merge = []
-    for num in range(1, files_count + 1):
-        merge_filename = f".service{num}"
-        merge_file = tmp_path / merge_filename
-
-        merge_file_content = merge_filename * num
-        merge_file.write_text(merge_file_content)
-
-        expected_output_file_content += merge_file_content
-        if append_linesep:
-            expected_output_file_content += os.linesep
-
+    for num, input_content in enumerate(input_contents, start=1):
+        merge_file = tmp_path / f".service{num}"
+        merge_file.write_text(input_content)
         files_to_merge.append(merge_file)
 
     merge(output_file, files_to_merge, append_linesep)
 
-    actual_output_file_content = output_file.read_text()
-    assert actual_output_file_content == expected_output_file_content
+    assert output_file.read_text() == expected_output
 
 
 if __name__ == "__main__":