fix UT

sonic-net · Oct 25, 2022 · c3544a2 · c3544a2
1 parent b4f466d
commit c3544a2
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/validated_config_db_connector_test.py b/tests/validated_config_db_connector_test.py
@@ -42,5 +42,7 @@ def test_validated_delete_table_invalid_delete(self):
 
     def test_create_gcu_patch(self):
         expected_gcu_patch = jsonpatch.JsonPatch([{"op": "add", "path": "/PORTCHANNEL/PortChannel01", "value": "test"}])
-        created_gcu_patch = validated_config_db_connector.ValidatedConfigDBConnector.create_gcu_patch(ValidatedConfigDBConnector(ConfigDBConnector()), "add", "PORTCHANNEL", "PortChannel01", "test")
-        assert expected_gcu_patch == created_gcu_patch
+        with mock.patch('validated_config_db_connector.ConfigDBConnector.get_table', return_value=True):
+            with mock.patch('validated_config_db_connector.ConfigDBConnector.get_entry', return_value=True):
+                created_gcu_patch = validated_config_db_connector.ValidatedConfigDBConnector.create_gcu_patch(ValidatedConfigDBConnector(ConfigDBConnector()), "add", "PORTCHANNEL", "PortChannel01", "test")
+                assert expected_gcu_patch == created_gcu_patch