mirror of
https://github.com/element-hq/synapse
synced 2024-08-20 08:30:21 +00:00
594 lines
19 KiB
Python
594 lines
19 KiB
Python
#
|
|
# This file is licensed under the Affero General Public License (AGPL) version 3.
|
|
#
|
|
# Copyright (C) 2023 New Vector, Ltd
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as
|
|
# published by the Free Software Foundation, either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# See the GNU Affero General Public License for more details:
|
|
# <https://www.gnu.org/licenses/agpl-3.0.html>.
|
|
#
|
|
# Originally licensed under the Apache License, Version 2.0:
|
|
# <http://www.apache.org/licenses/LICENSE-2.0>.
|
|
#
|
|
# [This file includes modifications made by New Vector Limited]
|
|
#
|
|
#
|
|
|
|
from typing import Any, Dict, List, Optional, Tuple, cast
|
|
|
|
from twisted.test.proto_helpers import MemoryReactor
|
|
|
|
from synapse.rest import admin
|
|
from synapse.rest.client import login, room
|
|
from synapse.server import HomeServer
|
|
from synapse.storage.databases.main import stats
|
|
from synapse.util import Clock
|
|
|
|
from tests import unittest
|
|
|
|
# The expected number of state events in a fresh public room.
|
|
EXPT_NUM_STATE_EVTS_IN_FRESH_PUBLIC_ROOM = 5
|
|
# The expected number of state events in a fresh private room.
|
|
EXPT_NUM_STATE_EVTS_IN_FRESH_PRIVATE_ROOM = 6
|
|
|
|
|
|
class StatsRoomTests(unittest.HomeserverTestCase):
|
|
servlets = [
|
|
admin.register_servlets_for_client_rest_resource,
|
|
room.register_servlets,
|
|
login.register_servlets,
|
|
]
|
|
|
|
def prepare(self, reactor: MemoryReactor, clock: Clock, hs: HomeServer) -> None:
|
|
self.store = hs.get_datastores().main
|
|
self.handler = self.hs.get_stats_handler()
|
|
|
|
def _add_background_updates(self) -> None:
|
|
"""
|
|
Add the background updates we need to run.
|
|
"""
|
|
# Ugh, have to reset this flag
|
|
self.store.db_pool.updates._all_done = False
|
|
|
|
self.get_success(
|
|
self.store.db_pool.simple_insert(
|
|
"background_updates",
|
|
{
|
|
"update_name": "populate_stats_process_rooms",
|
|
"progress_json": "{}",
|
|
},
|
|
)
|
|
)
|
|
self.get_success(
|
|
self.store.db_pool.simple_insert(
|
|
"background_updates",
|
|
{
|
|
"update_name": "populate_stats_process_users",
|
|
"progress_json": "{}",
|
|
"depends_on": "populate_stats_process_rooms",
|
|
},
|
|
)
|
|
)
|
|
|
|
async def get_all_room_state(self) -> List[Optional[str]]:
|
|
rows = cast(
|
|
List[Tuple[Optional[str]]],
|
|
await self.store.db_pool.simple_select_list(
|
|
"room_stats_state", None, retcols=("topic",)
|
|
),
|
|
)
|
|
return [r[0] for r in rows]
|
|
|
|
def _get_current_stats(
|
|
self, stats_type: str, stat_id: str
|
|
) -> Optional[Dict[str, Any]]:
|
|
table, id_col = stats.TYPE_TO_TABLE[stats_type]
|
|
|
|
cols = list(stats.ABSOLUTE_STATS_FIELDS[stats_type])
|
|
|
|
row = self.get_success(
|
|
self.store.db_pool.simple_select_one(
|
|
table + "_current",
|
|
{id_col: stat_id},
|
|
cols,
|
|
allow_none=True,
|
|
)
|
|
)
|
|
|
|
return None if row is None else dict(zip(cols, row))
|
|
|
|
def _perform_background_initial_update(self) -> None:
|
|
# Do the initial population of the stats via the background update
|
|
self._add_background_updates()
|
|
|
|
self.wait_for_background_updates()
|
|
|
|
def test_initial_room(self) -> None:
|
|
"""
|
|
The background updates will build the table from scratch.
|
|
"""
|
|
r = self.get_success(self.get_all_room_state())
|
|
self.assertEqual(len(r), 0)
|
|
|
|
# Disable stats
|
|
self.hs.config.stats.stats_enabled = False
|
|
self.handler.stats_enabled = False
|
|
|
|
u1 = self.register_user("u1", "pass")
|
|
u1_token = self.login("u1", "pass")
|
|
|
|
room_1 = self.helper.create_room_as(u1, tok=u1_token)
|
|
self.helper.send_state(
|
|
room_1, event_type="m.room.topic", body={"topic": "foo"}, tok=u1_token
|
|
)
|
|
|
|
# Stats disabled, shouldn't have done anything
|
|
r = self.get_success(self.get_all_room_state())
|
|
self.assertEqual(len(r), 0)
|
|
|
|
# Enable stats
|
|
self.hs.config.stats.stats_enabled = True
|
|
self.handler.stats_enabled = True
|
|
|
|
# Do the initial population of the user directory via the background update
|
|
self._add_background_updates()
|
|
|
|
self.wait_for_background_updates()
|
|
|
|
r = self.get_success(self.get_all_room_state())
|
|
|
|
self.assertEqual(len(r), 1)
|
|
self.assertEqual(r[0], "foo")
|
|
|
|
def test_create_user(self) -> None:
|
|
"""
|
|
When we create a user, it should have statistics already ready.
|
|
"""
|
|
|
|
u1 = self.register_user("u1", "pass")
|
|
|
|
u1stats = self._get_current_stats("user", u1)
|
|
|
|
assert u1stats is not None
|
|
|
|
# not in any rooms by default
|
|
self.assertEqual(u1stats["joined_rooms"], 0)
|
|
|
|
def test_create_room(self) -> None:
|
|
"""
|
|
When we create a room, it should have statistics already ready.
|
|
"""
|
|
|
|
self._perform_background_initial_update()
|
|
|
|
u1 = self.register_user("u1", "pass")
|
|
u1token = self.login("u1", "pass")
|
|
r1 = self.helper.create_room_as(u1, tok=u1token)
|
|
r1stats = self._get_current_stats("room", r1)
|
|
r2 = self.helper.create_room_as(u1, tok=u1token, is_public=False)
|
|
r2stats = self._get_current_stats("room", r2)
|
|
|
|
assert r1stats is not None
|
|
assert r2stats is not None
|
|
|
|
self.assertEqual(
|
|
r1stats["current_state_events"], EXPT_NUM_STATE_EVTS_IN_FRESH_PUBLIC_ROOM
|
|
)
|
|
self.assertEqual(
|
|
r2stats["current_state_events"], EXPT_NUM_STATE_EVTS_IN_FRESH_PRIVATE_ROOM
|
|
)
|
|
|
|
self.assertEqual(r1stats["joined_members"], 1)
|
|
self.assertEqual(r1stats["invited_members"], 0)
|
|
self.assertEqual(r1stats["banned_members"], 0)
|
|
|
|
self.assertEqual(r2stats["joined_members"], 1)
|
|
self.assertEqual(r2stats["invited_members"], 0)
|
|
self.assertEqual(r2stats["banned_members"], 0)
|
|
|
|
def test_updating_profile_information_does_not_increase_joined_members_count(
|
|
self,
|
|
) -> None:
|
|
"""
|
|
Check that the joined_members count does not increase when a user changes their
|
|
profile information (which is done by sending another join membership event into
|
|
the room.
|
|
"""
|
|
self._perform_background_initial_update()
|
|
|
|
# Create a user and room
|
|
u1 = self.register_user("u1", "pass")
|
|
u1token = self.login("u1", "pass")
|
|
r1 = self.helper.create_room_as(u1, tok=u1token)
|
|
|
|
# Get the current room stats
|
|
r1stats_ante = self._get_current_stats("room", r1)
|
|
assert r1stats_ante is not None
|
|
|
|
# Send a profile update into the room
|
|
new_profile = {"displayname": "bob"}
|
|
self.helper.change_membership(
|
|
r1, u1, u1, "join", extra_data=new_profile, tok=u1token
|
|
)
|
|
|
|
# Get the new room stats
|
|
r1stats_post = self._get_current_stats("room", r1)
|
|
assert r1stats_post is not None
|
|
|
|
# Ensure that the user count did not changed
|
|
self.assertEqual(r1stats_post["joined_members"], r1stats_ante["joined_members"])
|
|
self.assertEqual(
|
|
r1stats_post["local_users_in_room"], r1stats_ante["local_users_in_room"]
|
|
)
|
|
|
|
def test_send_state_event_nonoverwriting(self) -> None:
|
|
"""
|
|
When we send a non-overwriting state event, it increments current_state_events
|
|
"""
|
|
|
|
self._perform_background_initial_update()
|
|
|
|
u1 = self.register_user("u1", "pass")
|
|
u1token = self.login("u1", "pass")
|
|
r1 = self.helper.create_room_as(u1, tok=u1token)
|
|
|
|
self.helper.send_state(
|
|
r1, "cat.hissing", {"value": True}, tok=u1token, state_key="tabby"
|
|
)
|
|
|
|
r1stats_ante = self._get_current_stats("room", r1)
|
|
assert r1stats_ante is not None
|
|
|
|
self.helper.send_state(
|
|
r1, "cat.hissing", {"value": False}, tok=u1token, state_key="moggy"
|
|
)
|
|
|
|
r1stats_post = self._get_current_stats("room", r1)
|
|
assert r1stats_post is not None
|
|
|
|
self.assertEqual(
|
|
r1stats_post["current_state_events"] - r1stats_ante["current_state_events"],
|
|
1,
|
|
)
|
|
|
|
def test_join_first_time(self) -> None:
|
|
"""
|
|
When a user joins a room for the first time, current_state_events and
|
|
joined_members should increase by exactly 1.
|
|
"""
|
|
|
|
self._perform_background_initial_update()
|
|
|
|
u1 = self.register_user("u1", "pass")
|
|
u1token = self.login("u1", "pass")
|
|
r1 = self.helper.create_room_as(u1, tok=u1token)
|
|
|
|
u2 = self.register_user("u2", "pass")
|
|
u2token = self.login("u2", "pass")
|
|
|
|
r1stats_ante = self._get_current_stats("room", r1)
|
|
assert r1stats_ante is not None
|
|
|
|
self.helper.join(r1, u2, tok=u2token)
|
|
|
|
r1stats_post = self._get_current_stats("room", r1)
|
|
assert r1stats_post is not None
|
|
|
|
self.assertEqual(
|
|
r1stats_post["current_state_events"] - r1stats_ante["current_state_events"],
|
|
1,
|
|
)
|
|
self.assertEqual(
|
|
r1stats_post["joined_members"] - r1stats_ante["joined_members"], 1
|
|
)
|
|
|
|
def test_join_after_leave(self) -> None:
|
|
"""
|
|
When a user joins a room after being previously left,
|
|
joined_members should increase by exactly 1.
|
|
current_state_events should not increase.
|
|
left_members should decrease by exactly 1.
|
|
"""
|
|
|
|
self._perform_background_initial_update()
|
|
|
|
u1 = self.register_user("u1", "pass")
|
|
u1token = self.login("u1", "pass")
|
|
r1 = self.helper.create_room_as(u1, tok=u1token)
|
|
|
|
u2 = self.register_user("u2", "pass")
|
|
u2token = self.login("u2", "pass")
|
|
|
|
self.helper.join(r1, u2, tok=u2token)
|
|
self.helper.leave(r1, u2, tok=u2token)
|
|
|
|
r1stats_ante = self._get_current_stats("room", r1)
|
|
assert r1stats_ante is not None
|
|
|
|
self.helper.join(r1, u2, tok=u2token)
|
|
|
|
r1stats_post = self._get_current_stats("room", r1)
|
|
assert r1stats_post is not None
|
|
|
|
self.assertEqual(
|
|
r1stats_post["current_state_events"] - r1stats_ante["current_state_events"],
|
|
0,
|
|
)
|
|
self.assertEqual(
|
|
r1stats_post["joined_members"] - r1stats_ante["joined_members"], +1
|
|
)
|
|
self.assertEqual(
|
|
r1stats_post["left_members"] - r1stats_ante["left_members"], -1
|
|
)
|
|
|
|
def test_invited(self) -> None:
|
|
"""
|
|
When a user invites another user, current_state_events and
|
|
invited_members should increase by exactly 1.
|
|
"""
|
|
|
|
self._perform_background_initial_update()
|
|
|
|
u1 = self.register_user("u1", "pass")
|
|
u1token = self.login("u1", "pass")
|
|
r1 = self.helper.create_room_as(u1, tok=u1token)
|
|
|
|
u2 = self.register_user("u2", "pass")
|
|
|
|
r1stats_ante = self._get_current_stats("room", r1)
|
|
assert r1stats_ante is not None
|
|
|
|
self.helper.invite(r1, u1, u2, tok=u1token)
|
|
|
|
r1stats_post = self._get_current_stats("room", r1)
|
|
assert r1stats_post is not None
|
|
|
|
self.assertEqual(
|
|
r1stats_post["current_state_events"] - r1stats_ante["current_state_events"],
|
|
1,
|
|
)
|
|
self.assertEqual(
|
|
r1stats_post["invited_members"] - r1stats_ante["invited_members"], +1
|
|
)
|
|
|
|
def test_join_after_invite(self) -> None:
|
|
"""
|
|
When a user joins a room after being invited and
|
|
joined_members should increase by exactly 1.
|
|
current_state_events should not increase.
|
|
invited_members should decrease by exactly 1.
|
|
"""
|
|
|
|
self._perform_background_initial_update()
|
|
|
|
u1 = self.register_user("u1", "pass")
|
|
u1token = self.login("u1", "pass")
|
|
r1 = self.helper.create_room_as(u1, tok=u1token)
|
|
|
|
u2 = self.register_user("u2", "pass")
|
|
u2token = self.login("u2", "pass")
|
|
|
|
self.helper.invite(r1, u1, u2, tok=u1token)
|
|
|
|
r1stats_ante = self._get_current_stats("room", r1)
|
|
assert r1stats_ante is not None
|
|
|
|
self.helper.join(r1, u2, tok=u2token)
|
|
|
|
r1stats_post = self._get_current_stats("room", r1)
|
|
assert r1stats_post is not None
|
|
|
|
self.assertEqual(
|
|
r1stats_post["current_state_events"] - r1stats_ante["current_state_events"],
|
|
0,
|
|
)
|
|
self.assertEqual(
|
|
r1stats_post["joined_members"] - r1stats_ante["joined_members"], +1
|
|
)
|
|
self.assertEqual(
|
|
r1stats_post["invited_members"] - r1stats_ante["invited_members"], -1
|
|
)
|
|
|
|
def test_left(self) -> None:
|
|
"""
|
|
When a user leaves a room after joining and
|
|
left_members should increase by exactly 1.
|
|
current_state_events should not increase.
|
|
joined_members should decrease by exactly 1.
|
|
"""
|
|
|
|
self._perform_background_initial_update()
|
|
|
|
u1 = self.register_user("u1", "pass")
|
|
u1token = self.login("u1", "pass")
|
|
r1 = self.helper.create_room_as(u1, tok=u1token)
|
|
|
|
u2 = self.register_user("u2", "pass")
|
|
u2token = self.login("u2", "pass")
|
|
|
|
self.helper.join(r1, u2, tok=u2token)
|
|
|
|
r1stats_ante = self._get_current_stats("room", r1)
|
|
assert r1stats_ante is not None
|
|
|
|
self.helper.leave(r1, u2, tok=u2token)
|
|
|
|
r1stats_post = self._get_current_stats("room", r1)
|
|
assert r1stats_post is not None
|
|
|
|
self.assertEqual(
|
|
r1stats_post["current_state_events"] - r1stats_ante["current_state_events"],
|
|
0,
|
|
)
|
|
self.assertEqual(
|
|
r1stats_post["left_members"] - r1stats_ante["left_members"], +1
|
|
)
|
|
self.assertEqual(
|
|
r1stats_post["joined_members"] - r1stats_ante["joined_members"], -1
|
|
)
|
|
|
|
def test_banned(self) -> None:
|
|
"""
|
|
When a user is banned from a room after joining and
|
|
left_members should increase by exactly 1.
|
|
current_state_events should not increase.
|
|
banned_members should decrease by exactly 1.
|
|
"""
|
|
|
|
self._perform_background_initial_update()
|
|
|
|
u1 = self.register_user("u1", "pass")
|
|
u1token = self.login("u1", "pass")
|
|
r1 = self.helper.create_room_as(u1, tok=u1token)
|
|
|
|
u2 = self.register_user("u2", "pass")
|
|
u2token = self.login("u2", "pass")
|
|
|
|
self.helper.join(r1, u2, tok=u2token)
|
|
|
|
r1stats_ante = self._get_current_stats("room", r1)
|
|
assert r1stats_ante is not None
|
|
|
|
self.helper.change_membership(r1, u1, u2, "ban", tok=u1token)
|
|
|
|
r1stats_post = self._get_current_stats("room", r1)
|
|
assert r1stats_post is not None
|
|
|
|
self.assertEqual(
|
|
r1stats_post["current_state_events"] - r1stats_ante["current_state_events"],
|
|
0,
|
|
)
|
|
self.assertEqual(
|
|
r1stats_post["banned_members"] - r1stats_ante["banned_members"], +1
|
|
)
|
|
self.assertEqual(
|
|
r1stats_post["joined_members"] - r1stats_ante["joined_members"], -1
|
|
)
|
|
|
|
def test_initial_background_update(self) -> None:
|
|
"""
|
|
Test that statistics can be generated by the initial background update
|
|
handler.
|
|
|
|
This test also tests that stats rows are not created for new subjects
|
|
when stats are disabled. However, it may be desirable to change this
|
|
behaviour eventually to still keep current rows.
|
|
"""
|
|
|
|
self.hs.config.stats.stats_enabled = False
|
|
self.handler.stats_enabled = False
|
|
|
|
u1 = self.register_user("u1", "pass")
|
|
u1token = self.login("u1", "pass")
|
|
r1 = self.helper.create_room_as(u1, tok=u1token)
|
|
|
|
# test that these subjects, which were created during a time of disabled
|
|
# stats, do not have stats.
|
|
self.assertIsNone(self._get_current_stats("room", r1))
|
|
self.assertIsNone(self._get_current_stats("user", u1))
|
|
|
|
self.hs.config.stats.stats_enabled = True
|
|
self.handler.stats_enabled = True
|
|
|
|
self._perform_background_initial_update()
|
|
|
|
r1stats = self._get_current_stats("room", r1)
|
|
u1stats = self._get_current_stats("user", u1)
|
|
|
|
assert r1stats is not None
|
|
assert u1stats is not None
|
|
|
|
self.assertEqual(r1stats["joined_members"], 1)
|
|
self.assertEqual(
|
|
r1stats["current_state_events"], EXPT_NUM_STATE_EVTS_IN_FRESH_PUBLIC_ROOM
|
|
)
|
|
|
|
self.assertEqual(u1stats["joined_rooms"], 1)
|
|
|
|
def test_incomplete_stats(self) -> None:
|
|
"""
|
|
This tests that we track incomplete statistics.
|
|
|
|
We first test that incomplete stats are incrementally generated,
|
|
following the preparation of a background regen.
|
|
|
|
We then test that these incomplete rows are completed by the background
|
|
regen.
|
|
"""
|
|
|
|
u1 = self.register_user("u1", "pass")
|
|
u1token = self.login("u1", "pass")
|
|
u2 = self.register_user("u2", "pass")
|
|
u2token = self.login("u2", "pass")
|
|
u3 = self.register_user("u3", "pass")
|
|
r1 = self.helper.create_room_as(u1, tok=u1token, is_public=False)
|
|
|
|
# preparation stage of the initial background update
|
|
# Ugh, have to reset this flag
|
|
self.store.db_pool.updates._all_done = False
|
|
|
|
self.get_success(
|
|
self.store.db_pool.simple_delete(
|
|
"room_stats_current", {"1": 1}, "test_delete_stats"
|
|
)
|
|
)
|
|
self.get_success(
|
|
self.store.db_pool.simple_delete(
|
|
"user_stats_current", {"1": 1}, "test_delete_stats"
|
|
)
|
|
)
|
|
|
|
self.helper.invite(r1, u1, u2, tok=u1token)
|
|
self.helper.join(r1, u2, tok=u2token)
|
|
self.helper.invite(r1, u1, u3, tok=u1token)
|
|
self.helper.send(r1, "thou shalt yield", tok=u1token)
|
|
|
|
# now do the background updates
|
|
|
|
self.store.db_pool.updates._all_done = False
|
|
self.get_success(
|
|
self.store.db_pool.simple_insert(
|
|
"background_updates",
|
|
{
|
|
"update_name": "populate_stats_process_rooms",
|
|
"progress_json": "{}",
|
|
},
|
|
)
|
|
)
|
|
self.get_success(
|
|
self.store.db_pool.simple_insert(
|
|
"background_updates",
|
|
{
|
|
"update_name": "populate_stats_process_users",
|
|
"progress_json": "{}",
|
|
"depends_on": "populate_stats_process_rooms",
|
|
},
|
|
)
|
|
)
|
|
|
|
self.wait_for_background_updates()
|
|
|
|
r1stats_complete = self._get_current_stats("room", r1)
|
|
assert r1stats_complete is not None
|
|
u1stats_complete = self._get_current_stats("user", u1)
|
|
assert u1stats_complete is not None
|
|
u2stats_complete = self._get_current_stats("user", u2)
|
|
assert u2stats_complete is not None
|
|
|
|
# now we make our assertions
|
|
|
|
# check that _complete rows are complete and correct
|
|
self.assertEqual(r1stats_complete["joined_members"], 2)
|
|
self.assertEqual(r1stats_complete["invited_members"], 1)
|
|
|
|
self.assertEqual(
|
|
r1stats_complete["current_state_events"],
|
|
2 + EXPT_NUM_STATE_EVTS_IN_FRESH_PRIVATE_ROOM,
|
|
)
|
|
|
|
self.assertEqual(u1stats_complete["joined_rooms"], 1)
|
|
self.assertEqual(u2stats_complete["joined_rooms"], 1)
|