Skip to content

Commit 573ec57

Browse files
refactor: drop all in tests
1 parent 4b08cf1 commit 573ec57

2 files changed

Lines changed: 4 additions & 5 deletions

File tree

openedx_authz/engine/utils.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -233,7 +233,6 @@ def migrate_legacy_course_roles_to_authz(course_access_role_model, course_id_lis
233233
course_access_role_model.objects.filter(**course_access_role_filter)
234234
.filter(Q(course_id=CourseKeyField.Empty) | Q(course_id__startswith=CourseOverviewData.NAMESPACE))
235235
.select_related("user")
236-
.all()
237236
)
238237

239238
# List to keep track of any permissions that could not be migrated

openedx_authz/tests/test_migrations.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -301,15 +301,15 @@ class MockQuerySet:
301301
def __init__(self, permissions):
302302
self.permissions = permissions
303303

304+
def __iter__(self):
305+
return iter(self.permissions)
306+
304307
def filter(self, *args, **kwargs):
305308
return self
306309

307310
def select_related(self, *args, **kwargs):
308311
return self
309312

310-
def all(self):
311-
return self.permissions
312-
313313
def get_or_create(self):
314314
raise Exception("Unexpected error mock")
315315

@@ -1191,7 +1191,7 @@ def test_migrate_legacy_course_roles_to_authz_instance_wide_role_is_error(self):
11911191
mock_qs = MagicMock()
11921192
mock_qs.filter.return_value = mock_qs
11931193
mock_qs.select_related.return_value = mock_qs
1194-
mock_qs.all.return_value = [instance_wide_permission]
1194+
mock_qs.__iter__ = MagicMock(return_value=iter([instance_wide_permission]))
11951195

11961196
mock_model = MagicMock()
11971197
mock_model.objects.filter.return_value = mock_qs

0 commit comments

Comments
 (0)