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)))