Add test to generate_fake_course_data command (openedx-unsupported#330)

Upgrade tqdm https://openedx.atlassian.net/browse/BOM-1359 This package is only used in the generate_fake_course_data which currently is not being tested, I added a test to check that everything is working as expected.
edx · Apr 14, 2020 · de1c84b · de1c84b
1 parent 86a5ada
commit de1c84b
Show file tree

Hide file tree

Showing 8 changed files with 36 additions and 7 deletions.
diff --git a/.coveragerc b/.coveragerc
@@ -1,7 +1,7 @@
 [run]
 omit = analyticsdataserver/settings*
        *wsgi.py
-       analytics_data_api/management/commands/generate_fake_course_data.py
+
 source = analyticsdataserver, analytics_data_api
 branch = True
 

diff --git a/analytics_data_api/management/commands/tests/test_generate_fake_course_data.py b/analytics_data_api/management/commands/tests/test_generate_fake_course_data.py
@@ -0,0 +1,29 @@
+
+from django.core.management import call_command
+from django.test import TestCase
+
+from analytics_data_api.v0 import models
+
+
+class GenerateFakeCourseDataTests(TestCase):
+    def testNormalRun(self):
+        num_weeks = 2
+        course_id = "edX/DemoX/Demo_Course"
+
+        call_command(
+            'generate_fake_course_data',
+            "--num-weeks={weeks}".format(weeks=num_weeks),
+            "--no-videos",
+            "--course-id", course_id
+        )
+
+        for model in [models.CourseEnrollmentDaily,
+                      models.CourseEnrollmentModeDaily,
+                      models.CourseEnrollmentByGender,
+                      models.CourseEnrollmentByEducation,
+                      models.CourseEnrollmentByBirthYear,
+                      models.CourseEnrollmentByCountry,
+                      models.CourseMetaSummaryEnrollment,
+                      models.CourseProgramMetadata]:
+            self.assertTrue(model.objects.filter(course_id=course_id, ).exists())
+            self.assertEqual(model.objects.filter(course_id=course_id).count(), model.objects.all().count())
diff --git a/requirements/base.in b/requirements/base.in
@@ -12,7 +12,7 @@ djangorestframework-csv             # BSD
 django-storages                     # BSD
 elasticsearch-dsl                   # Apache 2.0
 ordered-set                         # MIT
-tqdm==4.11.2                        # MIT
+tqdm                                # MIT
 urllib3                             # MIT
 Markdown==2.6.6                     # BSD:markdown is used by swagger for rendering the api docs
 edx-ccx-keys

diff --git a/requirements/base.txt b/requirements/base.txt
@@ -62,7 +62,7 @@ six==1.14.0               # via cryptography, djangorestframework-csv, edx-ccx-k
 slumber==0.7.1            # via edx-rest-api-client
 sqlparse==0.3.1           # via django
 stevedore==1.32.0         # via edx-opaque-keys
-tqdm==4.11.2              # via -r requirements/base.in
+tqdm==4.43.0              # via -r requirements/base.in
 unicodecsv==0.14.1        # via djangorestframework-csv
 uritemplate==3.0.1        # via coreapi
 urllib3==1.25.8           # via -r requirements/base.in, elasticsearch, requests
diff --git a/requirements/dev.txt b/requirements/dev.txt
@@ -62,7 +62,7 @@ six==1.14.0               # via cryptography, djangorestframework-csv, edx-ccx-k
 slumber==0.7.1            # via edx-rest-api-client
 sqlparse==0.3.1           # via django
 stevedore==1.32.0         # via edx-opaque-keys
-tqdm==4.11.2              # via -r requirements/base.in
+tqdm==4.43.0              # via -r requirements/base.in
 unicodecsv==0.14.1        # via djangorestframework-csv
 uritemplate==3.0.1        # via coreapi
 urllib3==1.25.8           # via -r requirements/base.in, elasticsearch, requests
diff --git a/requirements/doc.txt b/requirements/doc.txt
@@ -65,7 +65,7 @@ slumber==0.7.1            # via edx-rest-api-client
 sphinx==1.2.1             # via -r requirements/doc.in
 sqlparse==0.3.1           # via django
 stevedore==1.32.0         # via edx-opaque-keys
-tqdm==4.11.2              # via -r requirements/base.in
+tqdm==4.43.0              # via -r requirements/base.in
 unicodecsv==0.14.1        # via djangorestframework-csv
 uritemplate==3.0.1        # via coreapi
 urllib3==1.25.8           # via -r requirements/base.in, elasticsearch, requests
diff --git a/requirements/production.txt b/requirements/production.txt
@@ -67,7 +67,7 @@ six==1.14.0               # via cryptography, djangorestframework-csv, edx-ccx-k
 slumber==0.7.1            # via edx-rest-api-client
 sqlparse==0.3.1           # via django
 stevedore==1.32.0         # via edx-opaque-keys
-tqdm==4.11.2              # via -r requirements/base.in
+tqdm==4.43.0              # via -r requirements/base.in
 unicodecsv==0.14.1        # via djangorestframework-csv
 uritemplate==3.0.1        # via coreapi
 urllib3==1.25.8           # via -r requirements/base.in, elasticsearch, requests
diff --git a/requirements/test.txt b/requirements/test.txt
@@ -88,7 +88,7 @@ slumber==0.7.1            # via edx-rest-api-client
 snowballstemmer==2.0.0    # via pydocstyle
 sqlparse==0.3.1           # via django
 stevedore==1.32.0         # via edx-opaque-keys
-tqdm==4.11.2              # via -r requirements/base.in
+tqdm==4.43.0              # via -r requirements/base.in
 typed-ast==1.4.1          # via astroid
 unicodecsv==0.14.1        # via djangorestframework-csv
 uritemplate==3.0.1        # via coreapi