# -*- coding: utf-8 -*- import datetime from django.contrib.auth.models import User from django.core.management import call_command from django.test import Client as HttpClient, override_settings, TestCase from django.utils import timezone from faker import Factory class FormsTestCase(TestCase): @classmethod def setUpClass(cls): super(FormsTestCase, cls).setUpClass() fake = Factory.create() call_command('migrate', verbosity=0) call_command('fake', verbosity=0) cls.c = HttpClient() fake_user = fake.simple_profile() cls.credentials = { 'username': fake_user['username'], 'password': fake.password() } cls.user = User.objects.create_user( is_superuser=True, **cls.credentials) cls.settings_template = { 'first_name': 'User', 'last_name': 'Name', 'email': 'user@user.user', 'dashboard_refresh_rate': '', 'language': 'en', 'timezone': 'UTC', 'next': '/user/settings/' } def test_change_password(self): self.c.login(**self.credentials) page = self.c.get('/user/password/') self.assertEqual(page.status_code, 200) params = { 'old_password': 'wrong', 'new_password1': 'mynewpassword', 'new_password2': 'notmynewpassword' } page = self.c.post('/user/password/', params) self.assertEqual(page.status_code, 200) self.assertFormError(page, 'form', 'old_password', 'Your old password was entered incorrectly. ' 'Please enter it again.') params['old_password'] = self.credentials['password'] page = self.c.post('/user/password/', params) self.assertEqual(page.status_code, 200) self.assertFormError(page, 'form', 'new_password2', "The two password fields didn’t match.") params['new_password2'] = 'mynewpassword' page = self.c.post('/user/password/', params) self.assertEqual(page.status_code, 200) def test_user_forms(self): self.user.is_staff = True self.user.save() self.c.login(**self.credentials) params = { 'username': 'username', 'first_name': 'User', 'last_name': 'Name', 'email': 'user@user.user', 'password1': 'd47o8dD&#hu3ulu3', 'password2': 'd47o8dD&#hu3ulu3' } page = self.c.post('/users/add/', params) self.assertEqual(page.status_code, 302) new_user = User.objects.get(username='username') self.assertIsInstance(new_user, User) params['first_name'] = 'Changed' page = self.c.post('/users/{}/edit/'.format(new_user.id), params) self.assertEqual(page.status_code, 302) new_user.refresh_from_db() self.assertEqual(new_user.first_name, params['first_name']) page = self.c.post('/users/{}/delete/'.format(new_user.id)) self.assertEqual(page.status_code, 302) self.assertQuerysetEqual(User.objects.filter(username='username'), []) def test_user_settings(self): self.c.login(**self.credentials) params = self.settings_template.copy() params['first_name'] = 'New First Name' page = self.c.post('/user/settings/', params, follow=True) self.assertEqual(page.status_code, 200) self.assertContains(page, 'New First Name') def test_user_regenerate_api_key(self): self.c.login(**self.credentials) api_key_before = User.objects.get(pk=self.user.id).settings.api_key() params = self.settings_template.copy() params['api_key_regenerate'] = 'Regenerate' page = self.c.post('/user/settings/', params, follow=True) self.assertEqual(page.status_code, 200) self.assertNotEqual( api_key_before, User.objects.get(pk=self.user.id).settings.api_key() ) def test_user_settings_invalid(self): self.c.login(**self.credentials) params = self.settings_template.copy() params['email'] = 'Not an email address' page = self.c.post('/user/settings/', params) self.assertEqual(page.status_code, 200) self.assertFormError(page, 'user_form', 'email', 'Enter a valid email address.') def test_user_settings_language(self): self.c.login(**self.credentials) params = self.settings_template.copy() params['language'] = 'fr' page = self.c.post('/user/settings/', data=params, follow=True) self.assertContains(page, 'Paramètres Utilisateur') @override_settings(TIME_ZONE='US/Eastern') def test_user_settings_timezone(self): self.c.login(**self.credentials) self.assertEqual(timezone.get_default_timezone_name(), 'US/Eastern') params = self.settings_template.copy() params['timezone'] = 'US/Pacific' page = self.c.post('/user/settings/', data=params, follow=True) self.assertEqual(page.status_code, 200) self.assertEqual(timezone.get_current_timezone_name(), params['timezone']) def test_user_settings_dashboard_hide_empty_on(self): self.c.login(**self.credentials) params = self.settings_template.copy() params['dashboard_hide_empty'] = 'on' page = self.c.post('/user/settings/', data=params, follow=True) self.assertEqual(page.status_code, 200) self.user.refresh_from_db() self.assertTrue(self.user.settings.dashboard_hide_empty) def test_user_settings_dashboard_refresh_rate(self): self.c.login(**self.credentials) params = self.settings_template.copy() params['dashboard_refresh_rate'] = '0:05:00' page = self.c.post('/user/settings/', data=params, follow=True) self.assertEqual(page.status_code, 200) self.user.refresh_from_db() self.assertEqual(self.user.settings.dashboard_refresh_rate, datetime.timedelta(seconds=300)) def test_user_settings_dashboard_hide_age(self): self.c.login(**self.credentials) params = self.settings_template.copy() params['dashboard_hide_age'] = '1 day, 0:00:00' page = self.c.post('/user/settings/', data=params, follow=True) self.assertEqual(page.status_code, 200) self.user.refresh_from_db() self.assertEqual(self.user.settings.dashboard_hide_age, datetime.timedelta(days=1))