tests.py 920 B

123456789101112131415161718192021
  1. from django.contrib.auth.models import User
  2. from django.core.urlresolvers import reverse
  3. from django.test import TestCase
  4. from django.contrib.contenttypes.models import ContentType
  5. from adhesions.models import Adherent
  6. from .models import Service, ServiceType
  7. class ServicesTests(TestCase):
  8. def setUp(self):
  9. user = User.objects.create_user('user', email='user@example.net', password='user')
  10. adh = Adherent.objects.create(adherent_type=ContentType.objects.get_for_model(user), adherent_id=user.pk)
  11. stype = ServiceType.objects.create(name='VM')
  12. s1 = Service.objects.create(adherent=adh, service_type=stype, label='Service 1')
  13. def test_service_list(self):
  14. self.client.login(username='user', password='user')
  15. response = self.client.get(reverse('service-list'))
  16. self.assertEqual(response.status_code, 200)
  17. self.assertContains(response, 'Service 1')