Skip to content

Commit ab6e2b9

Browse files
committed
Fix fmt
1 parent 42a0398 commit ab6e2b9

File tree

1 file changed

+4
-5
lines changed
  • src/databricks/labs/pytester/fixtures

1 file changed

+4
-5
lines changed

src/databricks/labs/pytester/fixtures/iam.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -344,8 +344,7 @@ def create(*, account_groups: list[str] | None = None):
344344
workspace_id = ws.get_workspace_id()
345345
service_principal = acc.service_principals.create(display_name=f'spn-{make_random()}')
346346
assert service_principal.id is not None
347-
service_principal_id = int(service_principal.id)
348-
created_secret = acc.service_principal_secrets.create(service_principal_id)
347+
created_secret = acc.service_principal_secrets.create(service_principal.id)
349348
if account_groups:
350349
group_mapping = {}
351350
for group in acc.groups.list(attributes='id,displayName'):
@@ -359,15 +358,15 @@ def create(*, account_groups: list[str] | None = None):
359358
acc.groups.patch(
360359
group_id,
361360
operations=[
362-
Patch(PatchOp.ADD, 'members', [ComplexValue(value=str(service_principal_id)).as_dict()]),
361+
Patch(PatchOp.ADD, 'members', [ComplexValue(value=str(service_principal.id)).as_dict()]),
363362
],
364363
schemas=[PatchSchema.URN_IETF_PARAMS_SCIM_API_MESSAGES_2_0_PATCH_OP],
365364
)
366365
permissions = [WorkspacePermission.USER]
367-
acc.workspace_assignment.update(workspace_id, service_principal_id, permissions=permissions)
366+
acc.workspace_assignment.update(workspace_id, int(service_principal.id), permissions=permissions)
368367
ws_as_spn = _make_workspace_client(ws, created_secret, service_principal)
369368

370-
log_account_link('account service principal', f'users/serviceprincipals/{service_principal_id}')
369+
log_account_link('account service principal', f'users/serviceprincipals/{service_principal.id}')
371370

372371
return RunAs(service_principal, ws_as_spn, env_or_skip)
373372

0 commit comments

Comments
 (0)