mirror of
https://git.anonymousland.org/anonymousland/synapse.git
synced 2024-10-01 11:49:51 -04:00
74f01e11c9
Outlier events don't ever have push actions associated with them, so we can skip some expensive queries during event persistence.
213 lines
7.0 KiB
Python
213 lines
7.0 KiB
Python
# 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 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, clock, hs):
|
|
self.store = hs.get_datastore()
|
|
self.persist_events_store = hs.get_datastores().persist_events
|
|
|
|
def test_get_unread_push_actions_for_user_in_range_for_http(self):
|
|
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):
|
|
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):
|
|
room_id = "!foo:example.com"
|
|
user_id = "@user1235:example.com"
|
|
|
|
def _assert_counts(noitf_count, highlight_count):
|
|
counts = self.get_success(
|
|
self.store.db_pool.runInteraction(
|
|
"", self.store._get_unread_counts_by_pos_txn, room_id, user_id, 0
|
|
)
|
|
)
|
|
self.assertEquals(
|
|
counts,
|
|
{
|
|
"notify_count": noitf_count,
|
|
"unread_count": 0, # Unread counts are tested in the sync tests.
|
|
"highlight_count": highlight_count,
|
|
},
|
|
)
|
|
|
|
def _inject_actions(stream, action):
|
|
event = Mock()
|
|
event.room_id = room_id
|
|
event.event_id = "$test:example.com"
|
|
event.internal_metadata.stream_ordering = stream
|
|
event.internal_metadata.is_outlier.return_value = False
|
|
event.depth = stream
|
|
|
|
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):
|
|
self.get_success(
|
|
self.store.db_pool.runInteraction(
|
|
"", self.store._rotate_notifs_before_txn, stream
|
|
)
|
|
)
|
|
|
|
def _mark_read(stream, depth):
|
|
self.get_success(
|
|
self.store.db_pool.runInteraction(
|
|
"",
|
|
self.store._remove_old_push_actions_before_txn,
|
|
room_id,
|
|
user_id,
|
|
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(7, 7)
|
|
_assert_counts(0, 0)
|
|
|
|
_inject_actions(8, HIGHLIGHT)
|
|
_assert_counts(1, 1)
|
|
_rotate(9)
|
|
_assert_counts(1, 1)
|
|
_rotate(10)
|
|
_assert_counts(1, 1)
|
|
|
|
def test_find_first_stream_ordering_after_ts(self):
|
|
def add_event(so, ts):
|
|
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)
|