121 lines
4.7 KiB
Python
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'
|
|
])
|