@@ -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