[checkmk-commits] Make DiscoveryResult iterable

Module: check_mk Branch: master Commit: 4e8fdbbf0e53b504d367debd0fe23d3e814121f1 URL: http://git.mathias-kettner.de/git/?p=check_mk.git;a=commit;h=4e8fdbbf0e53b5… Author: Moritz Kiemer <mo(a)mathias-kettner.de> Date: Fri Jun 15 13:15:59 2018 +0200 Make DiscoveryResult iterable This makes two instances comparable to one another. Change-Id: Ifa562122039c75ca16988e20165634bf2db07695 --- tests/unit/checks/checktestlib.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/unit/checks/checktestlib.py b/tests/unit/checks/checktestlib.py index 44333f4..203e24a 100644 --- a/tests/unit/checks/checktestlib.py +++ b/tests/unit/checks/checktestlib.py @@ -192,11 +192,14 @@ class DiscoveryResult(object): self.entries.append(DiscoveryEntry(entry)) def __eq__(self, other_value): - return all(entry in other_value for entry in self.entries) and \ - all(other_entry in self.entries for other_entry in other_value) + return all(entry in other_value for entry in self) and \ + all(other_entry in self for other_entry in other_value) def __contains__(self, value): return value in self.entries + def __iter__(self): + return iter(self.entries) + def __repr__(self): return "DiscoveryResult(%s)" % (",\n".join([repr(entry) for entry in self.entries]))