trentpalmerdotorg/accounts/tests/test_edit_profile_view.py

from django.test import TestCase
from django.contrib.auth.models import User
from accounts.models import Account
from django.urls import reverse


class EditProfileViewTestCase(TestCase):

    def setUp(self):
        user_a = User(email='user_a@example.com', username='user_a')
        user_a.set_password('password123456donkey')
        user_a.save()
        Account.objects.create(user=user_a)

    def test_edit_profile_view_no_login(self):
        response = self.client.post(reverse('accounts:edit_profile'), {
            'email': 'user_a@example.com',
            'first_name': 'User_a',
            'last_name': 'Smith',
            'twitter_handle': '@user_a',
            'password': 'password123456donkey',
        }, follow=True)
        self.assertEquals(response.status_code, 200)
        self.assertTemplateUsed(response, 'audio/index.html')
        user_a = User.objects.get(username='user_a')
        self.assertEquals(user_a.first_name, '')
        self.assertEquals(response.request['PATH_INFO'], '/')

    def test_edit_profile_view_no_data(self):
        self.client.login(username='user_a', password='password123456donkey')
        response = self.client.post(reverse('accounts:edit_profile'), follow=True)
        self.assertEquals(response.status_code, 200)
        self.assertTemplateUsed(response, 'base_form.html')
        user_a = User.objects.get(username='user_a')
        self.assertEquals(user_a.first_name, '')
        self.assertEquals(response.request['PATH_INFO'], '/accounts/edit-profile/')
        self.assertEquals(response.content.decode('utf-8').count("This field is required."), 3)

    def test_edit_profile_view_invalid_email(self):
        self.client.login(username='user_a', password='password123456donkey')
        response = self.client.post(reverse('accounts:edit_profile'), {
            'email': 'user_afoobar.com',
            'first_name': 'User_a',
            'last_name': 'Smith',
            'twitter_handle': '@user_a',
            'password': 'password123456donkey',
        }, follow=True)
        self.assertEquals(response.status_code, 200)
        self.assertTemplateUsed(response, 'base_form.html')
        user_a = User.objects.get(username='user_a')
        self.assertEquals(user_a.first_name, '')
        self.assertEquals(user_a.account.twitter_handle, '@Twitter')
        self.assertEquals(response.request['PATH_INFO'], '/accounts/edit-profile/')

    def test_edit_profile_view_invalid_twitter_handle(self):
        self.client.login(username='user_a', password='password123456donkey')
        response = self.client.post(reverse('accounts:edit_profile'), {
            'email': 'user_a@example.com',
            'first_name': 'User_a',
            'last_name': 'Smith',
            'twitter_handle': 'user_a',
            'password': 'password123456donkey',
        }, follow=True)
        self.assertEquals(response.status_code, 200)
        self.assertTemplateUsed(response, 'base_form.html')
        user_a = User.objects.get(username='user_a')
        self.assertEquals(user_a.first_name, '')
        self.assertEquals(user_a.account.twitter_handle, '@Twitter')
        self.assertEquals(response.request['PATH_INFO'], '/accounts/edit-profile/')
        t_warning = "twitter_handle must begin with "@""
        self.assertTrue(t_warning in response.content.decode('utf-8'))

    def test_edit_profile_view(self):
        self.client.login(username='user_a', password='password123456donkey')
        response = self.client.post(reverse('accounts:edit_profile'), {
            'email': 'user_a@example.com',
            'first_name': 'User_a',
            'last_name': 'Smith',
            'twitter_handle': '@user_a',
            'password': 'password123456donkey',
        }, follow=True)
        self.assertEquals(response.status_code, 200)
        self.assertTemplateUsed(response, 'audio/index.html')
        user_a = User.objects.get(username='user_a')
        self.assertEquals(user_a.first_name, 'User_a')
        self.assertEquals(user_a.last_name, 'Smith')
        self.assertEquals(user_a.account.twitter_handle, '@user_a')
        self.assertEquals(response.request['PATH_INFO'], '/')