1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
import pytest
import os
import difflib
from azure.projects._bicep.utils import generate_suffix
def _get_infra_dir() -> str:
test_dir = os.path.dirname(os.path.realpath(__file__))
return os.path.join(test_dir, "test_infra")
def _compare_outputs(output_dir, ref_dir, test_dir):
for _, _, files in os.walk(os.path.join(output_dir, ref_dir)):
for filename in files:
with open(os.path.join(output_dir, ref_dir, filename), "r") as _ref:
ref_file = _ref.readlines()
with open(os.path.join(output_dir, test_dir, filename), "r") as _test:
test_file = _test.readlines()
diff = difflib.unified_diff(
ref_file,
test_file,
fromfile=f"{ref_dir}/{filename}",
tofile=f"{test_dir}/{filename}",
)
changes = "".join(diff)
has_changes = bool(changes)
assert not has_changes, "\n" + changes
os.remove(os.path.join(output_dir, test_dir, filename))
os.rmdir(os.path.join(output_dir, test_dir))
@pytest.fixture
def _compare_exports(request):
output_dir = _get_infra_dir()
ref_dir = request.node.name
test_dir = f"infra_{request.node.name}_{generate_suffix()}"
yield output_dir, ref_dir, test_dir
_compare_outputs(output_dir, ref_dir, test_dir)
@pytest.fixture
def _record_exports(request):
output_dir = _get_infra_dir()
ref_dir = request.node.name
return output_dir, ref_dir, ref_dir
export_dir = _record_exports if os.environ.get("TEST_EXPORT_RECORD", "").lower() == "true" else _compare_exports
|
