Skip to content

Instantly share code, notes, and snippets.

@banveerad
Last active August 29, 2015 14:26
Show Gist options
  • Save banveerad/23ead3df44dba22f9b88 to your computer and use it in GitHub Desktop.
Save banveerad/23ead3df44dba22f9b88 to your computer and use it in GitHub Desktop.
def test_get_all_pools(self):
with self.subnet() as subnet:
with self.loadbalancer(subnet=subnet, name='lb1') as lb:
lb_id = lb['loadbalancer']['id']
with self.listener(loadbalancer_id=lb_id,
protocol_port=80) as listener1:
with self.listener(loadbalancer_id=lb_id,
protocol_port=443) as listener2:
listener1_id = listener1['listener']['id']
listener2_id = listener2['listener']['id']
with self.pool(listener_id=listener1_id) as pool1:
with self.pool(listener_id=listener2_id) as pool2:
resp, body = self._get_loadbalancer_api(lb_id)
self.assertEqual(len(body['loadbalancer']['all_pools']), 2)
expected_res = [pool1['pool']['id'], pool2['pool']['id']]
self.assertEqual(expected_res, [n['id'] for n in body['loadbalancer']['all_pools']])
def test_get_all_pools_with_l7_policy(self):
with self.loadbalancer(name='lb1') as lb1:
lb1_id = lb1['loadbalancer']['id']
with self.listener(loadbalancer_id=lb1_id) as listener1:
listener1_id = listener1['listener']['id']
with self.pool(listener_id=listener1_id) as pool:
pool_id = pool['pool']['id']
with self.loadbalancer(name='lb2') as lb2:
lb2_id = lb2['loadbalancer']['id']
with self.listener(loadbalancer_id=lb2_id) as listener2:
listener2_id = listener2['listener']['id']
with self.l7policy(listener2_id, name="0", redirect_pool_id=pool_id) as p:
resp, body = self._get_loadbalancer_api(lb2_id)
self.assertEqual(len(body['loadbalancer']['all_pools']), 1)
self.assertEqual(body['loadbalancer']['all_pools'][0]['id'], pool_id)
def test_get_all_pools_with_l7_policy_noredirect_pool(self):
with self.loadbalancer(name='lb1') as lb1:
lb1_id = lb1['loadbalancer']['id']
with self.listener(loadbalancer_id=lb1_id) as listener1:
listener1_id = listener1['listener']['id']
with self.pool(listener_id=listener1_id) as pool:
pool_id = pool['pool']['id']
with self.loadbalancer(name='lb2') as lb2:
lb2_id = lb2['loadbalancer']['id']
with self.listener(loadbalancer_id=lb2_id) as listener2:
listener2_id = listener2['listener']['id']
with self.l7policy(listener2_id, name="0") as p:
resp, body = self._get_loadbalancer_api(lb2_id)
self.assertEqual(len(body['loadbalancer']['all_pools']), 0)
def test_get_all_pools_with_l7_policy_redirect_n_listener_pool(self):
with self.loadbalancer(name='lb1') as lb1:
lb1_id = lb1['loadbalancer']['id']
with self.listener(loadbalancer_id=lb1_id) as listener1:
listener1_id = listener1['listener']['id']
with self.pool(listener_id=listener1_id) as pool1:
pool_id1 = pool1['pool']['id']
with self.loadbalancer(name='lb2') as lb2:
lb2_id = lb2['loadbalancer']['id']
with self.listener(loadbalancer_id=lb2_id) as listener2:
listener2_id = listener2['listener']['id']
with self.pool(listener_id=listener2_id) as pool2:
with self.l7policy(listener2_id, name="0", redirect_pool_id=pool_id1) as p:
resp, body = self._get_loadbalancer_api(lb2_id)
self.assertEqual(len(body['loadbalancer']['all_pools']), 2)
expected_res = [pool2['pool']['id'], pool1['pool']['id']]
self.assertEqual(expected_res, [n['id'] for n in body['loadbalancer']['all_pools']])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment