from twisted.internet import defer from synapse.storage.monthly_active_users import MonthlyActiveUsersStore import tests.unittest import tests.utils from tests.utils import setup_test_homeserver class MonthlyActiveUsersTestCase(tests.unittest.TestCase): def __init__(self, *args, **kwargs): super(MonthlyActiveUsersTestCase, self).__init__(*args, **kwargs) self.mau = None @defer.inlineCallbacks def setUp(self): hs = yield setup_test_homeserver() self.mau = MonthlyActiveUsersStore(hs) @defer.inlineCallbacks def test_can_insert_and_count_mau(self): count = yield self.mau.get_monthly_active_count() self.assertEqual(0, count) yield self.mau.upsert_monthly_active_user("@user:server") count = yield self.mau.get_monthly_active_count() self.assertEqual(1, count) @defer.inlineCallbacks def test_is_user_monthly_active(self): user_id1 = "@user1:server" user_id2 = "@user2:server" user_id3 = "@user3:server" result = yield self.mau.is_user_monthly_active(user_id1) self.assertFalse(result) yield self.mau.upsert_monthly_active_user(user_id1) yield self.mau.upsert_monthly_active_user(user_id2) result = yield self.mau.is_user_monthly_active(user_id1) self.assertTrue(result) result = yield self.mau.is_user_monthly_active(user_id3) self.assertFalse(result)