def test_assert_handlers_called(self):
handler_vars = {
'create_handler_called': False,
'update_handler_called': False,
'delete_handler_called': False
}
def create_handler(sensor_db):
handler_vars['create_handler_called'] = True
def update_handler(sensor_db):
handler_vars['update_handler_called'] = True
def delete_handler(sensor_db):
handler_vars['delete_handler_called'] = True
sensor_watcher = SensorWatcher(create_handler, update_handler, delete_handler)
message = Message(None, delivery_info={'routing_key': 'create'})
sensor_watcher.process_task(MOCK_SENSOR_DB, message)
self.assertTrue(handler_vars['create_handler_called'], 'create handler should be called.')
message = Message(None, delivery_info={'routing_key': 'update'})
sensor_watcher.process_task(MOCK_SENSOR_DB, message)
self.assertTrue(handler_vars['update_handler_called'], 'update handler should be called.')
message = Message(None, delivery_info={'routing_key': 'delete'})
sensor_watcher.process_task(MOCK_SENSOR_DB, message)
self.assertTrue(handler_vars['delete_handler_called'], 'delete handler should be called.')
def test_assert_handlers_called(self):
handler_vars = {
"create_handler_called": False,
"update_handler_called": False,
"delete_handler_called": False,
}
def create_handler(sensor_db):
handler_vars["create_handler_called"] = True
def update_handler(sensor_db):
handler_vars["update_handler_called"] = True
def delete_handler(sensor_db):
handler_vars["delete_handler_called"] = True
sensor_watcher = SensorWatcher(create_handler, update_handler,
delete_handler)
message = Message(None, delivery_info={"routing_key": "create"})
sensor_watcher.process_task(MOCK_SENSOR_DB, message)
self.assertTrue(handler_vars["create_handler_called"],
"create handler should be called.")
message = Message(None, delivery_info={"routing_key": "update"})
sensor_watcher.process_task(MOCK_SENSOR_DB, message)
self.assertTrue(handler_vars["update_handler_called"],
"update handler should be called.")
message = Message(None, delivery_info={"routing_key": "delete"})
sensor_watcher.process_task(MOCK_SENSOR_DB, message)
self.assertTrue(handler_vars["delete_handler_called"],
"delete handler should be called.")
