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' ])