Implement __eq__ for FRRoutingPrefix & CabourotteHealthcheck (#21)

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

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

tests/healthcheck/cabourotte/test_main.py

@@ -55,6 +55,28 @@ def test_repr(attributes: dict):
     )
 
 
+def test_equal():
+    """Two healthchecks with the same attributes are equal."""
+    healthcheck1 = CabourotteHealthcheck(
+        "test", url="https://example.com", interval=datetime.timedelta(seconds=10)
+    )
+    healthcheck2 = CabourotteHealthcheck(
+        "test", url="https://example.com", interval=datetime.timedelta(seconds=10)
+    )
+    assert healthcheck1 == healthcheck2
+
+
+def test_non_equal():
+    """Two healthchecks with different attributes are not equal."""
+    healthcheck1 = CabourotteHealthcheck(
+        "test", url="https://example.com", interval=datetime.timedelta(seconds=10)
+    )
+    healthcheck2 = CabourotteHealthcheck(
+        "test", url="https://example.com", interval=datetime.timedelta(seconds=20)
+    )
+    assert healthcheck1 != healthcheck2
+
+
 @pytest.mark.asyncio
 async def test__check_awaits_get_result(mocker: MockerFixture):
     """The check method awaits the result of get_result."""

tests/prefix/frrouting/test_main.py

@@ -1,3 +1,4 @@
+from ipaddress import IPv6Network
 from pathlib import Path
 
 from anycastd._executor import LocalExecutor
@@ -18,3 +19,25 @@ def test_repr(example_networks, example_vrfs):
         f"FRRoutingPrefix(prefix={example_networks!r}, vrf={example_vrfs!r}, "
         f"vtysh={vtysh!r}, executor={executor!r})"
     )
+
+
+def test_equal():
+    """Two prefixes with the same attributes are equal."""
+    prefix1 = FRRoutingPrefix(
+        prefix=IPv6Network("2001:db8::/32"), vrf="42", executor=LocalExecutor()
+    )
+    prefix2 = FRRoutingPrefix(
+        prefix=IPv6Network("2001:db8::/32"), vrf="42", executor=LocalExecutor()
+    )
+    assert prefix1 == prefix2
+
+
+def test_non_equal():
+    """Two prefixes with different attributes are not equal."""
+    prefix1 = FRRoutingPrefix(
+        prefix=IPv6Network("2001:db8::/32"), vrf="42", executor=LocalExecutor()
+    )
+    prefix2 = FRRoutingPrefix(
+        prefix=IPv6Network("2001:db8::/32"), vrf="43", executor=LocalExecutor()
+    )
+    assert prefix1 != prefix2