def test_update_device(self): input = Device(name="TestDevice",description = "A test description") yield self._persistenceLayer.save_device(input) input.name = "TestDeviceModified" yield self._persistenceLayer.save_device(input) exp = input obs = yield self._persistenceLayer.load_device(id = input.cid) self.assertEquals(obs,exp)
def test_save_and_load_devices(self): input =[] input.append(Device(name="TestDeviceOne",description = "A test description")) input.append(Device(name="TestDeviceTwo",description = "Another test description",status="active")) yield self._persistenceLayer.save_devices(input) exp = input obs = yield self._persistenceLayer.load_devices() self.assertEquals(obs,exp)
def load_devices(self, environmentId = None ,*args,**kwargs): """Save the device object .""" lDevices = [] if environmentId is not None: rows = yield self.select(environmentId = str(environmentId), order = "id") else: rows = yield self.select(order = "id") for row in rows: id,name,description,status = row device = Device(name = name,description=description,status=status) device.cid = uuid.UUID(id) lDevices.append(device) defer.returnValue(lDevices)
def load_device(self, id=None, environmentId=None, *args, **kwargs): """Retrieve data from device object.""" if id is not None: rows = yield self.select(id=str(id)) elif environmentId is not None: rows = yield self.select(environmentId=str(environmentId)) result = None if len(rows) > 0: id, name, description, status = rows[0] result = Device(name=name, description=description, status=status) result.cid = uuid.UUID(id) else: raise DeviceNotFound() defer.returnValue(result)
def load_devices(self, environmentId=None, *args, **kwargs): """Save the device object .""" lDevices = [] if environmentId is not None: rows = yield self.select(environmentId=str(environmentId), order="id") else: rows = yield self.select(order="id") for row in rows: id, name, description, status = row device = Device(name=name, description=description, status=status) device.cid = uuid.UUID(id) lDevices.append(device) defer.returnValue(lDevices)
def load_device(self,id = None, environmentId = None ,*args,**kwargs): """Retrieve data from device object.""" if id is not None: rows = yield self.select(id = str(id)) elif environmentId is not None: rows = yield self.select(environmentId = str(environmentId)) result=None if len(rows)>0: id,name,description,status = rows[0] result = Device(name = name,description=description,status=status) result.cid = uuid.UUID(id) else: raise DeviceNotFound() defer.returnValue(result)
def test_delete_device(self): input = Device(name="TestDevice",description = "A test description") yield self._persistenceLayer.save_device(input) exp = input obs = yield self._persistenceLayer.load_device(id = input.cid) self.assertEquals(obs,exp) yield self._persistenceLayer.delete_device(input) obs = self._persistenceLayer.load_device(id = input.cid) self.assertFailure(obs ,DeviceNotFound )
def test_save_and_load_devices_of_environment(self): parentEnvironement = Environment(name="TestEnvironment",description = "A test description") device1 = Device(name="TestDeviceOne",description = "A test description") device1._parent = parentEnvironement device2 = Device(name="TestDeviceTwo",description = "Another test description",status="active") device2._parent = parentEnvironement input =[device1,device2] parentEnvironement2 = Environment(name="TestEnvironment2",description = "A test description") device3 = Device(name="TestDeviceThree",description = "A test description") device3._parent = parentEnvironement2 yield self._persistenceLayer.save_devices(input) yield self._persistenceLayer.save_device(device3) exp = input obs = yield self._persistenceLayer.load_devices(parentEnvironement.cid) self.assertEquals(obs,exp) exp = [device3] obs = yield self._persistenceLayer.load_devices(parentEnvironement2.cid) self.assertEquals(obs,exp)
def test_delete_device_exception(self): input = Device(name="TestDevice",description = "A test description") obs = self._persistenceLayer.delete_device(input) self.assertFailure(obs ,DeviceNotFound )