BOM-881

python3 compatibility
2019-09-30 14:58:12 +05:00
parent cc4050460e
commit 86bd5a4643
1 changed files with 5 additions and 1 deletions
--- a/openedx/core/djangoapps/user_api/management/tests/test_email_opt_in_list.py
+++ b/openedx/core/djangoapps/user_api/management/tests/test_email_opt_in_list.py
@@ -274,10 +274,14 @@ class EmailOptInListTest(ModuleStoreTestCase):
            expected_msg_regex = (
                "^Error: too few arguments$"
            )
-        else:
+        elif num_args == 1:
            expected_msg_regex = (
                "^Error: the following arguments are required: ORG_ALIASES$"
            )
+        elif num_args == 0:
+            expected_msg_regex = (
+                "^Error: the following arguments are required: OUTPUT_FILENAME, ORG_ALIASES$"
+            )
        with self.assertRaisesRegexp(CommandError, expected_msg_regex):
            call_command('email_opt_in_list', *args)