ojuso-map/apps/map/tests.py

121 lines
4.7 KiB
Python
Raw Normal View History

2018-04-14 02:40:42 +00:00
from django.contrib.auth.models import User
2018-05-01 17:16:09 +00:00
from django.http import QueryDict
from django.test import TestCase
from django.utils.translation import activate
2018-04-14 02:40:42 +00:00
from django.urls import reverse
2018-05-01 17:16:09 +00:00
from .models import CaseStudy, CaseStudyDraft
2018-04-14 02:40:42 +00:00
class CaseStudyDraftAPITests(TestCase):
test_user = 'testuser1'
test_pass = '12345'
def setUp(self):
test_user1 = User.objects.create_user(username=self.test_user, password=self.test_pass)
test_user1.save()
2018-05-01 17:16:09 +00:00
activate('en-gb')
next_url = QueryDict.fromkeys(['next',], reverse('drafts'))
self.login_url = reverse('auth_login') + '?' + next_url.urlencode()
2018-04-14 02:40:42 +00:00
def login(self):
return self.client.login(username=self.test_user, password=self.test_pass)
def test_get_not_logged_in(self):
2018-05-01 17:16:09 +00:00
response = self.client.get(reverse('drafts'), follow=True)
self.assertRedirects(response, self.login_url)
2018-04-14 02:40:42 +00:00
def test_put_not_logged_in(self):
2018-05-01 17:16:09 +00:00
response = self.client.put(reverse('drafts'), follow=True)
self.assertRedirects(response, self.login_url)
2018-04-14 02:40:42 +00:00
def test_delete_not_logged_in(self):
2018-05-01 17:16:09 +00:00
response = self.client.delete(reverse('drafts'), follow=True)
self.assertRedirects(response, self.login_url)
2018-04-14 02:40:42 +00:00
def test_get_and_put(self):
login = self.login()
# No draft should currently exist
response = self.client.get(reverse('drafts'))
self.assertEqual(response.status_code, 404)
self.assertEqual(response.content, b"")
# Putting a draft should succeed
response = self.client.put(reverse('drafts'), '{"test":1}')
self.assertEqual(response.status_code, 201)
# We should get it back
response = self.client.get(reverse('drafts'))
self.assertEqual(response.status_code, 200)
self.assertEqual(response['Content-Type'], "application/json")
self.assertEqual(response.content, b'{"test":1}')
# Updating it should succeed
response = self.client.put(reverse('drafts'), '{"test":2}')
self.assertEqual(response.status_code, 200)
# We should get the update back
response = self.client.get(reverse('drafts'))
self.assertEqual(response.status_code, 200)
self.assertEqual(response['Content-Type'], "application/json")
self.assertEqual(response.content, b'{"test":2}')
def test_delete(self):
login = self.login()
# Create a draft
response = self.client.put(reverse('drafts'), '{"test":1}')
self.assertEqual(response.status_code, 201)
# Delete it again
response = self.client.delete(reverse('drafts'))
self.assertEqual(response.status_code, 204)
# Check it's not there
response = self.client.get(reverse('drafts'))
self.assertEqual(response.status_code, 404)
class CaseStudyTests(TestCase):
def test_get_renewable_generation_detail_when_empty(self):
"""get_renewable_generation_detail() should return the empty string."""
case_study = CaseStudy()
self.assertIs(case_study.get_renewable_generation_detail(), "")
def test_get_renewable_generation_detail_with_geo(self):
"""get_renewable_generation_detail() should return just the description"""
case_study = CaseStudy(generation_technology='GEOT')
self.assertEqual(case_study.get_renewable_generation_detail(), "Geothermal electricity")
def test_get_renewable_generation_detail_with_wind(self):
"""get_renewable_generation_detail() should return the description prefixed with 'wind power'"""
case_study = CaseStudy(generation_technology='SSWE')
2018-05-19 18:21:58 +00:00
self.assertEqual(case_study.get_renewable_generation_detail(), "Wind energy Small-scale (less than 500kW)")
def test_get_renewable_generation_detail_with_other(self):
"""get_renewable_generation_detail() should return the detail provided in .generation_technology_other"""
case_study = CaseStudy(generation_technology='OTHR', generation_technology_other='Warp drive')
self.assertEqual(case_study.get_renewable_generation_detail(), "Warp drive")
2018-05-01 17:16:09 +00:00
def test_get_negative_case_reasons_no_other_1(self):
"""Test with case having no 'other' entry"""
case_study = CaseStudy(negative_case_reasons=['VOLR','ALAB'])
self.assertEqual(case_study.get_negative_case_reasons_no_other(),
[ 'Violation of land rights'
, 'Abusive labour practices'
])
def test_get_negative_case_reasons_no_other_2(self):
"""Test with case having an 'other' entry"""
case_study = CaseStudy(negative_case_reasons=['VOLR','ALAB','OTHR'])
self.assertEqual(case_study.get_negative_case_reasons_no_other(),
[ 'Violation of land rights'
, 'Abusive labour practices'
])