Add explicit option for partial state rooms

1 files changed, 5 insertions, 0 deletions

diff --git a/tests/test_visibility.py b/tests/test_visibility.py

index 569a01a417..f758232d08 100644
--- a/tests/test_visibility.py
+++ b/tests/test_visibility.py
@@ -69,6 +69,7 @@ class FilterEventsForServerTestCase(unittest.HomeserverTestCase):
                 events_to_filter,
                 redact=True,
                 filter_out_erased_senders=True,
+                filter_out_partial_state_rooms=True,
             )
         )
 
@@ -96,6 +97,7 @@ class FilterEventsForServerTestCase(unittest.HomeserverTestCase):
                     [outlier],
                     redact=True,
                     filter_out_erased_senders=True,
+                    filter_out_partial_state_rooms=True,
                 )
             ),
             [outlier],
@@ -112,6 +114,7 @@ class FilterEventsForServerTestCase(unittest.HomeserverTestCase):
                 [outlier, evt],
                 redact=True,
                 filter_out_erased_senders=True,
+                filter_out_partial_state_rooms=True,
             )
         )
         self.assertEqual(len(filtered), 2, f"expected 2 results, got: {filtered}")
@@ -129,6 +132,7 @@ class FilterEventsForServerTestCase(unittest.HomeserverTestCase):
                 [outlier, evt],
                 redact=True,
                 filter_out_erased_senders=True,
+                filter_out_partial_state_rooms=True,
             )
         )
         self.assertEqual(filtered[0], outlier)
@@ -169,6 +173,7 @@ class FilterEventsForServerTestCase(unittest.HomeserverTestCase):
                 events_to_filter,
                 redact=True,
                 filter_out_erased_senders=True,
+                filter_out_partial_state_rooms=True,
             )
         )