mirror of
https://github.com/home-assistant/core.git
synced 2025-12-24 12:59:34 +00:00
Add tests.common.ANY (#84240)
This commit is contained in:
@@ -1303,6 +1303,38 @@ def assert_lists_same(a, b):
|
||||
assert i in a
|
||||
|
||||
|
||||
_SENTINEL = object()
|
||||
|
||||
|
||||
class _HA_ANY:
|
||||
"""A helper object that compares equal to everything.
|
||||
|
||||
Based on unittest.mock.ANY, but modified to not show up in pytest's equality
|
||||
assertion diffs.
|
||||
"""
|
||||
|
||||
_other = _SENTINEL
|
||||
|
||||
def __eq__(self, other):
|
||||
"""Test equal."""
|
||||
self._other = other
|
||||
return True
|
||||
|
||||
def __ne__(self, other):
|
||||
"""Test not equal."""
|
||||
self._other = other
|
||||
return False
|
||||
|
||||
def __repr__(self):
|
||||
"""Return repr() other to not show up in pytest quality diffs."""
|
||||
if self._other is _SENTINEL:
|
||||
return "<ANY>"
|
||||
return repr(self._other)
|
||||
|
||||
|
||||
ANY = _HA_ANY()
|
||||
|
||||
|
||||
def raise_contains_mocks(val):
|
||||
"""Raise for mocks."""
|
||||
if isinstance(val, Mock):
|
||||
|
||||
Reference in New Issue
Block a user