refactor test

googleapis · Sep 17, 2024 · f49637f · f49637f
1 parent 24aa690
commit f49637f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/samples/samples/snippets.py b/samples/samples/snippets.py
@@ -62,7 +62,7 @@ def create_instance(instance_id):
                 "sample_name": "snippets-create_instance-explicit",
                 "created": str(int(time.time())),
             },
-            edition=spanner_instance_admin.Instance.Edition.ENTERPRISE_PLUS,  # Optional
+            edition=spanner_instance_admin.Instance.Edition.STANDARD,  # Optional
         ),
     )
 

diff --git a/samples/samples/snippets_test.py b/samples/samples/snippets_test.py
@@ -198,7 +198,9 @@ def test_create_instance_with_autoscaling_config(capsys, lci_instance_id):
 
 
 def test_create_instance_partition(capsys, instance_partition_instance_id):
-    snippets.create_instance(instance_partition_instance_id)
+    # Unable to use create_instance since it has editions set where partitions are unsupported. 
+    # The minimal requirement for editions is ENTERPRISE_PLUS for the paritions to get supported.
+    snippets.create_instance_with_processing_units(instance_partition_instance_id, 1000)
     retry_429(snippets.create_instance_partition)(
         instance_partition_instance_id, "my-instance-partition"
     )