|
7 | 7 | waffle models are not imported in the test environment). |
8 | 8 | """ |
9 | 9 |
|
| 10 | +from datetime import datetime, timezone |
10 | 11 | from types import SimpleNamespace |
11 | | -from unittest.mock import patch |
| 12 | +from unittest.mock import MagicMock, patch |
12 | 13 |
|
13 | 14 | from casbin_adapter.models import CasbinRule |
14 | 15 | from ddt import data, ddt, unpack |
15 | 16 | from django.test import TestCase, override_settings |
| 17 | +from openedx_events.authz.data import RoleAssignmentData |
16 | 18 |
|
17 | 19 | from openedx_authz.handlers import ( |
18 | 20 | WAFFLE_OVERRIDE_FORCE_OFF, |
19 | 21 | WAFFLE_OVERRIDE_FORCE_ON, |
| 22 | + create_audit_record_on_role_assignment_change, |
20 | 23 | get_migration_type, |
21 | 24 | trigger_course_authoring_migration, |
22 | 25 | ) |
23 | 26 | from openedx_authz.models.authz_migration import MigrationType, ScopeType |
24 | | -from openedx_authz.models.core import ExtendedCasbinRule, Scope, Subject |
| 27 | +from openedx_authz.models.core import ExtendedCasbinRule, RoleAssignmentAudit, Scope, Subject |
25 | 28 | from openedx_authz.models.subjects import UserSubject |
26 | 29 | from openedx_authz.tests.stubs.models import ( |
27 | 30 | CourseAccessRole, |
28 | 31 | WaffleFlagCourseOverrideModel, |
29 | 32 | WaffleFlagOrgOverrideModel, |
30 | 33 | ) |
31 | 34 |
|
32 | | -from datetime import datetime, timezone |
33 | | -from unittest.mock import MagicMock, patch |
34 | | - |
35 | | -from casbin_adapter.models import CasbinRule |
36 | | -from django.test import TestCase |
37 | | -from openedx_events.authz.data import RoleAssignmentData |
38 | | - |
39 | | -from openedx_authz.handlers import create_audit_record_on_role_assignment_change |
40 | | -from openedx_authz.models.core import ExtendedCasbinRule, RoleAssignmentAudit, Scope, Subject |
41 | | - |
42 | 35 | AUTHZ_COURSE_AUTHORING_FLAG_NAME = "authz.enable_course_authoring" |
43 | 36 | OTHER_WAFFLE_FLAG_NAME = "some.other.flag" |
44 | 37 |
|
|
0 commit comments