def test_loadbalancer_get_all_by_device_id(self):
lb_fake1 = get_fake_lb('1', 'tenant1')
lb_ref1 = db_api.loadbalancer_create(self.conf, lb_fake1)
lbs1 = db_api.loadbalancer_get_all_by_device_id(self.conf, '1')
with self.assertRaises(exception.LoadBalancerNotFound):
db_api.loadbalancer_get_all_by_device_id(self.conf, '2')
self.assertEqual([dict(lb_ref1.iteritems())],
[dict(lb.iteritems()) for lb in lbs1])
def device_delete(conf, device_id):
try:
lb_refs = db_api.loadbalancer_get_all_by_device_id(conf, device_id)
except exc.LoadBalancerNotFound:
db_api.device_destroy(conf, device_id)
drivers.delete_device_driver(conf, device_id)
return
lbs = []
for lb_ref in lb_refs:
lb = db_api.unpack_extra(lb_ref)
lbs.append(lb['id'])
raise exc.DeviceConflict('Device %s is in use now by loadbalancers %s' %
(device_id, ', '.join(lbs)))
