123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263 |
- # Copyright 2016-2021 The Matrix.org Foundation C.I.C.
- #
- # Licensed under the Apache License, Version 2.0 (the "License");
- # you may not use this file except in compliance with the License.
- # You may obtain a copy of the License at
- #
- # http://www.apache.org/licenses/LICENSE-2.0
- #
- # Unless required by applicable law or agreed to in writing, software
- # distributed under the License is distributed on an "AS IS" BASIS,
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- # See the License for the specific language governing permissions and
- # limitations under the License.
- from unittest.mock import Mock
- from twisted.test.proto_helpers import MemoryReactor
- from synapse.server import HomeServer
- from synapse.storage.databases.main.event_push_actions import NotifCounts
- from synapse.util import Clock
- from tests.unittest import HomeserverTestCase
- USER_ID = "@user:example.com"
- PlAIN_NOTIF = ["notify", {"set_tweak": "highlight", "value": False}]
- HIGHLIGHT = [
- "notify",
- {"set_tweak": "sound", "value": "default"},
- {"set_tweak": "highlight"},
- ]
- class EventPushActionsStoreTestCase(HomeserverTestCase):
- def prepare(self, reactor: MemoryReactor, clock: Clock, hs: HomeServer) -> None:
- self.store = hs.get_datastores().main
- persist_events_store = hs.get_datastores().persist_events
- assert persist_events_store is not None
- self.persist_events_store = persist_events_store
- def test_get_unread_push_actions_for_user_in_range_for_http(self) -> None:
- self.get_success(
- self.store.get_unread_push_actions_for_user_in_range_for_http(
- USER_ID, 0, 1000, 20
- )
- )
- def test_get_unread_push_actions_for_user_in_range_for_email(self) -> None:
- self.get_success(
- self.store.get_unread_push_actions_for_user_in_range_for_email(
- USER_ID, 0, 1000, 20
- )
- )
- def test_count_aggregation(self) -> None:
- room_id = "!foo:example.com"
- user_id = "@user1235:test"
- last_read_stream_ordering = [0]
- def _assert_counts(noitf_count: int, highlight_count: int) -> None:
- counts = self.get_success(
- self.store.db_pool.runInteraction(
- "",
- self.store._get_unread_counts_by_pos_txn,
- room_id,
- user_id,
- last_read_stream_ordering[0],
- )
- )
- self.assertEqual(
- counts,
- NotifCounts(
- notify_count=noitf_count,
- unread_count=0, # Unread counts are tested in the sync tests.
- highlight_count=highlight_count,
- ),
- )
- def _inject_actions(stream: int, action: list) -> None:
- event = Mock()
- event.room_id = room_id
- event.event_id = f"$test{stream}:example.com"
- event.internal_metadata.stream_ordering = stream
- event.internal_metadata.is_outlier.return_value = False
- event.depth = stream
- self.get_success(
- self.store.db_pool.simple_insert(
- table="events",
- values={
- "stream_ordering": stream,
- "topological_ordering": stream,
- "type": "m.room.message",
- "room_id": room_id,
- "processed": True,
- "outlier": False,
- "event_id": event.event_id,
- },
- )
- )
- self.get_success(
- self.store.add_push_actions_to_staging(
- event.event_id,
- {user_id: action},
- False,
- )
- )
- self.get_success(
- self.store.db_pool.runInteraction(
- "",
- self.persist_events_store._set_push_actions_for_event_and_users_txn,
- [(event, None)],
- [(event, None)],
- )
- )
- def _rotate(stream: int) -> None:
- self.get_success(
- self.store.db_pool.runInteraction(
- "rotate-receipts", self.store._handle_new_receipts_for_notifs_txn
- )
- )
- self.get_success(
- self.store.db_pool.runInteraction(
- "rotate-notifs", self.store._rotate_notifs_before_txn, stream
- )
- )
- def _mark_read(stream: int, depth: int) -> None:
- last_read_stream_ordering[0] = stream
- self.get_success(
- self.store.db_pool.runInteraction(
- "",
- self.store._insert_linearized_receipt_txn,
- room_id,
- "m.read",
- user_id,
- f"$test{stream}:example.com",
- {},
- stream,
- )
- )
- _assert_counts(0, 0)
- _inject_actions(1, PlAIN_NOTIF)
- _assert_counts(1, 0)
- _rotate(2)
- _assert_counts(1, 0)
- _inject_actions(3, PlAIN_NOTIF)
- _assert_counts(2, 0)
- _rotate(4)
- _assert_counts(2, 0)
- _inject_actions(5, PlAIN_NOTIF)
- _mark_read(3, 3)
- _assert_counts(1, 0)
- _mark_read(5, 5)
- _assert_counts(0, 0)
- _inject_actions(6, PlAIN_NOTIF)
- _rotate(7)
- self.get_success(
- self.store.db_pool.simple_delete(
- table="event_push_actions", keyvalues={"1": 1}, desc=""
- )
- )
- _assert_counts(1, 0)
- _mark_read(6, 6)
- _assert_counts(0, 0)
- _inject_actions(8, HIGHLIGHT)
- _assert_counts(1, 1)
- _rotate(9)
- _assert_counts(1, 1)
- # Check that adding another notification and rotating after highlight
- # works.
- _inject_actions(10, PlAIN_NOTIF)
- _rotate(11)
- _assert_counts(2, 1)
- # Check that sending read receipts at different points results in the
- # right counts.
- _mark_read(8, 8)
- _assert_counts(1, 0)
- _mark_read(10, 10)
- _assert_counts(0, 0)
- def test_find_first_stream_ordering_after_ts(self) -> None:
- def add_event(so: int, ts: int) -> None:
- self.get_success(
- self.store.db_pool.simple_insert(
- "events",
- {
- "stream_ordering": so,
- "received_ts": ts,
- "event_id": "event%i" % so,
- "type": "",
- "room_id": "",
- "content": "",
- "processed": True,
- "outlier": False,
- "topological_ordering": 0,
- "depth": 0,
- },
- )
- )
- # start with the base case where there are no events in the table
- r = self.get_success(self.store.find_first_stream_ordering_after_ts(11))
- self.assertEqual(r, 0)
- # now with one event
- add_event(2, 10)
- r = self.get_success(self.store.find_first_stream_ordering_after_ts(9))
- self.assertEqual(r, 2)
- r = self.get_success(self.store.find_first_stream_ordering_after_ts(10))
- self.assertEqual(r, 2)
- r = self.get_success(self.store.find_first_stream_ordering_after_ts(11))
- self.assertEqual(r, 3)
- # add a bunch of dummy events to the events table
- for (stream_ordering, ts) in (
- (3, 110),
- (4, 120),
- (5, 120),
- (10, 130),
- (20, 140),
- ):
- add_event(stream_ordering, ts)
- r = self.get_success(self.store.find_first_stream_ordering_after_ts(110))
- self.assertEqual(r, 3, "First event after 110ms should be 3, was %i" % r)
- # 4 and 5 are both after 120: we want 4 rather than 5
- r = self.get_success(self.store.find_first_stream_ordering_after_ts(120))
- self.assertEqual(r, 4, "First event after 120ms should be 4, was %i" % r)
- r = self.get_success(self.store.find_first_stream_ordering_after_ts(129))
- self.assertEqual(r, 10, "First event after 129ms should be 10, was %i" % r)
- # check we can get the last event
- r = self.get_success(self.store.find_first_stream_ordering_after_ts(140))
- self.assertEqual(r, 20, "First event after 14ms should be 20, was %i" % r)
- # off the end
- r = self.get_success(self.store.find_first_stream_ordering_after_ts(160))
- self.assertEqual(r, 21)
- # check we can find an event at ordering zero
- add_event(0, 5)
- r = self.get_success(self.store.find_first_stream_ordering_after_ts(1))
- self.assertEqual(r, 0)
|