Skip to content

Commit

Permalink
Revert "Fix unused snapshots not triggering failure in CI" (#129223)
Browse files Browse the repository at this point in the history
Revert "Fix unused snapshots not triggering failure in CI (#128162)"

This reverts commit e888a95.
  • Loading branch information
frenck authored Oct 26, 2024
1 parent 46dd96a commit 35b7c30
Show file tree
Hide file tree
Showing 3 changed files with 1 addition and 173 deletions.
4 changes: 0 additions & 4 deletions .github/workflows/ci.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -949,7 +949,6 @@ jobs:
--timeout=9 \
--durations=10 \
--numprocesses auto \
--snapshot-details \
--dist=loadfile \
${cov_params[@]} \
-o console_output_style=count \
Expand Down Expand Up @@ -1072,7 +1071,6 @@ jobs:
-qq \
--timeout=20 \
--numprocesses 1 \
--snapshot-details \
${cov_params[@]} \
-o console_output_style=count \
--durations=10 \
Expand Down Expand Up @@ -1199,7 +1197,6 @@ jobs:
-qq \
--timeout=9 \
--numprocesses 1 \
--snapshot-details \
${cov_params[@]} \
-o console_output_style=count \
--durations=0 \
Expand Down Expand Up @@ -1346,7 +1343,6 @@ jobs:
-qq \
--timeout=9 \
--numprocesses auto \
--snapshot-details \
${cov_params[@]} \
-o console_output_style=count \
--durations=0 \
Expand Down
8 changes: 1 addition & 7 deletions tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,6 @@
import requests_mock
import respx
from syrupy.assertion import SnapshotAssertion
from syrupy.session import SnapshotSession

from homeassistant import block_async_io
from homeassistant.exceptions import ServiceNotFound
Expand Down Expand Up @@ -93,7 +92,7 @@
from homeassistant.util.json import json_loads

from .ignore_uncaught_exceptions import IGNORE_UNCAUGHT_EXCEPTIONS
from .syrupy import HomeAssistantSnapshotExtension, override_syrupy_finish
from .syrupy import HomeAssistantSnapshotExtension
from .typing import (
ClientSessionGenerator,
MockHAClientWebSocket,
Expand Down Expand Up @@ -150,11 +149,6 @@ def pytest_configure(config: pytest.Config) -> None:
if config.getoption("verbose") > 0:
logging.getLogger().setLevel(logging.DEBUG)

# Override default finish to detect unused snapshots despite xdist
# Temporary workaround until it is finalised inside syrupy
# See https://github.com/syrupy-project/syrupy/pull/901
SnapshotSession.finish = override_syrupy_finish


def pytest_runtest_setup() -> None:
"""Prepare pytest_socket and freezegun.
Expand Down
162 changes: 0 additions & 162 deletions tests/syrupy.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,22 +5,14 @@
from contextlib import suppress
import dataclasses
from enum import IntFlag
import json
import os
from pathlib import Path
from typing import Any

import attr
import attrs
import pytest
from syrupy.constants import EXIT_STATUS_FAIL_UNUSED
from syrupy.data import Snapshot, SnapshotCollection, SnapshotCollections
from syrupy.extensions.amber import AmberDataSerializer, AmberSnapshotExtension
from syrupy.location import PyTestLocation
from syrupy.report import SnapshotReport
from syrupy.session import ItemStatus, SnapshotSession
from syrupy.types import PropertyFilter, PropertyMatcher, PropertyPath, SerializableData
from syrupy.utils import is_xdist_controller, is_xdist_worker
import voluptuous as vol
import voluptuous_serialize

Expand Down Expand Up @@ -254,157 +246,3 @@ def dirname(cls, *, test_location: PyTestLocation) -> str:
"""
test_dir = Path(test_location.filepath).parent
return str(test_dir.joinpath("snapshots"))


# Classes and Methods to override default finish behavior in syrupy
# This is needed to handle the xdist plugin in pytest
# The default implementation does not handle the xdist plugin
# and will not work correctly when running tests in parallel
# with pytest-xdist.
# Temporary workaround until it is finalised inside syrupy
# See https://github.com/syrupy-project/syrupy/pull/901


class _FakePytestObject:
"""Fake object."""

def __init__(self, collected_item: dict[str, str]) -> None:
"""Initialise fake object."""
self.__module__ = collected_item["modulename"]
self.__name__ = collected_item["methodname"]


class _FakePytestItem:
"""Fake pytest.Item object."""

def __init__(self, collected_item: dict[str, str]) -> None:
"""Initialise fake pytest.Item object."""
self.nodeid = collected_item["nodeid"]
self.name = collected_item["name"]
self.path = Path(collected_item["path"])
self.obj = _FakePytestObject(collected_item)


def _serialize_collections(collections: SnapshotCollections) -> dict[str, Any]:
return {
k: [c.name for c in v] for k, v in collections._snapshot_collections.items()
}


def _serialize_report(
report: SnapshotReport,
collected_items: set[pytest.Item],
selected_items: dict[str, ItemStatus],
) -> dict[str, Any]:
return {
"discovered": _serialize_collections(report.discovered),
"created": _serialize_collections(report.created),
"failed": _serialize_collections(report.failed),
"matched": _serialize_collections(report.matched),
"updated": _serialize_collections(report.updated),
"used": _serialize_collections(report.used),
"_collected_items": [
{
"nodeid": c.nodeid,
"name": c.name,
"path": str(c.path),
"modulename": c.obj.__module__,
"methodname": c.obj.__name__,
}
for c in list(collected_items)
],
"_selected_items": {
key: status.value for key, status in selected_items.items()
},
}


def _merge_serialized_collections(
collections: SnapshotCollections, json_data: dict[str, list[str]]
) -> None:
if not json_data:
return
for location, names in json_data.items():
snapshot_collection = SnapshotCollection(location=location)
for name in names:
snapshot_collection.add(Snapshot(name))
collections.update(snapshot_collection)


def _merge_serialized_report(report: SnapshotReport, json_data: dict[str, Any]) -> None:
_merge_serialized_collections(report.discovered, json_data["discovered"])
_merge_serialized_collections(report.created, json_data["created"])
_merge_serialized_collections(report.failed, json_data["failed"])
_merge_serialized_collections(report.matched, json_data["matched"])
_merge_serialized_collections(report.updated, json_data["updated"])
_merge_serialized_collections(report.used, json_data["used"])
for collected_item in json_data["_collected_items"]:
custom_item = _FakePytestItem(collected_item)
if not any(
t.nodeid == custom_item.nodeid and t.name == custom_item.nodeid
for t in report.collected_items
):
report.collected_items.add(custom_item)
for key, selected_item in json_data["_selected_items"].items():
if key in report.selected_items:
status = ItemStatus(selected_item)
if status != ItemStatus.NOT_RUN:
report.selected_items[key] = status
else:
report.selected_items[key] = ItemStatus(selected_item)


def override_syrupy_finish(self: SnapshotSession) -> int:
"""Override the finish method to allow for custom handling."""
exitstatus = 0
self.flush_snapshot_write_queue()
self.report = SnapshotReport(
base_dir=self.pytest_session.config.rootpath,
collected_items=self._collected_items,
selected_items=self._selected_items,
assertions=self._assertions,
options=self.pytest_session.config.option,
)

if is_xdist_worker():
with open(".pytest_syrupy_worker_count", "w", encoding="utf-8") as f:
f.write(os.getenv("PYTEST_XDIST_WORKER_COUNT"))
with open(
f".pytest_syrupy_{os.getenv("PYTEST_XDIST_WORKER")}_result",
"w",
encoding="utf-8",
) as f:
json.dump(
_serialize_report(
self.report, self._collected_items, self._selected_items
),
f,
indent=2,
)
return exitstatus
if is_xdist_controller():
return exitstatus

worker_count = None
try:
with open(".pytest_syrupy_worker_count", encoding="utf-8") as f:
worker_count = f.read()
os.remove(".pytest_syrupy_worker_count")
except FileNotFoundError:
pass

if worker_count:
for i in range(int(worker_count)):
with open(f".pytest_syrupy_gw{i}_result", encoding="utf-8") as f:
_merge_serialized_report(self.report, json.load(f))
os.remove(f".pytest_syrupy_gw{i}_result")

if self.report.num_unused:
if self.update_snapshots:
self.remove_unused_snapshots(
unused_snapshot_collections=self.report.unused,
used_snapshot_collections=self.report.used,
)
elif not self.warn_unused_snapshots:
exitstatus |= EXIT_STATUS_FAIL_UNUSED
return exitstatus

0 comments on commit 35b7c30

Please sign in to comment.