Merge pull request #174 from GoogleCloudPlatform/bug/update_test_quic…

…kfix Pass the correct parameter to _update_kwargs
GoogleCloudPlatform · Feb 20, 2024 · 37a4152 · 37a4152
2 parents 529cbab + 6c6d868
commit 37a4152
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/dfcx_scrapi/core/environments.py b/src/dfcx_scrapi/core/environments.py
@@ -309,10 +309,10 @@ def update_environment(
         if environment_obj:
             env = environment_obj
             env.name = environment_id
-            mask = self._update_kwargs(environment_obj)
+            mask = self._update_kwargs(env)
         elif kwargs:
             env = self.get_environment(environment_id)
-            mask = self._update_kwargs(environment_obj, **kwargs)
+            mask = self._update_kwargs(env, **kwargs)
 
         request = types.environment.UpdateEnvironmentRequest(
             environment=env, update_mask=mask)

diff --git a/src/dfcx_scrapi/core/test_cases.py b/src/dfcx_scrapi/core/test_cases.py
@@ -462,12 +462,12 @@ def update_test_case(
         if obj:
             test_case = obj
             test_case.name = test_case_id
-            mask = self._update_kwargs(obj)
+            mask = self._update_kwargs(test_case)
         elif kwargs:
             if not test_case_id:
                 test_case_id = self.test_case_id
             test_case = self.get_test_case(test_case_id)
-            mask = self._update_kwargs(obj, **kwargs)
+            mask = self._update_kwargs(test_case, **kwargs)
 
         request = types.test_case.UpdateTestCaseRequest(
             test_case=test_case, update_mask=mask)