67 lines
2.1 KiB
Python
67 lines
2.1 KiB
Python
|
import os
|
||
|
|
from typing import Sequence
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
ROOT_DIR_PATH = os.path.dirname(os.path.realpath(__file__))
|
||
|
|
PRODUCTION_DOTENVS_DIR_PATH = os.path.join(ROOT_DIR_PATH, ".envs", ".production")
|
||
|
|
PRODUCTION_DOTENV_FILE_PATHS = [
|
||
|
|
os.path.join(PRODUCTION_DOTENVS_DIR_PATH, ".django"),
|
||
|
|
os.path.join(PRODUCTION_DOTENVS_DIR_PATH, ".postgres"),
|
||
|
|
]
|
||
|
|
DOTENV_FILE_PATH = os.path.join(ROOT_DIR_PATH, ".env")
|
||
|
|
|
||
|
|
|
||
|
|
def merge(
|
||
|
|
output_file_path: str, merged_file_paths: Sequence[str], append_linesep: bool = True
|
||
|
|
) -> None:
|
||
|
|
with open(output_file_path, "w") as output_file:
|
||
|
|
for merged_file_path in merged_file_paths:
|
||
|
|
with open(merged_file_path, "r") as merged_file:
|
||
|
|
merged_file_content = merged_file.read()
|
||
|
|
output_file.write(merged_file_content)
|
||
|
|
if append_linesep:
|
||
|
|
output_file.write(os.linesep)
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
merge(DOTENV_FILE_PATH, PRODUCTION_DOTENV_FILE_PATHS)
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("merged_file_count", range(3))
|
||
|
|
@pytest.mark.parametrize("append_linesep", [True, False])
|
||
|
|
def test_merge(tmpdir_factory, merged_file_count: int, append_linesep: bool):
|
||
|
|
tmp_dir_path = str(tmpdir_factory.getbasetemp())
|
||
|
|
|
||
|
|
output_file_path = os.path.join(tmp_dir_path, ".env")
|
||
|
|
|
||
|
|
expected_output_file_content = ""
|
||
|
|
merged_file_paths = []
|
||
|
|
for i in range(merged_file_count):
|
||
|
|
merged_file_ord = i + 1
|
||
|
|
|
||
|
|
merged_filename = ".service{}".format(merged_file_ord)
|
||
|
|
merged_file_path = os.path.join(tmp_dir_path, merged_filename)
|
||
|
|
|
||
|
|
merged_file_content = merged_filename * merged_file_ord
|
||
|
|
|
||
|
|
with open(merged_file_path, "w+") as file:
|
||
|
|
file.write(merged_file_content)
|
||
|
|
|
||
|
|
expected_output_file_content += merged_file_content
|
||
|
|
if append_linesep:
|
||
|
|
expected_output_file_content += os.linesep
|
||
|
|
|
||
|
|
merged_file_paths.append(merged_file_path)
|
||
|
|
|
||
|
|
merge(output_file_path, merged_file_paths, append_linesep)
|
||
|
|
|
||
|
|
with open(output_file_path, "r") as output_file:
|
||
|
|
actual_output_file_content = output_file.read()
|
||
|
|
|
||
|
|
assert actual_output_file_content == expected_output_file_content
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|