From 7b8b4ddad629a750f894f3e2e9a7b0805ab825e4 Mon Sep 17 00:00:00 2001 From: Adit Sarfaty Date: Wed, 7 Nov 2018 13:52:03 +0200 Subject: [PATCH] Support update tier0 in policy tier1 router Change-Id: I6de8a97dcabacbbccca79ba03ece762f94b7a6fe --- vmware_nsxlib/tests/unit/v3/test_policy_resources.py | 4 +++- vmware_nsxlib/v3/policy_resources.py | 5 +++-- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/vmware_nsxlib/tests/unit/v3/test_policy_resources.py b/vmware_nsxlib/tests/unit/v3/test_policy_resources.py index 1b1a1924..b357ccd9 100644 --- a/vmware_nsxlib/tests/unit/v3/test_policy_resources.py +++ b/vmware_nsxlib/tests/unit/v3/test_policy_resources.py @@ -1608,13 +1608,15 @@ class TestPolicyTier1(NsxPolicyLibTestCase): def test_update(self): id = '111' name = 'new name' + tier0 = 'tier0' with mock.patch.object(self.policy_api, "create_or_update") as update_call: self.resourceApi.update(id, - name=name, + name=name, tier0=tier0, tenant=TEST_TENANT) expected_def = policy_defs.Tier1Def(tier1_id=id, name=name, + tier0=tier0, tenant=TEST_TENANT) self.assert_called_with_def( update_call, expected_def) diff --git a/vmware_nsxlib/v3/policy_resources.py b/vmware_nsxlib/v3/policy_resources.py index a40986a3..58c0d201 100644 --- a/vmware_nsxlib/v3/policy_resources.py +++ b/vmware_nsxlib/v3/policy_resources.py @@ -563,15 +563,16 @@ class NsxPolicyTier1Api(NsxPolicyResourceBase): def update(self, tier1_id, name=None, description=None, force_whitelisting=None, - failover_mode=None, + failover_mode=None, tier0=None, tags=None, tenant=policy_constants.POLICY_INFRA_TENANT): - + # TODO(asarfaty): Support tier0=None fore reseting the tier0 value self._update(tier1_id=tier1_id, name=name, description=description, force_whitelisting=force_whitelisting, failover_mode=failover_mode, + tier0=tier0, tags=tags, tenant=tenant)