Joel Lavikainen
56 lines
2.2 KiB
Python
56 lines
2.2 KiB
Python
"""Tests for webapp."""
|
|
|
|
from django.test import TestCase
|
|
from django.core.files import File
|
|
from rest_framework.test import APITestCase
|
|
from rest_framework import status
|
|
from webapp.models import Tag
|
|
|
|
from collections import OrderedDict
|
|
from itertools import islice
|
|
import tempfile
|
|
|
|
|
|
class TagsTestCase(APITestCase):
|
|
|
|
|
|
def setUp(self):
|
|
self.icon = tempfile.NamedTemporaryFile(suffix=".jpg").name
|
|
Tag.objects.create(slug='Party', name='Bileet', icon=self.icon)
|
|
|
|
|
|
def test_get_single_tag(self):
|
|
self.assertEqual(Tag.objects.count(), 1)
|
|
response = self.client.get('/api/tags/', format='json')
|
|
self.assertTrue(status.is_success(response.status_code))
|
|
|
|
#We dont care about icon, so response is sliced
|
|
sliced_response = OrderedDict(islice(response.data[0].items(),2))
|
|
self.assertEqual(sliced_response, {'slug': 'Party', 'name': 'Bileet'})
|
|
def test_get_multiple_tags(self):
|
|
self.assertEqual(Tag.objects.count(), 1)
|
|
Tag.objects.create(slug='Freshmen', name='Fuksit', icon=self.icon)
|
|
Tag.objects.create(slug='International', name='Ulkkarit', icon=self.icon)
|
|
self.assertEqual(Tag.objects.count(), 3)
|
|
|
|
response = self.client.get('/api/tags/', format='json')
|
|
self.assertTrue(status.is_success(response.status_code))
|
|
|
|
print(response.data)
|
|
#We dont care about icon, so response is sliced
|
|
sliced_response = OrderedDict(islice(response.data[0].items(),2))
|
|
self.assertEqual(sliced_response, {'slug': 'Party', 'name': 'Bileet'})
|
|
sliced_response = OrderedDict(islice(response.data[1].items(),2))
|
|
self.assertEqual(sliced_response, {'slug': 'Freshmen', 'name': 'Fuksit'})
|
|
sliced_response = OrderedDict(islice(response.data[2].items(),2))
|
|
self.assertEqual(sliced_response, {'slug': 'International', 'name': 'Ulkkarit'})
|
|
|
|
|
|
def test_create_tag(self):
|
|
self.assertEqual(Tag.objects.count(), 1)
|
|
response = self.client.post('/api/tags/', {'slug': 'Test', 'name': 'Testinimi', 'icon': self.icon}, format='multipart')
|
|
self.assertFalse(status.is_success(response.status_code))
|
|
self.assertEqual(Tag.objects.count(), 1)
|
|
|
|
|
|
|