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 )
