epcstages/stages/tests.py

43 lines
1.5 KiB
Python

# -*- encoding: utf-8 -*-
from __future__ import unicode_literals
from django.contrib.auth.models import User
from django.core.urlresolvers import reverse
from django.test import TestCase
from .models import Period, Student, Availability, Referent
class StagesTest(TestCase):
fixtures = ['test_fixture.json']
def setUp(self):
self.admin = User.objects.create_user('me', 'me@example.org', 'mepassword')
self.client.login(username='me', password='mepassword')
def test_export(self):
response1 = self.client.get(reverse('stages_export'))
self.assertEqual(response1.status_code, 200)
response2 = self.client.get(reverse('stages_export'), {'filter': '2'})
self.assertEqual(response2.status_code, 200)
self.assertGreater(len(response1.content), len(response2.content))
def test_new_training(self):
student = Student.objects.get(last_name='Varrin')
avail = Availability.objects.get(pk=2)
response = self.client.post(reverse('new_training'),
{'student': student.pk,
'avail': avail.pk,
'referent': 1})
self.assertEqual(response.content, b'OK')
avail = Availability.objects.get(pk=2)
self.assertEqual(avail.training.student, student)
def test_period_schoolyear(self):
per = Period.objects.get(pk=1)
self.assertEqual(per.school_year, "2012 — 2013")
def test_period_weeks(self):
per = Period.objects.get(pk=1)
self.assertEqual(per.weeks, 1)