x/vmware-nsxlib
Code Issues Proposed changes

Add Load Balancer Application Rule

Browse Source 
Added a class Rule in load_balancer to support load balancer
layer7 application rule.

Change-Id: I8137e4c82536f04c5a471c90f1a7c591b3d46123
This commit is contained in:
Tong Liu 2017-07-15 15:19:57 -07:00
parent eff49e1899
commit ad40ef856e
3 changed files with 72 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
vmware_nsxlib/tests/unit/v3/test_constants.py
View File

@ -236,6 +236,28 @@ FAKE_PERSISTENCE_PROFILE = {
"_revision": 0
}
FAKE_RULE_UUID = uuidutils.generate_uuid()
FAKE_RULE = {
"resource_type": "LbRule",
"description": "LbRule to route login requests to dedicated pool",
"id": FAKE_RULE_UUID,
"display_name": "LoginRouteRule",
"phase": "HTTP_FORWARDING",
"match_strategy": "ALL",
"match_conditions": [
{
"type": "LbHttpRequestUriCondition",
"uri": "/login"
}
],
"actions": [
{
"type": "LbSelectPoolAction",
"pool_id": "54411c58-046c-4236-8ff1-e1e1aad3e873"
}
]
}
FAKE_CLIENT_SSL_PROFILE_UUID = uuidutils.generate_uuid()
FAKE_CLIENT_SSL_PROFILE = {
"display_name": "clientSslProfile1",

34
vmware_nsxlib/tests/unit/v3/test_load_balancer.py
View File

@ -128,6 +128,40 @@ class TestPersistenceProfile(nsxlib_testcase.NsxClientTestCase):
'loadbalancer/persistence-profiles/%s' % fake_profile['id'])
class TestRule(nsxlib_testcase.NsxClientTestCase):
def test_create_rule(self):
fake_rule = consts.FAKE_RULE.copy()
body = {
'display_name': fake_rule['display_name'],
'description': fake_rule['description'],
'resource_type': fake_rule['resource_type'],
'phase': fake_rule['phase'],
'match_strategy': fake_rule['match_strategy'],
'tags': tags
}
with mock.patch.object(self.nsxlib.client, 'create') as create:
self.nsxlib.load_balancer.rule.create(**body)
create.assert_called_with('loadbalancer/rules', body)
def test_list_rules(self):
with mock.patch.object(self.nsxlib.client, 'list') as list_call:
self.nsxlib.load_balancer.rule.list()
list_call.assert_called_with(resource='loadbalancer/rules')
def test_get_rule(self):
with mock.patch.object(self.nsxlib.client, 'get') as get:
fake_rule = consts.FAKE_RULE.copy()
self.nsxlib.load_balancer.rule.get(fake_rule['id'])
get.assert_called_with('loadbalancer/rules/%s' % fake_rule['id'])
def test_delete_rule(self):
with mock.patch.object(self.nsxlib.client, 'delete') as delete:
fake_rule = consts.FAKE_RULE.copy()
self.nsxlib.load_balancer.rule.delete(fake_rule['id'])
delete.assert_called_with(
'loadbalancer/rules/%s' % fake_rule['id'])
class TestClientSslProfile(nsxlib_testcase.NsxClientTestCase):
def test_create_client_ssl_profiles(self):

16
vmware_nsxlib/v3/load_balancer.py
View File

@ -150,6 +150,10 @@ class PersistenceProfile(LoadBalancerBase):
arg_name='resource_type')
class Rule(LoadBalancerBase):
resource = 'loadbalancer/rules'
class ClientSslProfile(LoadBalancerBase):
resource = 'loadbalancer/client-ssl-profiles'
@ -272,6 +276,17 @@ class VirtualServer(LoadBalancerBase):
body['ip_address'] = vip
return self.client.update(object_url, body)
def add_rule(self, vs_id, rule_id):
object_url = self.resource + '/' + vs_id
body = self.client.get(object_url)
if 'rule_ids' in body:
rule_list = body['rule_ids']
rule_list.append(rule_id)
else:
rule_list = [rule_id]
body['rule_ids'] = rule_list
return self.client.update(object_url, body)
class Service(LoadBalancerBase):
resource = 'loadbalancer/services'
@ -329,3 +344,4 @@ class LoadBalancer(object):
self.persistence_profile = PersistenceProfile(client, nsxlib_config)
self.client_ssl_profile = ClientSslProfile(client, nsxlib_config)
self.server_ssh_profile = ServerSslProfile(client, nsxlib_config)
self.rule = Rule(client, nsxlib_config)