From 0710496090a84a3fc1cbf6d66d1649a96b2d8bbc Mon Sep 17 00:00:00 2001 From: Adit Sarfaty Date: Sun, 28 Apr 2019 09:46:27 +0300 Subject: [PATCH] Add category to policy update entries Or else it will fail when the category is different than the current one. Change-Id: Id3e36b7bac37bfdc519eb9eab0d6e9cd0fb0e245 --- vmware_nsxlib/tests/unit/v3/policy/test_resources.py | 4 +++- vmware_nsxlib/v3/policy/core_resources.py | 4 +++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/vmware_nsxlib/tests/unit/v3/policy/test_resources.py b/vmware_nsxlib/tests/unit/v3/policy/test_resources.py index 41796b6e..3c71eb16 100644 --- a/vmware_nsxlib/tests/unit/v3/policy/test_resources.py +++ b/vmware_nsxlib/tests/unit/v3/policy/test_resources.py @@ -1873,7 +1873,9 @@ class TestPolicyCommunicationMap(NsxPolicyLibTestCase): self.resourceApi.update_entries( domain_id, map_id, entries, tenant=TEST_TENANT) update_call.assert_called_once_with( - domain_id, map_id, entries, tenant=TEST_TENANT) + domain_id, map_id, entries, + category=constants.CATEGORY_APPLICATION, + tenant=TEST_TENANT) def test_update_with_entries(self): domain_id = '111' diff --git a/vmware_nsxlib/v3/policy/core_resources.py b/vmware_nsxlib/v3/policy/core_resources.py index 4b2333c5..c9b688be 100644 --- a/vmware_nsxlib/v3/policy/core_resources.py +++ b/vmware_nsxlib/v3/policy/core_resources.py @@ -2905,8 +2905,10 @@ class NsxPolicySecurityPolicyBaseApi(NsxPolicyResourceBase): self.policy_api.create_or_update(entry_def) def update_entries(self, domain_id, map_id, entries, + category=constants.CATEGORY_APPLICATION, tenant=constants.POLICY_INFRA_TENANT): - self.update_with_entries(domain_id, map_id, entries, tenant=tenant) + self.update_with_entries(domain_id, map_id, entries, category=category, + tenant=tenant) def update_with_entries(self, domain_id, map_id, entries=IGNORE, name=IGNORE, description=IGNORE,