def test_save_digital_twin(self, mock_mongo):
mock_mongo.return_value = None
twin = [{'device_name': 'test_device', 'active': False,
'location': 'on-desk', 'technology': 'WI-FI', 'battery_level': 'USB-power'}]
db_handler = DbHandler(queue=self.test_queue, thread_event=self.test_event)
db_handler.mongo = MockMongoDeviceNameExist
db_handler._save_digital_twin(twin=twin)
def test_add_document_row(self, mock_mongo):
mock_mongo.return_value = None
db_handler = DbHandler(queue=self.test_queue, thread_event=self.test_event)
db_handler.mongo = MockMongo
db_handler.add_document_row(msg=ObserverMessage(event="pass", data=""))
def test_get_data(self, mock_mongo):
mock_mongo.return_value = None
db_handler = DbHandler(queue=self.test_queue, thread_event=self.test_event)
db_handler.mongo = MockMongo
result = db_handler.get_document_data(document="abc")
self.assertEqual(result, ['a', 'b'])
def test_store_state_data_new(self, mock_mongo):
mock_mongo.return_value = None
db_handler = DbHandler(queue=self.test_queue, thread_event=self.test_event)
db_handler.mongo = MockMongoNoDeviceNameExist
db_handler.store_state_data(msg=ObserverMessage(event="pass", data={}))
def test_action_skip(self, mock_mongo):
mock_mongo.return_value = None
db_handler = DbHandler(queue=self.test_queue, thread_event=self.test_event)
db_handler.mongo = MockMongo
db_handler.action_skip()