From 4a6f2b63d7ad5907b3d64f8e4d318e7d59b4dd5f Mon Sep 17 00:00:00 2001 From: hashlash Date: Sat, 21 Mar 2020 03:20:48 +0700 Subject: Fixed #31380 -- Added deployment system check for DJANGO_ALLOW_ASYNC_UNSAFE environment variable. --- tests/check_framework/test_async_checks.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) create mode 100644 tests/check_framework/test_async_checks.py (limited to 'tests') diff --git a/tests/check_framework/test_async_checks.py b/tests/check_framework/test_async_checks.py new file mode 100644 index 0000000000..fa7f840172 --- /dev/null +++ b/tests/check_framework/test_async_checks.py @@ -0,0 +1,15 @@ +import os +from unittest import mock + +from django.core.checks.async_checks import E001, check_async_unsafe +from django.test import SimpleTestCase + + +class AsyncCheckTests(SimpleTestCase): + @mock.patch.dict(os.environ, {'DJANGO_ALLOW_ASYNC_UNSAFE': ''}) + def test_no_allowed_async_unsafe(self): + self.assertEqual(check_async_unsafe(None), []) + + @mock.patch.dict(os.environ, {'DJANGO_ALLOW_ASYNC_UNSAFE': 'true'}) + def test_allowed_async_unsafe_set(self): + self.assertEqual(check_async_unsafe(None), [E001]) -- cgit v1.3