mirror of
https://mau.dev/maunium/synapse.git
synced 2024-10-01 01:36:05 -04:00
23740eaa3d
During the migration the automated script to update the copyright headers accidentally got rid of some of the existing copyright lines. Reinstate them.
161 lines
5.0 KiB
Python
161 lines
5.0 KiB
Python
#
|
|
# This file is licensed under the Affero General Public License (AGPL) version 3.
|
|
#
|
|
# Copyright 2015, 2016 OpenMarket Ltd
|
|
# 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]
|
|
#
|
|
#
|
|
|
|
import secrets
|
|
from typing import Generator, List, Tuple, cast
|
|
|
|
from twisted.test.proto_helpers import MemoryReactor
|
|
|
|
from synapse.server import HomeServer
|
|
from synapse.util import Clock
|
|
|
|
from tests import unittest
|
|
|
|
|
|
class UpdateUpsertManyTests(unittest.HomeserverTestCase):
|
|
def prepare(self, reactor: MemoryReactor, clock: Clock, hs: HomeServer) -> None:
|
|
self.storage = hs.get_datastores().main
|
|
|
|
self.table_name = "table_" + secrets.token_hex(6)
|
|
self.get_success(
|
|
self.storage.db_pool.runInteraction(
|
|
"create",
|
|
lambda x, *a: x.execute(*a),
|
|
"CREATE TABLE %s (id INTEGER, username TEXT, value TEXT)"
|
|
% (self.table_name,),
|
|
)
|
|
)
|
|
self.get_success(
|
|
self.storage.db_pool.runInteraction(
|
|
"index",
|
|
lambda x, *a: x.execute(*a),
|
|
"CREATE UNIQUE INDEX %sindex ON %s(id, username)"
|
|
% (self.table_name, self.table_name),
|
|
)
|
|
)
|
|
|
|
def _dump_table_to_tuple(self) -> Generator[Tuple[int, str, str], None, None]:
|
|
yield from cast(
|
|
List[Tuple[int, str, str]],
|
|
self.get_success(
|
|
self.storage.db_pool.simple_select_list(
|
|
self.table_name, None, ["id, username, value"]
|
|
)
|
|
),
|
|
)
|
|
|
|
def test_upsert_many(self) -> None:
|
|
"""
|
|
Upsert_many will perform the upsert operation across a batch of data.
|
|
"""
|
|
# Add some data to an empty table
|
|
key_names = ["id", "username"]
|
|
value_names = ["value"]
|
|
key_values = [[1, "user1"], [2, "user2"]]
|
|
value_values = [["hello"], ["there"]]
|
|
|
|
self.get_success(
|
|
self.storage.db_pool.runInteraction(
|
|
"test",
|
|
self.storage.db_pool.simple_upsert_many_txn,
|
|
self.table_name,
|
|
key_names,
|
|
key_values,
|
|
value_names,
|
|
value_values,
|
|
)
|
|
)
|
|
|
|
# Check results are what we expect
|
|
self.assertEqual(
|
|
set(self._dump_table_to_tuple()),
|
|
{(1, "user1", "hello"), (2, "user2", "there")},
|
|
)
|
|
|
|
# Update only user2
|
|
key_values = [[2, "user2"]]
|
|
value_values = [["bleb"]]
|
|
|
|
self.get_success(
|
|
self.storage.db_pool.runInteraction(
|
|
"test",
|
|
self.storage.db_pool.simple_upsert_many_txn,
|
|
self.table_name,
|
|
key_names,
|
|
key_values,
|
|
value_names,
|
|
value_values,
|
|
)
|
|
)
|
|
|
|
# Check results are what we expect
|
|
self.assertEqual(
|
|
set(self._dump_table_to_tuple()),
|
|
{(1, "user1", "hello"), (2, "user2", "bleb")},
|
|
)
|
|
|
|
def test_simple_update_many(self) -> None:
|
|
"""
|
|
simple_update_many performs many updates at once.
|
|
"""
|
|
# First add some data.
|
|
self.get_success(
|
|
self.storage.db_pool.simple_insert_many(
|
|
table=self.table_name,
|
|
keys=("id", "username", "value"),
|
|
values=[(1, "alice", "A"), (2, "bob", "B"), (3, "charlie", "C")],
|
|
desc="insert",
|
|
)
|
|
)
|
|
|
|
# Check the data made it to the table
|
|
self.assertEqual(
|
|
set(self._dump_table_to_tuple()),
|
|
{(1, "alice", "A"), (2, "bob", "B"), (3, "charlie", "C")},
|
|
)
|
|
|
|
# Now use simple_update_many
|
|
self.get_success(
|
|
self.storage.db_pool.simple_update_many(
|
|
table=self.table_name,
|
|
key_names=("username",),
|
|
key_values=(
|
|
("alice",),
|
|
("bob",),
|
|
("stranger",),
|
|
),
|
|
value_names=("value",),
|
|
value_values=(
|
|
("aaa!",),
|
|
("bbb!",),
|
|
("???",),
|
|
),
|
|
desc="update_many1",
|
|
)
|
|
)
|
|
|
|
# Check the table is how we expect:
|
|
# charlie has been left alone
|
|
self.assertEqual(
|
|
set(self._dump_table_to_tuple()),
|
|
{(1, "alice", "aaa!"), (2, "bob", "bbb!"), (3, "charlie", "C")},
|
|
)
|