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
