|
22 | 22 | from google.cloud.spanner_admin_database_v1.types.common import DatabaseDialect |
23 | 23 | from google.cloud.spanner_v1 import backup, client, database, instance |
24 | 24 | from test_utils import retry |
| 25 | +from google.cloud.spanner_admin_instance_v1.types import spanner_instance_admin |
25 | 26 |
|
26 | 27 | INSTANCE_CREATION_TIMEOUT = 560 # seconds |
27 | 28 |
|
@@ -128,17 +129,24 @@ def sample_instance( |
128 | 129 | instance_config, |
129 | 130 | sample_name, |
130 | 131 | ): |
131 | | - sample_instance = spanner_client.instance( |
132 | | - instance_id, |
133 | | - instance_config, |
134 | | - labels={ |
135 | | - "cloud_spanner_samples": "true", |
136 | | - "sample_name": sample_name, |
137 | | - "created": str(int(time.time())), |
138 | | - }, |
| 132 | + operation = spanner_client.instance_admin_api.create_instance( |
| 133 | + parent=spanner_client.project_name, |
| 134 | + instance_id=instance_id, |
| 135 | + instance=spanner_instance_admin.Instance( |
| 136 | + config=instance_config, |
| 137 | + display_name="This is a display name.", |
| 138 | + node_count=1, |
| 139 | + labels={ |
| 140 | + "cloud_spanner_samples": "true", |
| 141 | + "sample_name": sample_name, |
| 142 | + "created": str(int(time.time())), |
| 143 | + }, |
| 144 | + edition=spanner_instance_admin.Instance.Edition.ENTERPRISE_PLUS, # Optional |
| 145 | + ), |
139 | 146 | ) |
140 | | - op = retry_429(sample_instance.create)() |
141 | | - op.result(INSTANCE_CREATION_TIMEOUT) # block until completion |
| 147 | + operation.result(INSTANCE_CREATION_TIMEOUT) # block until completion |
| 148 | + |
| 149 | + sample_instance = spanner_client.instance(instance_id) |
142 | 150 |
|
143 | 151 | # Eventual consistency check |
144 | 152 | retry_found = retry.RetryResult(bool) |
|
0 commit comments