feat: cloud function schedule memory

avast · Jun 21, 2024 · f489a7b · f489a7b
1 parent a9d0ec2
commit f489a7b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/tests/deployment/test_gcp_connector.py b/tests/deployment/test_gcp_connector.py
@@ -126,7 +126,7 @@ def test_upsert_cloud_function(self):
             "service_account_email": "[email protected]",
             "labels": {},
             "environment_variables": {},
-            "available_memory_mb": 512
+            "available_memory_mb": 512,
         }
 
         # Set Mocks

diff --git a/tests/pipeline/test_pipeline_service.py b/tests/pipeline/test_pipeline_service.py
@@ -308,6 +308,7 @@ def test_run_pipeline(self, docker_mock):
                 "ENCRYPTION_SPEC_KEY_NAME": "projects/project_id/locations/region/keyRings/key_ring/cryptoKeys/key",
                 "PIPELINE_EXPERIMENT": "wanna-sample-experiment",
             },
+            "available_memory_mb": 512,
         }
 
         # Set Mocks