From c798ccb093d4556297b14359ae290b9621ed050e Mon Sep 17 00:00:00 2001
From: Benjamin Bach <benjamin@overtag.dk>
Date: Tue, 10 May 2016 15:40:22 +0200
Subject: [PATCH] Add basic view test for image uploading

---
 wiki/plugins/images/tests/__init__.py   |  6 ++
 wiki/plugins/images/tests/test_views.py | 87 +++++++++++++++++++++++++
 2 files changed, 93 insertions(+)
 create mode 100644 wiki/plugins/images/tests/__init__.py
 create mode 100644 wiki/plugins/images/tests/test_views.py

diff --git a/wiki/plugins/images/tests/__init__.py b/wiki/plugins/images/tests/__init__.py
new file mode 100644
index 00000000..e7538a06
--- /dev/null
+++ b/wiki/plugins/images/tests/__init__.py
@@ -0,0 +1,6 @@
+from __future__ import absolute_import
+
+import django
+if django.VERSION < (1, 6):
+    # New style autodiscovery of tests doesn't work for Django < 1.6
+    from .test_views import *  # noqa
diff --git a/wiki/plugins/images/tests/test_views.py b/wiki/plugins/images/tests/test_views.py
new file mode 100644
index 00000000..f790ff44
--- /dev/null
+++ b/wiki/plugins/images/tests/test_views.py
@@ -0,0 +1,87 @@
+from __future__ import print_function, unicode_literals
+
+import base64
+from io import BytesIO
+
+from django.core.files.uploadedfile import InMemoryUploadedFile
+from django.core.urlresolvers import reverse
+from wiki.core.plugins import registry as plugin_registry
+from wiki.tests.base import ArticleWebTestBase
+
+from .. import models
+from ..wiki_plugin import ImagePlugin
+
+
+class ImageTests(ArticleWebTestBase):
+
+    def setUp(self):
+        super(ImageTests, self).setUp()
+        self.article = self.root_article
+        # A black 1x1 gif
+        self.test_data = "R0lGODlhAQABAIAAAAUEBAAAACwAAAAAAQABAAACAkQBADs="
+
+    def _create_gif_filestream_from_base64(self, str_base64, **kwargs):
+        """
+        Helper function to create filestream for upload.
+
+        Parameters :
+        strData : str, test string data
+
+        Optional Arguments :
+        filename : str, Defaults to 'test.txt'
+        """
+        filename = kwargs.get('filename', 'test.gif')
+        data = base64.b64decode(str_base64)
+        filedata = BytesIO(data)
+        filestream = InMemoryUploadedFile(
+            filedata,
+            None,
+            filename,
+            'image',
+            len(data),
+            None
+        )
+        return filestream
+
+    def _create_test_image(self):
+        # Get the form index
+        plugin_index = -1
+        for cnt, plugin_instance in enumerate(plugin_registry.get_sidebar()):
+            if isinstance(plugin_instance, ImagePlugin):
+                plugin_index = cnt
+                break
+        self.assertTrue(plugin_index >= 0, "Image plugin not activated")
+        base_edit_url = reverse('wiki:edit', kwargs={'path': ''})
+        url = base_edit_url + '?f=form{0:d}'.format(plugin_index)
+        filestream = self._create_gif_filestream_from_base64(self.test_data)
+        response = self.c.post(
+            url,
+            {
+                'unsaved_article_title': self.article.current_revision.title,
+                'unsaved_article_content': self.article.current_revision.content,
+                'image': filestream,
+                'images_save': '1',
+            },
+        )
+        self.assertRedirects(response, base_edit_url)
+
+    def test_index(self):
+        url = reverse('wiki:images_index', kwargs={'path': ''})
+        response = self.c.get(url,)
+        self.assertContains(response, 'Images')
+
+    def test_upload(self):
+        """
+        Tests that simple file upload uploads correctly
+        Uploading a file should preserve the original filename.
+        Uploading should not modify file in any way.
+        """
+        self._create_test_image()
+        # Check the object was created.
+        image = models.Image.objects.get()
+        image_revision = image.current_revision.imagerevision
+        self.assertEqual(image_revision.get_filename(), 'test.gif')
+        self.assertEqual(
+            image_revision.image.file.read(),
+            base64.b64decode(self.test_data)
+        )
-- 
2.45.2