mirror of
https://git.anonymousland.org/anonymousland/synapse-product.git
synced 2024-09-09 15:22:01 +00:00
fix known broken test
This commit is contained in:
parent
0aba3d361a
commit
4e6e00152c
@ -15,7 +15,7 @@
|
||||
|
||||
from twisted.internet import defer
|
||||
|
||||
import tests.unittest
|
||||
|
||||
import tests.utils
|
||||
|
||||
|
||||
@ -33,27 +33,25 @@ class InitTestCase(tests.unittest.TestCase):
|
||||
self.store = hs.get_datastore()
|
||||
self.clock = hs.get_clock()
|
||||
|
||||
@defer.inlineCallbacks
|
||||
def test_count_monthly_users(self):
|
||||
count = self.store.count_monthly_users()
|
||||
count = yield self.store.count_monthly_users()
|
||||
self.assertEqual(0, count)
|
||||
|
||||
self._insert_user_ips("@user:server1")
|
||||
self._insert_user_ips("@user:server2")
|
||||
yield self._insert_user_ips("@user:server1")
|
||||
yield self._insert_user_ips("@user:server2")
|
||||
|
||||
count = self.store.count_monthly_users()
|
||||
count = yield self.store.count_monthly_users()
|
||||
self.assertEqual(2, count)
|
||||
|
||||
@defer.inlineCallbacks
|
||||
def _insert_user_ips(self, user):
|
||||
"""
|
||||
Helper function to populate user_ips without using batch insertion infra
|
||||
args:
|
||||
user (str): specify username i.e. @user:server.com
|
||||
"""
|
||||
try:
|
||||
txn = self.store.db_conn.cursor()
|
||||
self.store.database_engine.lock_table(txn, "user_ips")
|
||||
self.store._simple_upsert_txn(
|
||||
txn,
|
||||
yield self.store._simple_upsert(
|
||||
table="user_ips",
|
||||
keyvalues={
|
||||
"user_id": user,
|
||||
@ -64,8 +62,5 @@ class InitTestCase(tests.unittest.TestCase):
|
||||
},
|
||||
values={
|
||||
"last_seen": self.clock.time_msec(),
|
||||
},
|
||||
lock=False,
|
||||
}
|
||||
)
|
||||
finally:
|
||||
txn.close()
|
||||
|
Loading…
Reference in New Issue
Block a user