evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Ensure that all files in a folder are in the same test bucket (#129946)

Browse source
This commit is contained in:
epenet 2024-11-06 15:38:24 +01:00 • committed by GitHub
parent 51d6948848
commit dea31e5744
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 15 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
script/split_tests.py
View file

@ -49,16 +49,27 @@ class BucketHolder:
test_folder.get_all_flatten(), reverse=True, key=lambda x: x.total_tests test_folder.get_all_flatten(), reverse=True, key=lambda x: x.total_tests
) )
for tests in sorted_tests: for tests in sorted_tests:
print(f"{tests.total_tests:>{digits}} tests in {tests.path}")
if tests.added_to_bucket: if tests.added_to_bucket:
# Already added to bucket # Already added to bucket
continue continue
print(f"{tests.total_tests:>{digits}} tests in {tests.path}")
smallest_bucket = min(self._buckets, key=lambda x: x.total_tests) smallest_bucket = min(self._buckets, key=lambda x: x.total_tests)
is_file = isinstance(tests, TestFile)
if ( if (
smallest_bucket.total_tests + tests.total_tests < self._tests_per_bucket smallest_bucket.total_tests + tests.total_tests < self._tests_per_bucket
) or isinstance(tests, TestFile): ) or is_file:
smallest_bucket.add(tests) smallest_bucket.add(tests)
# Ensure all files from the same folder are in the same bucket
# to ensure that syrupy correctly identifies unused snapshots
if is_file:
for other_test in tests.parent.children.values():
if other_test is tests or isinstance(other_test, TestFolder):
continue
print(
f"{other_test.total_tests:>{digits}} tests in {other_test.path} (same bucket)"
)
smallest_bucket.add(other_test)
# verify that all tests are added to a bucket # verify that all tests are added to a bucket
if not test_folder.added_to_bucket: if not test_folder.added_to_bucket:
@ -79,6 +90,7 @@ class TestFile:
total_tests: int total_tests: int
path: Path path: Path
added_to_bucket: bool = field(default=False, init=False) added_to_bucket: bool = field(default=False, init=False)
parent: TestFolder | None = field(default=None, init=False)
def add_to_bucket(self) -> None: def add_to_bucket(self) -> None:
"""Add test file to bucket.""" """Add test file to bucket."""
@ -125,6 +137,7 @@ class TestFolder:
def add_test_file(self, file: TestFile) -> None: def add_test_file(self, file: TestFile) -> None:
"""Add test file to folder.""" """Add test file to folder."""
path = file.path path = file.path
file.parent = self
relative_path = path.relative_to(self.path) relative_path = path.relative_to(self.path)
if not relative_path.parts: if not relative_path.parts:
raise ValueError("Path is not a child of this folder") raise ValueError("Path is not a child of this folder")