files for whitenoise

aclowes · Aug 26, 2017 · 0d16104 · 0d16104
1 parent 216081d
commit 0d16104
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/yawn/management/tests/test_webserver.py b/yawn/management/tests/test_webserver.py
@@ -21,15 +21,18 @@ def test_webserver(mock_run):
 
 
 def test_static_files():
-    app = webserver.get_wsgi_application()
-    whitenoise = webserver.DefaultFileServer(app, settings)
-    whitenoise.application = mock.Mock()
-
     # 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(os.path.dirname(yawn.__file__), 'staticfiles', filename)
+        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()
+
     # an API request
     whitenoise({'PATH_INFO': '/api/'}, None)
     assert whitenoise.application.call_count == 1