Aarni Halinen
110 lines
3.9 KiB
Python
110 lines
3.9 KiB
Python
from django.test import TestCase
|
|
from django.contrib.auth.models import User
|
|
from django.utils import timezone
|
|
from rest_framework import status
|
|
from rest_framework.test import APITestCase, APIRequestFactory, force_authenticate
|
|
|
|
from webapp.models import Tag, Event
|
|
from webapp.serializers import EventSerializer
|
|
|
|
import tempfile
|
|
|
|
|
|
class EventTestCase(APITestCase):
|
|
|
|
def setUp(self):
|
|
self.icon = tempfile.NamedTemporaryFile(suffix=".jpg").name
|
|
Tag.objects.create(slug='testtag1', name='test1', icon=self.icon)
|
|
tag1 = Tag.objects.get(slug="testtag1")
|
|
Tag.objects.create(slug="testtag2", name='test2', icon=self.icon)
|
|
tag2 = Tag.objects.get(slug="testtag2")
|
|
self.assertEqual(Tag.objects.count(), 2)
|
|
|
|
# Visible and relevant
|
|
Event.objects.create(
|
|
title="TestEvent1",
|
|
visible=True,
|
|
description="diidadaapa",
|
|
content="lorem ipsum",
|
|
start_time=timezone.datetime(2019, 11, 9, 12, 0, 0),
|
|
end_time=timezone.datetime(2020, 11, 9, 13, 0, 0),
|
|
location="Testimaa"
|
|
)
|
|
# Invisible but relevant
|
|
Event.objects.create(
|
|
title="TestEvent2",
|
|
visible=False,
|
|
description="diidadaapa",
|
|
content="lorem ipsum",
|
|
start_time=timezone.datetime(2018, 12, 9, 12, 0, 0),
|
|
end_time=timezone.datetime(2020, 12, 9, 13, 0, 0),
|
|
location="Testimaan kerhohuone"
|
|
)
|
|
# Visible but unrelevant
|
|
Event.objects.create(
|
|
title="TestEvent3",
|
|
visible=True,
|
|
description="diidadaapa",
|
|
content="lorem ipsum",
|
|
start_time=timezone.datetime(2018, 12, 9, 12, 0, 0),
|
|
end_time=timezone.datetime(2018, 12, 9, 13, 0, 0),
|
|
location="Testimaan kerhohuone"
|
|
)
|
|
# Visible and relevant
|
|
Event.objects.create(
|
|
title="TestEvent4",
|
|
visible=True,
|
|
description="diidadaapa",
|
|
content="lorem ipsum",
|
|
start_time=timezone.datetime(2018, 12, 9, 12, 0, 0),
|
|
end_time=timezone.datetime(2020, 12, 9, 13, 0, 0),
|
|
location="Testimaan kerhohuone"
|
|
)
|
|
Event.objects.get(title="TestEvent1").tags.add(tag1)
|
|
Event.objects.get(title="TestEvent2").tags.add(tag2)
|
|
self.assertEqual(Event.objects.count(), 4)
|
|
|
|
username, password = 'test_admin', 'password123'
|
|
self.authClient = User.objects.create_superuser(username, 'myemail@test.com', password)
|
|
self.testEventId = Event.objects.get(title="TestEvent1").id
|
|
|
|
def test_get_events(self):
|
|
# Get from API
|
|
response = self.client.get("/api/events/", format="json")
|
|
# Response 200
|
|
self.assertTrue(status.is_success(response.status_code))
|
|
# Response should not have old events
|
|
self.assertEqual(len(response.data["results"]), 2)
|
|
# Check that serialized data is equal to received response
|
|
expected_events = EventSerializer(
|
|
Event.objects.filter(title__in=("TestEvent1", "TestEvent4")).order_by("start_time"),
|
|
many=True,
|
|
context={
|
|
"request": APIRequestFactory().get(r"http://testserver/api/events/")
|
|
}
|
|
)
|
|
self.assertEqual(response.data["results"], expected_events.data)
|
|
|
|
def test_get_single_event(self):
|
|
response = self.client.get(f"/api/events/{self.testEventId}/", format="json")
|
|
result = response.data
|
|
serialized = EventSerializer(
|
|
Event.objects.get(title="TestEvent1"),
|
|
context={
|
|
"request": APIRequestFactory().get(r"http://testserver/api/events/")
|
|
}
|
|
).data
|
|
self.assertEqual(
|
|
result,
|
|
serialized
|
|
)
|
|
|
|
def test_post_event(self):
|
|
pass
|
|
|
|
def test_update_event(self):
|
|
pass
|
|
|
|
def test_delete_event(self):
|
|
pass
|