cookiecutter/cookiecutter-django
1
1
Fork 0
mirror of https://github.com/cookiecutter/cookiecutter-django.git synced 2025-08-16 01:44:51 +03:00
Code Issues Packages Projects Releases Wiki Activity

Refactor test for merge function

Browse Source
This commit is contained in:
Bruno Alla 2023-01-25 23:13:27 +00:00
parent 49a2433ae9
commit 9957289457
No known key found for this signature in database
1 changed files with 13 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
{{cookiecutter.project_slug}}/merge_production_dotenvs_in_dotenv.py
View File

@ -30,37 +30,29 @@ def main():
merge(DOTENV_FILE, PRODUCTION_DOTENV_FILES) merge(DOTENV_FILE, PRODUCTION_DOTENV_FILES)
@pytest.mark.parametrize("merged_file_count", range(3)) @pytest.mark.parametrize("files_count", range(3))
@pytest.mark.parametrize("append_linesep", [True, False]) @pytest.mark.parametrize("append_linesep", [True, False])
def test_merge(tmpdir_factory, merged_file_count: int, append_linesep: bool): def test_merge(tmp_path: Path, files_count: int, append_linesep: bool):
tmp_dir = Path(str(tmpdir_factory.getbasetemp())) output_file = tmp_path / ".env"
output_file = tmp_dir / ".env"
expected_output_file_content = "" expected_output_file_content = ""
merged_files = [] files_to_merge = []
for i in range(merged_file_count): for num in range(1, files_count + 1):
merged_file_ord = i + 1 merge_filename = f".service{num}"
merge_file = tmp_path / merge_filename
merged_filename = f".service{merged_file_ord}" merge_file_content = merge_filename * num
merged_file = tmp_dir / merged_filename merge_file.write_text(merge_file_content)
merged_file_content = merged_filename * merged_file_ord expected_output_file_content += merge_file_content
with open(merged_file, "w+") as file:
file.write(merged_file_content)
expected_output_file_content += merged_file_content
if append_linesep: if append_linesep:
expected_output_file_content += os.linesep expected_output_file_content += os.linesep
merged_files.append(merged_file) files_to_merge.append(merge_file)
merge(output_file, merged_files, append_linesep) merge(output_file, files_to_merge, append_linesep)
with open(output_file) as output_file:
actual_output_file_content = output_file.read()
actual_output_file_content = output_file.read_text()
assert actual_output_file_content == expected_output_file_content assert actual_output_file_content == expected_output_file_content