def test_loadbalancer_get_all_by_vm_id(self): lb_fake1 = get_fake_lb('1', 'tenant1') lb_fake2 = get_fake_lb('2', 'tenant2') lb_ref1 = db_api.loadbalancer_create(self.conf, lb_fake1) lb_ref2 = db_api.loadbalancer_create(self.conf, lb_fake2) sf_fake1 = get_fake_sf(lb_ref1['id']) sf_fake2 = get_fake_sf(lb_ref2['id']) sf_ref1 = db_api.serverfarm_create(self.conf, sf_fake1) sf_ref2 = db_api.serverfarm_create(self.conf, sf_fake2) node_fake1 = get_fake_server(sf_ref1['id'], 1, '10.0.0.1') node_fake2 = get_fake_server(sf_ref1['id'], 20, '10.0.0.2') node_fake3 = get_fake_server(sf_ref2['id'], 1, '10.0.0.3') node_fake4 = get_fake_server(sf_ref2['id'], 30, '10.0.0.4') node_ref1 = db_api.server_create(self.conf, node_fake1) node_ref2 = db_api.server_create(self.conf, node_fake2) node_ref3 = db_api.server_create(self.conf, node_fake3) node_ref4 = db_api.server_create(self.conf, node_fake4) lbs1 = db_api.loadbalancer_get_all_by_vm_id(self.conf, 'tenant1', 1) lbs2 = db_api.loadbalancer_get_all_by_vm_id(self.conf, 'tenant2', 30) lbs3 = db_api.loadbalancer_get_all_by_vm_id(self.conf, 'tenant2', 20) self.assertEqual([dict(lb_ref1.iteritems())], [dict(lb.iteritems()) for lb in lbs1]) self.assertEqual([dict(lb_ref2.iteritems())], [dict(lb.iteritems()) for lb in lbs2]) self.assertFalse(lbs3)
def test_loadbalancer_get_all_by_vm_id(self): lb_fake1 = get_fake_lb('1', 'tenant1') lb_fake2 = get_fake_lb('2', 'tenant2') lb_ref1 = db_api.loadbalancer_create(self.conf, lb_fake1) lb_ref2 = db_api.loadbalancer_create(self.conf, lb_fake2) sf_fake1 = get_fake_sf(lb_ref1['id']) sf_fake2 = get_fake_sf(lb_ref2['id']) sf_ref1 = db_api.serverfarm_create(self.conf, sf_fake1) sf_ref2 = db_api.serverfarm_create(self.conf, sf_fake2) node_fake1 = get_fake_server(sf_ref1['id'], 1, '10.0.0.1') node_fake2 = get_fake_server(sf_ref1['id'], 20, '10.0.0.2') node_fake3 = get_fake_server(sf_ref2['id'], 1, '10.0.0.3') node_fake4 = get_fake_server(sf_ref2['id'], 30, '10.0.0.4') node_ref1 = db_api.server_create(self.conf, node_fake1) node_ref2 = db_api.server_create(self.conf, node_fake2) node_ref3 = db_api.server_create(self.conf, node_fake3) node_ref4 = db_api.server_create(self.conf, node_fake4) lbs1 = db_api.loadbalancer_get_all_by_vm_id(self.conf, 1, 'tenant1') lbs2 = db_api.loadbalancer_get_all_by_vm_id(self.conf, 30, 'tenant2') lbs3 = db_api.loadbalancer_get_all_by_vm_id(self.conf, 20, 'tenant2') self.assertEqual(len(lbs1), 1) self.assertEqual(lbs1[0]['id'], lb_ref1['id']) self.assertEqual(len(lbs2), 1) self.assertEqual(lbs2[0]['id'], lb_ref2['id']) self.assertEqual(len(lbs3), 0)
def test_loadbalancer_get_all_by_vm_id(self): lb_fake1 = get_fake_lb("1", "tenant1") lb_fake2 = get_fake_lb("2", "tenant2") lb_ref1 = db_api.loadbalancer_create(self.conf, lb_fake1) lb_ref2 = db_api.loadbalancer_create(self.conf, lb_fake2) sf_fake1 = get_fake_sf(lb_ref1["id"]) sf_fake2 = get_fake_sf(lb_ref2["id"]) sf_ref1 = db_api.serverfarm_create(self.conf, sf_fake1) sf_ref2 = db_api.serverfarm_create(self.conf, sf_fake2) node_fake1 = get_fake_server(sf_ref1["id"], 1, "10.0.0.1") node_fake2 = get_fake_server(sf_ref1["id"], 20, "10.0.0.2") node_fake3 = get_fake_server(sf_ref2["id"], 1, "10.0.0.3") node_fake4 = get_fake_server(sf_ref2["id"], 30, "10.0.0.4") node_ref1 = db_api.server_create(self.conf, node_fake1) node_ref2 = db_api.server_create(self.conf, node_fake2) node_ref3 = db_api.server_create(self.conf, node_fake3) node_ref4 = db_api.server_create(self.conf, node_fake4) lbs1 = db_api.loadbalancer_get_all_by_vm_id(self.conf, 1, "tenant1") lbs2 = db_api.loadbalancer_get_all_by_vm_id(self.conf, 30, "tenant2") lbs3 = db_api.loadbalancer_get_all_by_vm_id(self.conf, 20, "tenant2") self.assertEqual([dict(lb_ref1.iteritems())], [dict(lb.iteritems()) for lb in lbs1]) self.assertEqual([dict(lb_ref2.iteritems())], [dict(lb.iteritems()) for lb in lbs2]) self.assertFalse(lbs3)
def lb_find_for_vm(conf, vm_id, tenant_id=''): lbs = db_api.loadbalancer_get_all_by_vm_id(conf, vm_id, tenant_id) lbs = [db_api.unpack_extra(lb) for lb in lbs] return lbs