瀏覽代碼

Added test for django.core.management.utils.get_random_secret_key().

Hasan Ramezani 6 年之前
父節點
當前提交
03e918d717
共有 1 個文件被更改,包括 9 次插入1 次删除
  1. 9 1
      tests/user_commands/tests.py

+ 9 - 1
tests/user_commands/tests.py

@@ -7,7 +7,9 @@ from admin_scripts.tests import AdminScriptTestCase
 from django.apps import apps
 from django.core import management
 from django.core.management import BaseCommand, CommandError, find_commands
-from django.core.management.utils import find_command, popen_wrapper
+from django.core.management.utils import (
+    find_command, get_random_secret_key, popen_wrapper,
+)
 from django.db import connection
 from django.test import SimpleTestCase, override_settings
 from django.test.utils import captured_stderr, extend_sys_path
@@ -260,3 +262,9 @@ class UtilsTests(SimpleTestCase):
         msg = 'Error executing a_42_command_that_doesnt_exist_42'
         with self.assertRaisesMessage(CommandError, msg):
             popen_wrapper(['a_42_command_that_doesnt_exist_42'])
+
+    def test_get_random_secret_key(self):
+        key = get_random_secret_key()
+        self.assertEqual(len(key), 50)
+        for char in key:
+            self.assertIn(char, 'abcdefghijklmnopqrstuvwxyz0123456789!@#$%^&*(-_=+)')