ojuso-map/apps/map/tests.py
2018-09-26 15:09:35 +01:00

121 lines
4.7 KiB
Python

from django.contrib.auth.models import User
from django.http import QueryDict
from django.test import TestCase
from django.utils.translation import activate
from django.urls import reverse
from .models import CaseStudy, CaseStudyDraft
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()
activate('en-gb')
next_url = QueryDict.fromkeys(['next',], reverse('drafts'))
self.login_url = reverse('auth_login') + '?' + next_url.urlencode()
def login(self):
return self.client.login(username=self.test_user, password=self.test_pass)
def test_get_not_logged_in(self):
response = self.client.get(reverse('drafts'), follow=True)
self.assertRedirects(response, self.login_url)
def test_put_not_logged_in(self):
response = self.client.put(reverse('drafts'), follow=True)
self.assertRedirects(response, self.login_url)
def test_delete_not_logged_in(self):
response = self.client.delete(reverse('drafts'), follow=True)
self.assertRedirects(response, self.login_url)
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')
self.assertTrue(case_study.get_renewable_generation_detail().startswith("Wind energy"))
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")
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'
])