4141REALM_LOCATION = "global"
4242
4343# The format of realm ID. This is used in the unit tests and cleanup below.
44- realm_id_format = ' test-realm-{}-{}'
44+ realm_id_format = " test-realm-{}-{}"
4545
4646# The format of deployment ID. This is used in the unit tests and cleanup below.
47- deployment_id_format = ' test-deployment-{}-{}'
47+ deployment_id_format = " test-deployment-{}-{}"
4848
4949
5050@pytest .fixture (scope = "session" , autouse = True )
5151def clean_up_old_deployments ():
5252 all_deployments = list_deployments .list_deployments (PROJECT_ID )
5353 for deployment in all_deployments :
5454 deployment_name = deployment .name
55- deployment_id = deployment_name [deployment_name .rfind ('/' ) + 1 : len (deployment_name )]
56- if deployment_id .find ('test-deployment-' ) == 0 :
57- time_str = deployment_id [deployment_id .rfind ('-' ) + 1 : len (deployment_id )]
55+ deployment_id = deployment_name [
56+ deployment_name .rfind ("/" ) + 1 : len (deployment_name )
57+ ]
58+ if deployment_id .find ("test-deployment-" ) == 0 :
59+ time_str = deployment_id [deployment_id .rfind ("-" ) + 1 : len (deployment_id )]
5860 test_date = datetime .datetime .utcfromtimestamp (int (time_str ))
5961 now_date = datetime .datetime .utcfromtimestamp (int (time .time ()))
6062 difftime = now_date - test_date
6163
6264 # *NOTE* Restrict to deployments used in the tests older than 2 days
6365 # to prevent thrashing in the case of async tests
64- if (difftime .days > 2 ):
65- print (f"Cleaning up old deployment { deployment_id } and its configs, difftime: { difftime } " )
66+ if difftime .days > 2 :
67+ print (
68+ f"Cleaning up old deployment { deployment_id } and its configs, difftime: { difftime } "
69+ )
6670 clean_up_deployment_and_configs (deployment_id )
6771
6872
@@ -95,7 +99,9 @@ def test_deployment_with_config():
9599 print (f"Creating deployment { deployment_id } in project { PROJECT_ID } " )
96100 create_deployment .create_deployment (PROJECT_ID , deployment_id )
97101
98- print (f"Creating config { CONFIG_ID } in deployment { deployment_id } in project { PROJECT_ID } " )
102+ print (
103+ f"Creating config { CONFIG_ID } in deployment { deployment_id } in project { PROJECT_ID } "
104+ )
99105 create_config .create_config (PROJECT_ID , deployment_id , CONFIG_ID )
100106
101107 yield deployment_id
@@ -112,18 +118,22 @@ def clean_up_deployment_and_configs(deployment_id):
112118 return
113119
114120 try :
115- update_rollout_remove_default .update_rollout_remove_default (PROJECT_ID , deployment_id )
121+ update_rollout_remove_default .update_rollout_remove_default (
122+ PROJECT_ID , deployment_id
123+ )
116124 except exceptions .NotFound : # Ignore the non-existent deployment
117125 return
118126
119127 try :
120- update_rollout_remove_override .update_rollout_remove_override (PROJECT_ID , deployment_id )
128+ update_rollout_remove_override .update_rollout_remove_override (
129+ PROJECT_ID , deployment_id
130+ )
121131 except exceptions .NotFound : # Ignore the non-existent deployment
122132 return
123133
124134 configs = list_configs .list_configs (PROJECT_ID , deployment_id )
125135 for config in configs :
126- config_id = config .name .rsplit ('/' , 1 )[- 1 ]
136+ config_id = config .name .rsplit ("/" , 1 )[- 1 ]
127137 print (f"Deleting config { config_id } in deployment { deployment_id } " )
128138 try :
129139 delete_config .delete_config (PROJECT_ID , deployment_id , config_id )
@@ -141,7 +151,9 @@ def clean_up_deployment_and_configs(deployment_id):
141151def test_realm ():
142152 realm_id = realm_id_format .format (uuid .uuid4 ().hex , int (time .time ()))
143153
144- print (f"Creating realm { realm_id } in location { REALM_LOCATION } in project { PROJECT_ID } " )
154+ print (
155+ f"Creating realm { realm_id } in location { REALM_LOCATION } in project { PROJECT_ID } "
156+ )
145157 create_realm .create_realm (PROJECT_ID , REALM_LOCATION , realm_id )
146158
147159 yield realm_id
@@ -165,7 +177,10 @@ def test_create_deployment(test_deployment):
165177
166178def test_get_deployment (test_deployment ):
167179 deployment = get_deployment .get_deployment (PROJECT_ID , test_deployment )
168- assert deployment .name == f"projects/{ PROJECT_ID } /locations/global/gameServerDeployments/{ test_deployment } "
180+ assert (
181+ deployment .name
182+ == f"projects/{ PROJECT_ID } /locations/global/gameServerDeployments/{ test_deployment } "
183+ )
169184
170185
171186def test_list_deployments (test_deployment ):
@@ -182,7 +197,10 @@ def test_list_deployments(test_deployment):
182197def test_update_deployment (test_deployment ):
183198 update_deployment .update_deployment (PROJECT_ID , test_deployment )
184199 deployment = get_deployment .get_deployment (PROJECT_ID , test_deployment )
185- assert deployment .labels == {"label-key-1" : "label-value-1" , "label-key-2" : "label-value-2" }
200+ assert deployment .labels == {
201+ "label-key-1" : "label-value-1" ,
202+ "label-key-2" : "label-value-2" ,
203+ }
186204
187205
188206def test_delete_deployment (test_deployment ):
@@ -193,42 +211,66 @@ def test_delete_deployment(test_deployment):
193211
194212def test_get_rollout (test_deployment ):
195213 rollout = get_rollout .get_rollout (PROJECT_ID , test_deployment )
196- assert rollout .name == f"projects/{ PROJECT_ID } /locations/global/gameServerDeployments/{ test_deployment } /rollout"
214+ assert (
215+ rollout .name
216+ == f"projects/{ PROJECT_ID } /locations/global/gameServerDeployments/{ test_deployment } /rollout"
217+ )
197218
198219
199220def test_update_rollout_default (test_deployment_with_config ):
200- update_rollout_default .update_rollout_default (PROJECT_ID , test_deployment_with_config , CONFIG_ID )
221+ update_rollout_default .update_rollout_default (
222+ PROJECT_ID , test_deployment_with_config , CONFIG_ID
223+ )
201224 rollout = get_rollout .get_rollout (PROJECT_ID , test_deployment_with_config )
202- assert rollout .default_game_server_config == f"projects/{ PROJECT_ID } /locations/global/gameServerDeployments/{ test_deployment_with_config } /configs/{ CONFIG_ID } "
225+ assert (
226+ rollout .default_game_server_config
227+ == f"projects/{ PROJECT_ID } /locations/global/gameServerDeployments/{ test_deployment_with_config } /configs/{ CONFIG_ID } "
228+ )
203229
204230
205231def test_update_rollout_override (test_realm , test_deployment_with_config ):
206- update_rollout_override .update_rollout_override (PROJECT_ID , test_deployment_with_config , CONFIG_ID , REALM_LOCATION , test_realm )
232+ update_rollout_override .update_rollout_override (
233+ PROJECT_ID , test_deployment_with_config , CONFIG_ID , REALM_LOCATION , test_realm
234+ )
207235 rollout = get_rollout .get_rollout (PROJECT_ID , test_deployment_with_config )
208236 assert len (rollout .game_server_config_overrides ) == 1
209- assert rollout .game_server_config_overrides [0 ].config_version == f"projects/{ PROJECT_ID } /locations/global/gameServerDeployments/{ test_deployment_with_config } /configs/{ CONFIG_ID } "
210- assert rollout .game_server_config_overrides [0 ].realms_selector .realms == [f"projects/{ PROJECT_ID } /locations/{ REALM_LOCATION } /realms/{ test_realm } " ]
237+ assert (
238+ rollout .game_server_config_overrides [0 ].config_version
239+ == f"projects/{ PROJECT_ID } /locations/global/gameServerDeployments/{ test_deployment_with_config } /configs/{ CONFIG_ID } "
240+ )
241+ assert rollout .game_server_config_overrides [0 ].realms_selector .realms == [
242+ f"projects/{ PROJECT_ID } /locations/{ REALM_LOCATION } /realms/{ test_realm } "
243+ ]
211244
212245
213246def test_update_rollout_remove_default (test_deployment ):
214- update_rollout_remove_default .update_rollout_remove_default (PROJECT_ID , test_deployment )
247+ update_rollout_remove_default .update_rollout_remove_default (
248+ PROJECT_ID , test_deployment
249+ )
215250 rollout = get_rollout .get_rollout (PROJECT_ID , test_deployment )
216251 assert rollout .default_game_server_config == ""
217252
218253
219254def test_update_rollout_remove_override (test_deployment ):
220- update_rollout_remove_override .update_rollout_remove_override (PROJECT_ID , test_deployment )
255+ update_rollout_remove_override .update_rollout_remove_override (
256+ PROJECT_ID , test_deployment
257+ )
221258 rollout = get_rollout .get_rollout (PROJECT_ID , test_deployment )
222259 assert len (rollout .game_server_config_overrides ) == 0
223260
224261
225262def test_create_conifg (test_deployment_with_config ):
226- print (f"Created config { CONFIG_ID } in deployment { test_deployment_with_config } in project { PROJECT_ID } " )
263+ print (
264+ f"Created config { CONFIG_ID } in deployment { test_deployment_with_config } in project { PROJECT_ID } "
265+ )
227266
228267
229268def test_get_config (test_deployment_with_config ):
230269 config = get_config .get_config (PROJECT_ID , test_deployment_with_config , CONFIG_ID )
231- assert config .name == f"projects/{ PROJECT_ID } /locations/global/gameServerDeployments/{ test_deployment_with_config } /configs/{ CONFIG_ID } "
270+ assert (
271+ config .name
272+ == f"projects/{ PROJECT_ID } /locations/global/gameServerDeployments/{ test_deployment_with_config } /configs/{ CONFIG_ID } "
273+ )
232274
233275
234276def test_list_configs (test_deployment_with_config ):
0 commit comments