wip: support for app_name in URLPatternsTestCase

shuckc · shuckc · commit 043561b1ea58 · 2018-02-25T19:38:55.000Z
diff --git a/rest_framework/test.py b/rest_framework/test.py
@@ -385,9 +385,14 @@ def setUpClass(cls):
 
         if hasattr(cls._module, 'urlpatterns'):
             cls._module_urlpatterns = cls._module.urlpatterns
+        if hasattr(cls._module, 'app_name'):
+            cls._module_app_name = cls._module.app_name
 
         cls._module.urlpatterns = cls.urlpatterns
 
+        if hasattr(cls, 'app_name'):
+            cls._module.app_name = cls.app_name
+
         cls._override.enable()
         super(URLPatternsTestCase, cls).setUpClass()
 
@@ -400,3 +405,9 @@ def tearDownClass(cls):
             cls._module.urlpatterns = cls._module_urlpatterns
         else:
             del cls._module.urlpatterns
+
+        if hasattr(cls, '_module_app_name'):
+            cls._module.app_name = cls._module_app_name
+        else:
+            if hasattr(cls._module, 'app_name'):
+                del cls._module.app_name
