Fix session persistemce profiles list

resource_type is a static method and not a property

Change-Id: Ia1e90b2127a865b5997c8f6bec29fb410f417f65
This commit is contained in:
asarfaty 2021-03-17 10:21:31 +02:00 committed by Adit Sarfaty
parent 99860a6837
commit 6cd14b3149
2 changed files with 31 additions and 5 deletions

View File

@ -394,7 +394,7 @@ class TestPolicyLBCookiePersistenceProfile(
with mock.patch.object(
self.policy_api, "list",
return_value={'results': [
{'resource_type': self.resourceApi.entry_def.resource_type,
{'resource_type': self.resourceApi.entry_def.resource_type(),
'display_name': name}]}) as api_call:
obj = self.resourceApi.get_by_name(name, tenant=TEST_TENANT)
self.assertIsNotNone(obj)
@ -407,7 +407,7 @@ class TestPolicyLBCookiePersistenceProfile(
with mock.patch.object(
self.policy_api, "list",
return_value={'results': [
{'resource_type': self.resourceApi.entry_def.resource_type,
{'resource_type': self.resourceApi.entry_def.resource_type(),
'display_name': 'profile1'},
{'resource_type': 'wrong_type',
'display_name': 'profile2'}]}) as api_call:
@ -538,7 +538,7 @@ class TestPolicyLBSourceIpProfileApi(test_resources.NsxPolicyLibTestCase):
with mock.patch.object(
self.policy_api, "list",
return_value={'results': [
{'resource_type': self.resourceApi.entry_def.resource_type,
{'resource_type': self.resourceApi.entry_def.resource_type(),
'display_name': name}]}) as api_call:
obj = self.resourceApi.get_by_name(name, tenant=TEST_TENANT)
self.assertIsNotNone(obj)
@ -548,6 +548,32 @@ class TestPolicyLBSourceIpProfileApi(test_resources.NsxPolicyLibTestCase):
self.assert_called_with_def(api_call, expected_def)
def test_list(self):
profiles = [{'resource_type': 'LBSourceIpPersistenceProfile',
'id': 'default-source-ip-lb-persistence-profile',
'display_name': 'default-source-ip-profile'}]
with mock.patch.object(self.policy_api, "list",
return_value={'results': profiles}) as api_call:
result = self.resourceApi.list(tenant=TEST_TENANT)
expected_def = (
lb_defs.LBSourceIpPersistenceProfileDef(
tenant=TEST_TENANT))
self.assert_called_with_def(api_call, expected_def)
self.assertEqual(profiles, result)
def test_list_different_type(self):
profiles = [{'resource_type': 'LBSourceCookiePersistenceProfile',
'id': 'default-source-ip-lb-persistence-profile',
'display_name': 'default-source-ip-profile'}]
with mock.patch.object(self.policy_api, "list",
return_value={'results': profiles}) as api_call:
result = self.resourceApi.list(tenant=TEST_TENANT)
expected_def = (
lb_defs.LBSourceIpPersistenceProfileDef(
tenant=TEST_TENANT))
self.assert_called_with_def(api_call, expected_def)
self.assertEqual([], result)
def test_list_empty(self):
with mock.patch.object(self.policy_api, "list",
return_value={'results': []}) as api_call:
result = self.resourceApi.list(tenant=TEST_TENANT)

View File

@ -379,7 +379,7 @@ class NsxPolicyLoadBalancerCookiePersistenceProfileApi(
results = self._list(lb_cookie_persistence_profile_def)
# filter the results by resource type
return [res for res in results
if res.get('resource_type') == self.entry_def.resource_type]
if res.get('resource_type') == self.entry_def.resource_type()]
def update(self, persistence_profile_id,
name=IGNORE,
@ -443,7 +443,7 @@ class NsxPolicyLoadBalancerSourceIpPersistenceProfileApi(
results = self._list(lb_source_ip_persistence_profile_def)
# filter the results by resource type
return [res for res in results
if res.get('resource_type') == self.entry_def.resource_type]
if res.get('resource_type') == self.entry_def.resource_type()]
def update(self, persistence_profile_id,
name=IGNORE,