|
@@ -120,7 +120,7 @@ class CleanupExtremBackgroundUpdateStoreTestCase(HomeserverTestCase):
|
|
|
self.store.get_latest_event_ids_in_room(self.room_id)
|
|
|
)
|
|
|
|
|
|
- self.assertEqual(latest_event_ids, [event_id_4])
|
|
|
+ self.assertEqual(latest_event_ids, {event_id_4})
|
|
|
|
|
|
def test_basic_cleanup(self) -> None:
|
|
|
"""Test that extremities are correctly calculated in the presence of
|
|
@@ -147,7 +147,7 @@ class CleanupExtremBackgroundUpdateStoreTestCase(HomeserverTestCase):
|
|
|
latest_event_ids = self.get_success(
|
|
|
self.store.get_latest_event_ids_in_room(self.room_id)
|
|
|
)
|
|
|
- self.assertEqual(set(latest_event_ids), {event_id_a, event_id_b})
|
|
|
+ self.assertEqual(latest_event_ids, {event_id_a, event_id_b})
|
|
|
|
|
|
# Run the background update and check it did the right thing
|
|
|
self.run_background_update()
|
|
@@ -155,7 +155,7 @@ class CleanupExtremBackgroundUpdateStoreTestCase(HomeserverTestCase):
|
|
|
latest_event_ids = self.get_success(
|
|
|
self.store.get_latest_event_ids_in_room(self.room_id)
|
|
|
)
|
|
|
- self.assertEqual(latest_event_ids, [event_id_b])
|
|
|
+ self.assertEqual(latest_event_ids, {event_id_b})
|
|
|
|
|
|
def test_chain_of_fail_cleanup(self) -> None:
|
|
|
"""Test that extremities are correctly calculated in the presence of
|
|
@@ -185,7 +185,7 @@ class CleanupExtremBackgroundUpdateStoreTestCase(HomeserverTestCase):
|
|
|
latest_event_ids = self.get_success(
|
|
|
self.store.get_latest_event_ids_in_room(self.room_id)
|
|
|
)
|
|
|
- self.assertEqual(set(latest_event_ids), {event_id_a, event_id_b})
|
|
|
+ self.assertEqual(latest_event_ids, {event_id_a, event_id_b})
|
|
|
|
|
|
# Run the background update and check it did the right thing
|
|
|
self.run_background_update()
|
|
@@ -193,7 +193,7 @@ class CleanupExtremBackgroundUpdateStoreTestCase(HomeserverTestCase):
|
|
|
latest_event_ids = self.get_success(
|
|
|
self.store.get_latest_event_ids_in_room(self.room_id)
|
|
|
)
|
|
|
- self.assertEqual(latest_event_ids, [event_id_b])
|
|
|
+ self.assertEqual(latest_event_ids, {event_id_b})
|
|
|
|
|
|
def test_forked_graph_cleanup(self) -> None:
|
|
|
r"""Test that extremities are correctly calculated in the presence of
|
|
@@ -240,7 +240,7 @@ class CleanupExtremBackgroundUpdateStoreTestCase(HomeserverTestCase):
|
|
|
latest_event_ids = self.get_success(
|
|
|
self.store.get_latest_event_ids_in_room(self.room_id)
|
|
|
)
|
|
|
- self.assertEqual(set(latest_event_ids), {event_id_a, event_id_b, event_id_c})
|
|
|
+ self.assertEqual(latest_event_ids, {event_id_a, event_id_b, event_id_c})
|
|
|
|
|
|
# Run the background update and check it did the right thing
|
|
|
self.run_background_update()
|
|
@@ -248,7 +248,7 @@ class CleanupExtremBackgroundUpdateStoreTestCase(HomeserverTestCase):
|
|
|
latest_event_ids = self.get_success(
|
|
|
self.store.get_latest_event_ids_in_room(self.room_id)
|
|
|
)
|
|
|
- self.assertEqual(set(latest_event_ids), {event_id_b, event_id_c})
|
|
|
+ self.assertEqual(latest_event_ids, {event_id_b, event_id_c})
|
|
|
|
|
|
|
|
|
class CleanupExtremDummyEventsTestCase(HomeserverTestCase):
|