def test_createing_iot_device_object(self):
device_id = "KEOIDUV"
with session_scope() as session:
obj = IoTDevice(device_id=device_id)
session.add(obj)
with session_scope() as session:
ret = session.query(IoTDevice).one()
self.assertEqual(ret.device_id, device_id)
def test_data_io_handler_writing_list(self):
data = [{"device_id": f"{i}"} for i in range(10)]
IoTDevice.sql.write(data)
with session_scope() as session:
ret = session.query(IoTDeviceSQL).count()
self.assertEqual(ret, 10)
def test_dataclase_save_to_sqlalchemy_db(self):
device_id = "xELP0d"
x = IoTDevice(device_id=device_id)
x.save()
with session_scope() as session:
ret = session.query(IoTDeviceSQL).one()
self.assertEqual(ret.device_id, device_id)
def test_data_io_handler_writing_list_with_error(self):
data = [{"device_id": f"{i}"} for i in range(10)]
data[0]["device_id"] = None
with self.assertRaises(DataIOWriteError):
IoTDevice.sql.write(data)
with session_scope() as session:
ret = session.query(IoTDeviceSQL).count()
self.assertEqual(ret, 9)