Add tests for get_latest_event_ids_in_room and get_current_state

1 files changed, 9 insertions, 0 deletions

diff --git a/synapse/events/__init__.py b/synapse/events/__init__.py
index 13154b1723..81e2126202 100644
--- a/synapse/events/__init__.py
+++ b/synapse/events/__init__.py
@@ -36,6 +36,10 @@ class _EventInternalMetadata(object):
     def is_invite_from_remote(self):
         return getattr(self, "invite_from_remote", False)
 
+    def __eq__(self, other):
+        "Equality check for unit tests."
+        return self.__dict__ == other.__dict__
+
 
 def _event_dict_property(key):
     def getter(self):
@@ -180,3 +184,8 @@ class FrozenEvent(EventBase):
             self.get("type", None),
             self.get("state_key", None),
         )
+
+    def __eq__(self, other):
+        """Equality check for unit tests. Compares internal_metadata as well
+        as the event fields"""
+        return self.__dict__ == other.__dict__