Implement __eq__ for FRRoutingPrefix & CabourotteHealthcheck

gecio · Dec 5, 2023 · 644786d · 644786d
1 parent 2b8e0bb
commit 644786d
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/anycastd/healthcheck/_cabourotte/main.py b/src/anycastd/healthcheck/_cabourotte/main.py
@@ -25,6 +25,12 @@ def __repr__(self) -> str:
             f"interval={self.interval!r})"
         )
 
+    def __eq__(self, other: object) -> bool:
+        if not isinstance(other, CabourotteHealthcheck):
+            return NotImplemented
+
+        return self.__dict__ == other.__dict__
+
     @property
     def interval(self) -> datetime.timedelta:
         return self.__interval

diff --git a/src/anycastd/prefix/_frrouting/main.py b/src/anycastd/prefix/_frrouting/main.py
@@ -47,6 +47,12 @@ def __repr__(self) -> str:
             f"vtysh={self.vtysh!r}, executor={self.executor!r})"
         )
 
+    def __eq__(self, other: object) -> bool:
+        if not isinstance(other, FRRoutingPrefix):
+            return NotImplemented
+
+        return self.__dict__ == other.__dict__
+
     @property
     def prefix(self) -> IPv4Network | IPv6Network:
         return self.__prefix