Merge pull request #982 from openedx/pwnage101/update-restricted-runs…

…-test-data

docs: update test data script for testing restricted runs

scripts/setup_restricted_runs_test_courses.py

@@ -153,3 +153,19 @@
     course=restricted_course_unicorn,
     run=course_unicorn_run1,
 )[0])
+
+# Finally update the catalog query content filters to reflect the relationships
+# =============================================================================
+catalog_query.content_filter |= {
+    'restricted_runs_allowed': {
+        f'course:{course_mixed.content_key}': [course_mixed_run2.content_key],
+        f'course:{course_unicorn.content_key}': [course_mixed_run1.content_key],
+    },
+}
+catalog_query.save()
+catalog_query_2.content_filter |= {
+    'restricted_runs_allowed': {
+        f'course:{course_mixed.content_key}': [course_mixed_run3.content_key],
+    },
+}
+catalog_query_2.save()