test: make CAddress in functional tests comparable

This way we can compare CAddress objects using `==` or even
arrays of CAddress using `array1 == array2`.

1 files changed, 3 insertions, 0 deletions

diff --git a/test/functional/test_framework/messages.py b/test/functional/test_framework/messages.py
index a21d7e7946..fdc254e5fc 100755
--- a/test/functional/test_framework/messages.py
+++ b/test/functional/test_framework/messages.py
@@ -229,6 +229,9 @@ class CAddress:
         self.ip = "0.0.0.0"
         self.port = 0
 
+    def __eq__(self, other):
+        return self.net == other.net and self.ip == other.ip and self.nServices == other.nServices and self.port == other.port and self.time == other.time
+
     def deserialize(self, f, *, with_time=True):
         """Deserialize from addrv1 format (pre-BIP155)"""
         if with_time: