Skip to content

Commit 7471fff

Browse files
chore(spanner): create multi-region ENTERPRISE_PLUS instance in test_create_instance_partition
1 parent cdcfe90 commit 7471fff

1 file changed

Lines changed: 20 additions & 6 deletions

File tree

packages/google-cloud-spanner/samples/samples/snippets_test.py

Lines changed: 20 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -213,18 +213,32 @@ def test_create_and_update_instance_default_backup_schedule_type(
213213
retry_429(instance.delete)()
214214

215215

216-
def test_create_instance_partition(capsys, instance_partition_instance_id):
217-
# Unable to use create_instance since it has editions set where partitions are unsupported.
218-
# The minimal requirement for editions is ENTERPRISE_PLUS for the paritions to get supported.
219-
retry_429(snippets.create_instance_with_processing_units)(
220-
instance_partition_instance_id, 1000
216+
def test_create_instance_partition(
217+
capsys, instance_partition_instance_id, multi_region_instance_config
218+
):
219+
from google.cloud.spanner_admin_instance_v1.types import spanner_instance_admin
220+
221+
spanner_client = spanner.Client()
222+
request = spanner_instance_admin.CreateInstanceRequest(
223+
parent=spanner_client.project_name,
224+
instance_id=instance_partition_instance_id,
225+
instance=spanner_instance_admin.Instance(
226+
config=multi_region_instance_config,
227+
display_name="Instance for instance partition test",
228+
node_count=1,
229+
edition=spanner_instance_admin.Instance.Edition.ENTERPRISE_PLUS,
230+
),
221231
)
232+
operation = retry_429(spanner_client.instance_admin_api.create_instance)(
233+
request=request
234+
)
235+
operation.result(900)
236+
222237
retry_429(snippets.create_instance_partition)(
223238
instance_partition_instance_id, "my-instance-partition"
224239
)
225240
out, _ = capsys.readouterr()
226241
assert "Created instance partition my-instance-partition" in out
227-
spanner_client = spanner.Client()
228242
instance = spanner_client.instance(instance_partition_instance_id)
229243
retry_429(instance.delete)()
230244

0 commit comments

Comments
 (0)