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)