def test_device_get_all(self):
device_ref1 = db_api.device_create(self.conf, device_fake1)
device_ref2 = db_api.device_create(self.conf, device_fake2)
devices = db_api.device_get_all(self.conf)
self.assertEqual([dict(dev.iteritems()) for dev in devices],
[dict(dev.iteritems()) for dev in [device_ref1,
device_ref2]])
def test_device_get_several(self):
device_ref1 = db_api.device_create(self.conf, device_fake1)
device_ref2 = db_api.device_create(self.conf, device_fake2)
device_ref3 = db_api.device_get(self.conf, device_ref1["id"])
device_ref4 = db_api.device_get(self.conf, device_ref2["id"])
self.assertEqual(dict(device_ref3.iteritems()), dict(device_ref1.iteritems()))
self.assertEqual(dict(device_ref4.iteritems()), dict(device_ref2.iteritems()))
def test_device_destroy(self):
device_ref = db_api.device_create(self.conf, device_fake1)
db_api.device_destroy(self.conf, device_ref['id'])
with self.assertRaises(exception.DeviceNotFound) as cm:
db_api.device_get(self.conf, device_ref['id'])
err = cm.exception
self.assertEqual(err.kwargs, {'device_id': device_ref['id']})
def test_device_create(self):
device_ref = db_api.device_create(self.conf, device_fake1)
device = dict(device_ref.iteritems())
expected = device_fake1.copy()
self.assertIsNotNone(device['id'])
expected['id'] = device['id']
self.assertEqual(device, expected)
def test_device_update(self):
device_ref = db_api.device_create(self.conf, device_fake1)
self.assertIsNotNone(device_ref["id"])
update = {"password": "******", "extra": {"supports_vlan": True, "vip_vlan": 100, "requires_vip_ip": True}}
device_ref = db_api.device_update(self.conf, device_ref["id"], update)
device = dict(device_ref.iteritems())
expected = device_fake1.copy()
expected.update(update)
self.assertIsNotNone(device["id"])
expected["id"] = device["id"]
self.assertEqual(device, expected)
def test_device_update(self):
device_ref = db_api.device_create(self.conf, device_fake1)
self.assertIsNotNone(device_ref['id'])
update = {'password': '******',
'extra': {'supports_vlan': True,
'vip_vlan': 100,
'requires_vip_ip': True}}
device_ref = db_api.device_update(self.conf, device_ref['id'], update)
device = dict(device_ref.iteritems())
expected = device_fake1.copy()
expected.update(update)
self.assertIsNotNone(device['id'])
expected['id'] = device['id']
self.assertEqual(device, expected)
def device_create(conf, **params):
device_dict = db_api.device_pack_extra(params)
device = db_api.device_create(conf, device_dict)
return device
def test_device_get(self):
device_ref1 = db_api.device_create(self.conf, device_fake1)
device_ref2 = db_api.device_get(self.conf, device_ref1['id'])
self.assertEqual(dict(device_ref1.iteritems()),
dict(device_ref2.iteritems()))
def test_device_get_all(self):
device_ref1 = db_api.device_create(self.conf, device_fake1)
device_ref2 = db_api.device_create(self.conf, device_fake2)
devices = db_api.device_get_all(self.conf)
self.assertEqual(len(devices), 2)
def device_create(conf, **params):
device_dict = db_api.device_pack_extra(params)
device = db_api.device_create(conf, device_dict)
scheduller.Scheduller.Instance(conf).addDevice(device)
return device['id']
