files for whitenoise

aclowes · Aug 26, 2017 · 3c7c8c4 · 3c7c8c4
1 parent 0d16104
commit 3c7c8c4
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 14 deletions.
diff --git a/yawn/management/tests/static/favicon.ico b/yawn/management/tests/static/favicon.ico
diff --git a/yawn/management/tests/static/index.html b/yawn/management/tests/static/index.html
diff --git a/yawn/management/tests/test_webserver.py b/yawn/management/tests/test_webserver.py
@@ -1,14 +1,11 @@
 import os
 import sys
-import subprocess
 
 from unittest import mock
 
 from django.conf import settings
 from django.core import management
 
-import yawn
-
 from yawn.management.commands import webserver
 
 
@@ -21,17 +18,11 @@ def test_webserver(mock_run):
 
 
 def test_static_files():
-    # make files to serve
-    root = os.path.join(os.path.dirname(yawn.__file__), 'staticfiles/')
-    os.makedirs(root, exist_ok=True)
-    for filename in ('index.html', 'favicon.ico'):
-        path = os.path.join(root, filename)
-        subprocess.check_call(['touch', path])
-
-    # setup whitenoise after the files exist
-    app = webserver.get_wsgi_application()
-    whitenoise = webserver.DefaultFileServer(app, settings)
-    whitenoise.application = mock.Mock()
+    root = os.path.join(settings.BASE_DIR, 'management/tests/static')
+    with mock.patch.object(settings, 'WHITENOISE_ROOT', root):
+        app = webserver.get_wsgi_application()
+        whitenoise = webserver.DefaultFileServer(app, settings)
+        whitenoise.application = mock.Mock()
 
     # an API request
     whitenoise({'PATH_INFO': '/api/'}, None)