GetTogether/get_together/tests/events.py

38 lines
1,007 B
Python

from django.test import TestCase, Client
from model_mommy import mommy
from django.contrib.auth.models import User
from events.models import Event, Attendee, UserProfile
# Create your tests here.
class EventDisplayTests(TestCase):
def setUp(self):
super().setUp()
def tearDown(self):
super().tearDown()
def test_show_event(self):
event = mommy.make(Event)
event.save()
event_url = event.get_absolute_url()
c = Client()
response = c.get(event_url)
assert(response.status_code == 200)
def test_show_event_attendee_without_avatar(self):
event = mommy.make(Event)
event.save()
profile = mommy.make(UserProfile, avatar='')
profile.save()
attendee = mommy.make(Attendee, event=event, user=profile, role=Attendee.NORMAL, status= Attendee.YES)
attendee.save()
c = Client()
response = c.get(event.get_absolute_url())
assert(response.status_code == 200)