def create_organizer_set(auto_create_user_set=False):
"""Create organizers set to test.
Args:
auto_create_user_set -- flag tindicating that create_user_set function must be executed
Precondition:
create_user_set
Postcondition:
Two Organizers created associated to users organizer01, and organizer02.
Not events association.
"""
if auto_create_user_set:
create_user_set()
set_current_user(User.objects.get(username="******"))
Organizer.objects.bulk_create([
Organizer(user=User.objects.get(username="******"),
first_name="Organizer01"),
Organizer(user=User.objects.get(username="******"),
first_name="Organizer02"),
Organizer(user=User.objects.get(username="******"),
first_name="Organizer03")
])
def test_cant_create_sponsor_category_not_event_organizer(self):
set_current_user(User.objects.filter(username='******').first())
event = Event.objects.filter(name='MyTest02').first()
url = reverse('event_create_sponsor_category', kwargs={'pk': event.pk})
data = {'name': 'Oro', 'amount': '10000'}
self.client.login(username='******', password='******')
response = self.client.post(url, data)
self.assertContainsMessage(response, MUST_BE_EVENT_ORGANIZAER_MESSAGE)
self.assertFalse(SponsorCategory.objects.filter(name='Oro').exists())
def test_create_sponsor_category_by_event_organizer(self):
set_current_user(User.objects.filter(username='******').first())
event = Event.objects.filter(name='MyTest01').first()
url = reverse('event_create_sponsor_category', kwargs={'pk': event.pk})
data = {'name': 'Oro', 'amount': '50000'}
self.client.login(username='******', password='******')
self.client.post(url, data)
self.assertTrue(SponsorCategory.objects.filter(name='Oro').exists())
def test_cant_duplicate_sponsor_category(self):
set_current_user(User.objects.filter(username='******').first())
event = Event.objects.filter(name='MyTest01').first()
url = reverse('event_create_sponsor_category', kwargs={'pk': event.pk})
data = {'name': 'Gold', 'amount': '10000'}
self.client.login(username='******', password='******')
response = self.client.post(url, data)
expected_url = reverse('event_detail', kwargs={'pk': event.pk})
self.assertRedirects(response, expected_url)
self.assertContainsMessage(response,
DUPLICATED_SPONSOR_CATEGORY_MESSAGE)
def create_event_set(user):
"""Create events set to test.
Args:
user -- existing User
Postcondition:
Two Events created by user with: comission 10 and 20.
Gold and Silver SponsorCategory associated to events
"""
set_current_user(user)
event01 = Event.objects.create(name='MyTest01', commission=10)
Event.objects.create(name='MyTest02', commission=20)
SponsorCategory.objects.create(name='Gold', amount=10000, event=event01)
SponsorCategory.objects.create(name='Silver', amount=1000, event=event01)
