test: add some changes

course_discovery/apps/course_metadata/managers.py

@@ -8,18 +8,28 @@
 class DraftManager(models.Manager):
     """ Model manager that hides draft rows unless you ask for them. """
 
-    def get_queryset(self):
-        return super().get_queryset().filter(draft=models.Value(0))
-
-    def _with_drafts(self):
-        return super().get_queryset()
-
-    def filter_drafts(self, **kwargs):
+    def get_queryset(self, allow_restricted=False):
+        if not allow_restricted:
+            return super().get_queryset().filter(draft=models.Value(0), restricted_run__isnull=True)
+        else:
+            return super().get_queryset().filter(draft=models.Value(0))
+
+    def _with_drafts(self, allow_restricted=False):
+        if not allow_restricted:
+            return super().get_queryset().filter(restricted_run__isnull=True)
+        else:
+            return super().get_queryset()
+
+    def filter_drafts(self, allow_restricted=False, **kwargs):
         """
         Acts like filter(), but prefers draft versions.
         If a draft is not available, we give back the non-draft version.
         """
-        return self._with_drafts().filter(Q(draft=models.Value(1)) | Q(draft_version=None)).filter(**kwargs)
+        if allow_restricted:
+            return self._with_drafts(allow_restricted=True).filter(Q(draft=models.Value(1)) | Q(draft_version=None)).filter(**kwargs)
+        else:
+            return self._with_drafts().filter(Q(draft=models.Value(1)) | Q(draft_version=None)).filter(**kwargs)
+
 
     def get_draft(self, **kwargs):
         """