def setUp(self):
super(MemoryEngineTest, self).setUp()
self.application = Application(**{'options': Options})
self.engine = MemoryEngine(self.application)
self.engine.initialize()
self.engine.history_size = 2
self.engine.presence_timeout = 1
self.project_id = "project_id"
self.channel = "channel"
self.uid_1 = 'uid-1'
self.uid_2 = 'uid-2'
self.user_id = 'user_id'
self.user_id_extra = 'user_id_extra'
self.user_info = "{}"
self.message_1 = json.dumps('test message 1')
self.message_2 = json.dumps('test message 2')
self.message_3 = json.dumps('test message 3')
def setUp(self):
super(MemoryEngineTest, self).setUp()
self.application = Application(**{'options': Options})
self.engine = MemoryEngine(self.application)
self.engine.initialize()
self.engine.history_size = 2
self.engine.presence_timeout = 1
self.project_id = "project_id"
self.channel = "channel"
self.uid_1 = 'uid-1'
self.uid_2 = 'uid-2'
self.user_id = 'user_id'
self.user_id_extra = 'user_id_extra'
self.user_info = "{}"
self.message_1 = json.dumps('test message 1')
self.message_2 = json.dumps('test message 2')
self.message_3 = json.dumps('test message 3')
class MemoryEngineTest(AsyncTestCase):
def setUp(self):
super(MemoryEngineTest, self).setUp()
self.application = Application(**{'options': Options})
self.engine = MemoryEngine(self.application)
self.engine.initialize()
self.engine.history_size = 2
self.engine.presence_timeout = 1
self.project_id = "project_id"
self.channel = "channel"
self.uid_1 = 'uid-1'
self.uid_2 = 'uid-2'
self.user_id = 'user_id'
self.user_id_extra = 'user_id_extra'
self.user_info = "{}"
self.message_1 = json.dumps('test message 1')
self.message_2 = json.dumps('test message 2')
self.message_3 = json.dumps('test message 3')
@gen_test
def test_add_subscription(self):
yield self.engine.add_subscription(self.project_id, self.channel, FakeClient())
self.assertTrue(
self.engine.get_subscription_key(
self.project_id, self.channel
) in self.engine.subscriptions
)
@gen_test
def test_remove_subscription(self):
yield self.engine.remove_subscription(self.project_id, self.channel, FakeClient())
self.assertTrue(
self.engine.get_subscription_key(
self.project_id, self.channel
) not in self.engine.subscriptions
)
@gen_test
def test_presence(self):
result, error = yield self.engine.get_presence(
self.project_id, self.channel
)
self.assertEqual(result, {})
result, error = yield self.engine.add_presence(
self.project_id, self.channel, self.uid_1, self.user_info
)
self.assertEqual(result, True)
result, error = yield self.engine.get_presence(
self.project_id, self.channel
)
self.assertTrue(self.uid_1 in result)
result, error = yield self.engine.add_presence(
self.project_id, self.channel,
self.uid_1, self.user_info
)
self.assertEqual(result, True)
result, error = yield self.engine.get_presence(
self.project_id, self.channel
)
self.assertTrue(self.uid_1 in result)
self.assertEqual(len(result), 1)
result, error = yield self.engine.add_presence(
self.project_id, self.channel,
self.uid_2, self.user_info
)
self.assertEqual(result, True)
result, error = yield self.engine.get_presence(
self.project_id, self.channel
)
self.assertTrue(self.uid_1 in result)
self.assertTrue(self.uid_2 in result)
self.assertEqual(len(result), 2)
result, error = yield self.engine.remove_presence(
self.project_id, self.channel, self.uid_2
)
self.assertEqual(result, True)
result, error = yield self.engine.get_presence(
self.project_id, self.channel
)
self.assertTrue(self.uid_1 in result)
self.assertTrue(self.uid_2 not in result)
self.assertEqual(len(result), 1)
time.sleep(2)
result, error = yield self.engine.get_presence(
self.project_id, self.channel
)
self.assertEqual(result, {})
@gen_test
def test_history(self):
result, error = yield self.engine.add_history_message(
self.project_id, self.channel, self.message_1
)
self.assertEqual(error, None)
self.assertEqual(result, True)
result, error = yield self.engine.get_history(
self.project_id, self.channel
)
self.assertEqual(error, None)
self.assertEqual(len(result), 1)
result, error = yield self.engine.add_history_message(
self.project_id, self.channel, self.message_2
)
self.assertEqual(error, None)
self.assertEqual(result, True)
result, error = yield self.engine.get_history(
self.project_id, self.channel
)
self.assertEqual(error, None)
self.assertEqual(len(result), 2)
result, error = yield self.engine.add_history_message(
self.project_id, self.channel, self.message_3
)
self.assertEqual(error, None)
self.assertEqual(result, True)
result, error = yield self.engine.get_history(
self.project_id, self.channel
)
self.assertEqual(error, None)
self.assertEqual(len(result), 2)
@gen_test
def test_history_expire(self):
result, error = yield self.engine.add_history_message(
self.project_id, self.channel, self.message_1, history_expire=1
)
self.assertEqual(error, None)
self.assertEqual(result, True)
result, error = yield self.engine.get_history(
self.project_id, self.channel
)
self.assertEqual(error, None)
self.assertEqual(len(result), 1)
time.sleep(2)
result, error = yield self.engine.get_history(
self.project_id, self.channel
)
self.assertEqual(error, None)
self.assertEqual(len(result), 0)
class MemoryEngineTest(AsyncTestCase):
def setUp(self):
super(MemoryEngineTest, self).setUp()
self.application = Application(**{'options': Options})
self.engine = MemoryEngine(self.application)
self.engine.initialize()
self.engine.history_size = 2
self.engine.presence_timeout = 1
self.project_id = "project_id"
self.channel = "channel"
self.uid_1 = 'uid-1'
self.uid_2 = 'uid-2'
self.user_id = 'user_id'
self.user_id_extra = 'user_id_extra'
self.user_info = "{}"
self.message_1 = json.dumps('test message 1')
self.message_2 = json.dumps('test message 2')
self.message_3 = json.dumps('test message 3')
@gen_test
def test_add_subscription(self):
yield self.engine.add_subscription(self.project_id, self.channel, FakeClient())
self.assertTrue(
self.engine.get_subscription_key(
self.project_id, self.channel
) in self.engine.subscriptions
)
@gen_test
def test_remove_subscription(self):
yield self.engine.remove_subscription(self.project_id, self.channel, FakeClient())
self.assertTrue(
self.engine.get_subscription_key(
self.project_id, self.channel
) not in self.engine.subscriptions
)
@gen_test
def test_presence(self):
result, error = yield self.engine.get_presence(
self.project_id, self.channel
)
self.assertEqual(result, {})
result, error = yield self.engine.add_presence(
self.project_id, self.channel, self.uid_1, self.user_info
)
self.assertEqual(result, True)
result, error = yield self.engine.get_presence(
self.project_id, self.channel
)
self.assertTrue(self.uid_1 in result)
result, error = yield self.engine.add_presence(
self.project_id, self.channel,
self.uid_1, self.user_info
)
self.assertEqual(result, True)
result, error = yield self.engine.get_presence(
self.project_id, self.channel
)
self.assertTrue(self.uid_1 in result)
self.assertEqual(len(result), 1)
result, error = yield self.engine.add_presence(
self.project_id, self.channel,
self.uid_2, self.user_info
)
self.assertEqual(result, True)
result, error = yield self.engine.get_presence(
self.project_id, self.channel
)
self.assertTrue(self.uid_1 in result)
self.assertTrue(self.uid_2 in result)
self.assertEqual(len(result), 2)
result, error = yield self.engine.remove_presence(
self.project_id, self.channel, self.uid_2
)
self.assertEqual(result, True)
result, error = yield self.engine.get_presence(
self.project_id, self.channel
)
self.assertTrue(self.uid_1 in result)
self.assertTrue(self.uid_2 not in result)
self.assertEqual(len(result), 1)
time.sleep(2)
result, error = yield self.engine.get_presence(
self.project_id, self.channel
)
self.assertEqual(result, {})
@gen_test
def test_history(self):
result, error = yield self.engine.add_history_message(
self.project_id, self.channel, self.message_1, history_size=2, history_lifetime=1
)
self.assertEqual(error, None)
self.assertEqual(result, True)
result, error = yield self.engine.get_history(
self.project_id, self.channel
)
self.assertEqual(error, None)
self.assertEqual(len(result), 1)
result, error = yield self.engine.add_history_message(
self.project_id, self.channel, self.message_2, history_size=2, history_lifetime=1
)
self.assertEqual(error, None)
self.assertEqual(result, True)
result, error = yield self.engine.get_history(
self.project_id, self.channel
)
self.assertEqual(error, None)
self.assertEqual(len(result), 2)
result, error = yield self.engine.add_history_message(
self.project_id, self.channel, self.message_3, history_size=2, history_lifetime=1
)
self.assertEqual(error, None)
self.assertEqual(result, True)
result, error = yield self.engine.get_history(
self.project_id, self.channel
)
self.assertEqual(error, None)
self.assertEqual(len(result), 2)
@gen_test
def test_history_expire(self):
result, error = yield self.engine.add_history_message(
self.project_id, self.channel, self.message_1, history_size=2, history_lifetime=1
)
self.assertEqual(error, None)
self.assertEqual(result, True)
result, error = yield self.engine.get_history(
self.project_id, self.channel
)
self.assertEqual(error, None)
self.assertEqual(len(result), 1)
time.sleep(2)
result, error = yield self.engine.get_history(
self.project_id, self.channel
)
self.assertEqual(error, None)
self.assertEqual(len(result), 0)
