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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
from pathlib import Path
import pytest
from syrupy.exceptions import FailedToLoadModuleMember
from syrupy.utils import (
import_module_member,
walk_snapshot_dir,
)
def makefiles(testdir, filetree, root=""):
for filename, contents in filetree.items():
filepath = Path(root).joinpath(filename)
if isinstance(contents, dict):
testdir.mkdir(filepath)
makefiles(testdir, contents, str(filepath))
else:
name, ext = str(filepath.with_name(filepath.stem)), filepath.suffix
testdir.makefile(ext, **{name: contents})
@pytest.fixture
def testfiles(testdir):
filetree = {
"file1.txt": "file1",
"file2.txt": "file2",
"__snapshot__": {
"wrong_snapfile1.ambr": "",
"wrong_snapfolder": {"wrong_snapfile2.svg": "<svg></svg>"},
},
"__snapshots__": {
"snapfile1.ambr": "",
"snapfolder": {"snapfile2.svg": "<svg></svg>"},
},
}
makefiles(testdir, filetree)
return filetree, testdir
def test_walk_dir_skips_non_snapshot_path(testfiles):
_, testdir = testfiles
snap_folder = Path("__snapshots__")
assert {
str(Path(p).relative_to(Path.cwd()))
for p in walk_snapshot_dir(Path(testdir.tmpdir).joinpath(snap_folder))
} == {
str(snap_folder.joinpath("snapfile1.ambr")),
str(snap_folder.joinpath("snapfolder", "snapfile2.svg")),
}
def test_walk_dir_ignores_ignored_extensions(testdir):
filetree = {
"file1.txt": "file1",
"file2.txt": "file2",
"__snapshots__": {
"snapfile1.ambr": "",
"snapfile1.ambr.dvc": "",
"snapfolder": {"snapfile2.svg": "<svg></svg>"},
},
}
makefiles(testdir, filetree)
discovered_files = {
str(Path(p).relative_to(Path.cwd()))
for p in walk_snapshot_dir(
Path(testdir.tmpdir).joinpath("__snapshots__"), ignore_extensions=["dvc"]
)
}
assert discovered_files == {
str(Path("__snapshots__").joinpath("snapfile1.ambr")),
str(Path("__snapshots__").joinpath("snapfolder", "snapfile2.svg")),
}
assert (
str(Path("__snapshots__").joinpath("snapfile1.ambr.dvc"))
not in discovered_files
)
def dummy_member():
return 123
def test_import_module_member_imports_member():
imported_member = import_module_member(f"{__name__}.dummy_member")
assert imported_member() == 123
@pytest.mark.parametrize(
"path",
[
"dummy_member",
f"{__name__}badpath.dummy_member",
f"{__name__}.dummy_memberbadmember",
],
)
def test_import_module_member_with_bad_path_raises_exception(path):
with pytest.raises(FailedToLoadModuleMember):
import_module_member(path)
|
