From b648c28d7049c30d53272fec7fe00dbf10b1efc5 Mon Sep 17 00:00:00 2001
From: Carl van Tonder <carl@supervacuo.com>
Date: Sat, 22 Sep 2018 22:25:04 -0400
Subject: [PATCH] Fix crashing test

---
 apps/files/tests.py | 4 ++--
 apps/files/urls.py  | 6 ++++--
 apps/files/views.py | 4 ++++
 3 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/apps/files/tests.py b/apps/files/tests.py
index 22812e8..edd8457 100644
--- a/apps/files/tests.py
+++ b/apps/files/tests.py
@@ -48,7 +48,7 @@ class FileTests(TestCase):
         login = self.login()
 
         with open('apps/map/static/map/ojuso-logo-white.png', 'rb') as fp:
-            response = self.client.post(reverse('files:upload'), {
+            response = self.client.post(reverse('files:upload_image'), {
                 'file': fp
             })
 
@@ -56,7 +56,7 @@ class FileTests(TestCase):
             self.assertIsInstance(data['id'], int)
             self.assertEqual(data['is_valid'], True)
 
-        response = self.client.post(reverse('files:delete', kwargs={
+        response = self.client.post(reverse('files:delete_image', kwargs={
             'pk': data['id']
         }))
 
diff --git a/apps/files/urls.py b/apps/files/urls.py
index aeb0607..c78b6ec 100644
--- a/apps/files/urls.py
+++ b/apps/files/urls.py
@@ -1,11 +1,13 @@
 from django.conf.urls import url
     
-from .views import FileUploadView, FileDeleteView, ImageFileUploadView
+from .views import (FileUploadView, FileDeleteView, ImageFileUploadView,
+                    ImageFileDeleteView)
 
 app_name = 'files'
 
 urlpatterns = [
     url(r'^upload/$', FileUploadView.as_view(), name='upload'),
-    url(r'^upload/image/$', ImageFileUploadView.as_view(), name='upload'),
+    url(r'^upload/image/$', ImageFileUploadView.as_view(), name='upload_image'),
     url(r'^delete/(?P<pk>\d+)/$', FileDeleteView.as_view(), name='delete'),
+    url(r'^delete/image/(?P<pk>\d+)/$', ImageFileDeleteView.as_view(), name='delete_image'),
 ]
diff --git a/apps/files/views.py b/apps/files/views.py
index 83f0e43..8f557ad 100644
--- a/apps/files/views.py
+++ b/apps/files/views.py
@@ -50,3 +50,7 @@ class FileDeleteView(LoginRequiredMixin, DetailView):
         return JsonResponse({
             'success': True
         })
+
+
+class ImageFileDeleteView(FileDeleteView):
+    model = ImageFile