def setUp(self): default_site = Site.objects.get(is_default_site=True) self.test_setting = TestSetting() self.test_setting.title = 'Site title' self.test_setting.email = '*****@*****.**' self.test_setting.site = default_site self.test_setting.save() self.login()
class TestSettingEditView(TestCase, WagtailTestUtils): def setUp(self): default_site = Site.objects.get(is_default_site=True) self.test_setting = TestSetting() self.test_setting.title = 'Site title' self.test_setting.email = '*****@*****.**' self.test_setting.site = default_site self.test_setting.save() self.login() def get(self, params={}): return self.client.get( reverse('wagtailsettings_edit', args=('tests', 'testsetting')), params) def post(self, post_data={}): return self.client.post( reverse('wagtailsettings_edit', args=('tests', 'testsetting')), post_data) def test_status_code(self): self.assertEqual(self.get().status_code, 200) def test_non_existant_model(self): response = self.client.get( reverse('wagtailsettings_edit', args=('tests', 'foo'))) self.assertEqual(response.status_code, 404) def test_edit_invalid(self): response = self.post(post_data={'foo': 'bar'}) self.assertContains(response, "The setting could not be saved due to errors.") self.assertContains(response, "This field is required.") def test_edit(self): response = self.post(post_data={'title': 'Edited site title', 'email': '*****@*****.**'}) self.assertEqual(response.status_code, 302) default_site = Site.objects.get(is_default_site=True) setting = TestSetting.objects.get(site=default_site) self.assertEqual(setting.title, 'Edited site title') self.assertEqual(setting.email, '*****@*****.**')