def gets_a_notification_only_once(self):
handler = StubHandler()
manager = NotificationManager(handler)
manager.set('foo', 'bar')
self.assertEqual(manager.get('foo'), 'bar')
self.assertIsNone(manager.get('foo'))
def gets_a_notification_only_once(self):
handler = StubHandler()
manager = NotificationManager(handler)
manager.set('foo', 'bar')
self.assertEqual(manager.get('foo'), 'bar')
self.assertIsNone(manager.get('foo'))
def uses_custom_notifications_database_if_provided(self):
handler = StubHandler()
handler.settings = {
'pycket': {
'engine': 'redis',
'storage': {
'db_sessions': 10,
'db_notifications': 11,
}
}
}
manager = NotificationManager(handler)
manager.set('foo', 'bar')
self.assertEqual(manager.driver.client.connection_pool._available_connections[0].db, 11)
def uses_custom_notifications_database_if_provided(self):
handler = StubHandler()
handler.settings = {
'pycket': {
'engine': 'redis',
'storage': {
'db_sessions': 10,
'db_notifications': 11,
}
}
}
manager = NotificationManager(handler)
manager.set('foo', 'bar')
self.assertEqual(
manager.driver.client.connection_pool._available_connections[0].db,
11)
def removes_notification_from_database_after_retrieval(self):
handler = StubHandler()
manager = NotificationManager(handler)
manager.set('foo', 'bar')
raw_notifications = self.client.get(handler.session_id)
notifications = pickle.loads(raw_notifications)
self.assertEqual(list(notifications.keys()), ['foo'])
manager.get('foo')
raw_notifications = self.client.get(handler.session_id)
notifications = pickle.loads(raw_notifications)
self.assertEqual(list(notifications.keys()), [])
def removes_notification_from_database_after_retrieval(self):
handler = StubHandler()
manager = NotificationManager(handler)
manager.set('foo', 'bar')
raw_notifications = self.client.get(handler.session_id)
notifications = pickle.loads(raw_notifications)
self.assertEqual(list(notifications.keys()), ['foo'])
manager.get('foo')
raw_notifications = self.client.get(handler.session_id)
notifications = pickle.loads(raw_notifications)
self.assertEqual(list(notifications.keys()), [])
