|
21 | 21 | ) |
22 | 22 | from databricks.sdk.retries import retried |
23 | 23 | from databricks.sdk.service import compute |
24 | | -from databricks.sdk.service.iam import PermissionLevel |
25 | 24 |
|
26 | 25 | from databricks.labs.ucx.__about__ import __version__ |
27 | 26 | from databricks.labs.ucx.config import WorkspaceConfig |
28 | 27 | from databricks.labs.ucx.install import WorkspaceInstaller |
29 | | -from databricks.labs.ucx.workspace_access.groups import MigratedGroup |
30 | 28 |
|
31 | 29 | from ..conftest import MockInstallationContext |
32 | 30 |
|
@@ -90,47 +88,6 @@ def factory( |
90 | 88 | pending.remove() |
91 | 89 |
|
92 | 90 |
|
93 | | -def test_experimental_permissions_migration_for_group_with_same_name( |
94 | | - installation_ctx, |
95 | | - make_cluster_policy, |
96 | | - make_cluster_policy_permissions, |
97 | | -): |
98 | | - ws_group, acc_group = installation_ctx.make_ucx_group() |
99 | | - migrated_group = MigratedGroup.partial_info(ws_group, acc_group) |
100 | | - cluster_policy = make_cluster_policy() |
101 | | - make_cluster_policy_permissions( |
102 | | - object_id=cluster_policy.policy_id, |
103 | | - permission_level=PermissionLevel.CAN_USE, |
104 | | - group_name=migrated_group.name_in_workspace, |
105 | | - ) |
106 | | - |
107 | | - schema_a = installation_ctx.make_schema() |
108 | | - table_a = installation_ctx.make_table(schema_name=schema_a.name) |
109 | | - installation_ctx.make_grant(migrated_group.name_in_workspace, 'USAGE', schema_info=schema_a) |
110 | | - installation_ctx.make_grant(migrated_group.name_in_workspace, 'OWN', schema_info=schema_a) |
111 | | - installation_ctx.make_grant(migrated_group.name_in_workspace, 'SELECT', table_info=table_a) |
112 | | - |
113 | | - installation_ctx.workspace_installation.run() |
114 | | - |
115 | | - installation_ctx.deployed_workflows.run_workflow("migrate-groups") |
116 | | - |
117 | | - object_permissions = installation_ctx.generic_permissions_support.load_as_dict( |
118 | | - "cluster-policies", cluster_policy.policy_id |
119 | | - ) |
120 | | - new_schema_grants = installation_ctx.grants_crawler.for_schema_info(schema_a) |
121 | | - |
122 | | - if {"USAGE", "OWN"} != new_schema_grants[migrated_group.name_in_account] or object_permissions[ |
123 | | - migrated_group.name_in_account |
124 | | - ] != PermissionLevel.CAN_USE: |
125 | | - installation_ctx.deployed_workflows.relay_logs("migrate-groups-experimental") |
126 | | - assert {"USAGE", "OWN"} == new_schema_grants[ |
127 | | - migrated_group.name_in_account |
128 | | - ], "Incorrect schema grants for migrated group" |
129 | | - assert ( |
130 | | - object_permissions[migrated_group.name_in_account] == PermissionLevel.CAN_USE |
131 | | - ), "Incorrect permissions for migrated group" |
132 | | - |
133 | | - |
134 | 91 | @retried(on=[NotFound, TimeoutError], timeout=timedelta(minutes=3)) |
135 | 92 | def test_job_failure_propagates_correct_error_message_and_logs(ws, sql_backend, installation_ctx): |
136 | 93 | installation_ctx.workspace_installation.run() |
|
0 commit comments