~widelands-dev/widelands-website/django_staticfiles

#!/usr/bin/python -tt

 

#!/usr/bin/env python -tt

# encoding: utf-8

#

# File: utests/test_wl_markdown.py

#

# Created by Holger Rapp on 2009-02-28.

# Copyright (c) 2009 HolgerRapp@gmx.net. All rights reserved.

#

# Last Modified: $Date$

#

 

# Since we want to include something from one path up,

# we append the parent path to sys.path

import sys

sys.path.append('..')

 

import PIL

from cStringIO import StringIO

 

from django.test import TestCase

from django.contrib.contenttypes.models import ContentType

from django.contrib.auth.models import User

from django.core.files.uploadedfile import SimpleUploadedFile

from django.test.client import Client

from django.urls import reverse

from django.db import IntegrityError

 

from models import Image

from forms import UploadImageForm

 

from views import upload

 

 

class _TestUploadingBase(TestCase):

 

    @staticmethod

    def _make_new_uploaded_image(name, type='bmp'):

        sio = StringIO()

        i = PIL.Image.new('RGB', (4, 4))

 

        i.save(sio, type)

 

        return SimpleUploadedFile(name, sio.read(), content_type='image/%s' % type)

 

    def setUp(self):

        # We need some dummy objects

        # User

        self.u = User.objects.create(username='paul')

        # A Content type

        self.ct = ContentType.objects.create(

            app_label='test', model='TestModel')

 

        self.t1 = self._make_new_uploaded_image('test.png')

        self.t2 = self._make_new_uploaded_image('test.png')

        self.o1 = self._make_new_uploaded_image('othername.png')

 

        self.c = Client()

 

 

###########################################################################

#                  MODEL TESTS (need database, are slow)                  #

###########################################################################

class TestImages_TestModelAdding_ExceptCorrectResult(_TestUploadingBase):

 

    def runTest(self):

        self.assertFalse(Image.objects.has_image('test'))

        u = Image.objects.create(user=self.u, content_type=self.ct, object_id=1,

                                 name='test', revision=1)

        self.assertEqual(Image.objects.get(name='test', revision=1), u)

        self.assertTrue(Image.objects.has_image('test'))

 

 

class TestImages_TestModelAddingTwiceTheSameNameAndRevision_ExceptRaises(_TestUploadingBase):

 

    def runTest(self):

        u = Image.objects.create(user=self.u, content_type=self.ct, object_id=1,

                                 name='test', revision=1)

        self.assertRaises(Image.AlreadyExisting, Image.objects.create, **{'user': self.u, 'content_type': self.ct, 'object_id': 1,

                                                                          'name': 'test', 'revision': 1})

 

 

class TestImages_TestModelAddingTwiceTheSameNameDifferentRevision_ExceptRaises(_TestUploadingBase):

 

    def runTest(self):

        u = Image.objects.create(user=self.u, content_type=self.ct, object_id=1,

                                 name='test', revision=1)

        u = Image.objects.create(user=self.u, content_type=self.ct, object_id=1,

                                 name='test', revision=2)

        self.assertEqual(Image.objects.filter(name='test').count(), 2)

 

###############

# Other Tests #

###############

# This test is not of much use

# class TestImages_TestUploadForm_ExceptCorrectResult(_TestUploadingBase):

#     def runTest(self):

#         form = UploadImageForm()

#         self.assertEqual( form.is_valid(), False )

 

if __name__ == '__main__':

    unittest.main()

    # k = TestWlMarkdown_WikiWordsInLink_ExceptCorrectResult()

    # unittest.TextTestRunner().run(k)