from django.test import TestCase, Client
from django.core.urlresolvers import reverse
from django.utils.translation import ugettext_lazy as _
from rolepermissions.shortcuts import assign_role
from users.models import User
from django.core import mail

class LoginTestCase(TestCase):

    def setUp(self):
        self.client = Client()

        self.user = User.objects.create_user(
            username = 'test',
            email = 'testing@amadeus.com',
            is_staff = True,
            is_active = True,
            password = 'testing'
        )
        assign_role(self.user, 'system_admin')

        self.url = reverse('core:home')

    def test_ok(self):
        response = self.client.get(self.url)
        self.assertEquals(response.status_code, 200)
        self.assertTemplateUsed(response, 'index.html')
        data = {'username': 'test', 'password': 'testing'}
        response = self.client.post(self.url, data)
        self.assertRedirects(response, reverse("app:index"))

    # def test_not_ok(self):
    #     response = self.client.get(self.url)
    #     self.assertEquals(response.status_code, 200)
    #     self.assertTemplateUsed(response, 'index.html')
    #     data = {'username': 'test', 'password': 'senhaerrada'}
    #     response = self.client.post(self.url, data)
    #     self.assertTrue('message' in response.context)
    #     self.assertEquals(response.context['message'], "Email ou senha incorretos!")

class RegisterUserTestCase(TestCase):

    def setUp(self):
        self.client = Client()
        self.url = reverse('core:register')
        self.data = {
                'username': 'testeamadeus',
                'birth_date': '12/12/2000',
                'email': 'teste@amadeus.com',
                'password': 'aminhasenha1',
                'password2': 'aminhasenha1',
                'cpf': '705.089.884-89',
                'name': 'Teste Amadeus',
                'city': 'Praia',
                'state': 'PE',
                'gender': 'F',
            }

    def test_register_ok(self):

        response = self.client.post(self.url, self.data, follow=True)
        self.assertRedirects(response, reverse('core:home'))
        self.assertEqual(response.status_code, 200)
        self.assertEqual(User.objects.count(), 1)

    def test_register_error(self):
        response = self.client.get(self.url)
        self.assertEquals(response.status_code, 200)

        data = {
                'username': 'testeamadeus',
                'birth_date': '12/12/2000',
                'email': 'teste@amadeus.com',
                'password': 'aminhasenha1',
                'password2': 'aminhasenha',
                'name': 'Teste Amadeus',
                'city': 'Praia',
                'state': 'PE',
                'gender': 'F',
            }
        response = self.client.post(self.url, data)
        self.assertFormError(response, 'form', 'password2', 'The confirmation password is incorrect.')

        data = {
                'username': 'testeamadeus',
                'email': 'teste.amadeus.com',
                'birth_date': '12/12/2000',
                'password': 'aminhasenha1',
                'password2': 'aminhasenha',
                'name': 'Teste Amadeus',
                'city': 'Praia',
                'state': 'PE',
                'gender': 'F',
            }
        
        response = self.client.post(self.url, data)
        self.assertFormError(response, 'form', 'email', _('Enter a valid email address.'))

        data = {
                'username': '',
                'email': 'teste@amadeus.com',
                'birth_date': '12/12/2000',
                'password': 'aminhasenha1',
                'password2': 'aminhasenha',
                'name': 'Teste Amadeus',
                'city': 'Praia',
                'state': 'PE',
                'gender': 'F',
            }
        response = self.client.post(self.url, data)
        self.assertFormError(response, 'form', 'username', _('This field is required.'))

class RememberPasswordTestCase(TestCase):

    def setUp(self):
        self.client = Client()
        self.url = reverse('core:remember_password')

    def test_remember_password_ok(self):
        response = self.client.get(self.url)
        self.assertEquals(response.status_code, 200)
        self.assertTemplateUsed(response, 'remember_password.html')
        data = {'email': 'fulano@fulano.com', 'registration': '0124578964226'}
        response = self.client.post(self.url, data)
        self.assertEquals(response.status_code, 200)
        self.assertEquals(len(mail.outbox), 1)
        self.assertTrue('success' in response.context)
        self.assertTrue('danger' not in response.context)

    def test_remember_password_error(self):
        data = {'email': 'fulano@fulano.com','registration':''}
        response = self.client.post(self.url, data)
        self.assertEquals(response.status_code, 200)
        self.assertEquals(len(mail.outbox), 0)
        self.assertTrue('success' not in response.context)
        self.assertTrue('danger' in response.context)

        data = {'email': '', 'registration': '0124578964226'}
        response = self.client.post(self.url, data)
        self.assertEquals(response.status_code, 200)
        self.assertEquals(len(mail.outbox), 0)
        self.assertTrue('success' not in response.context)
        self.assertTrue('danger' in response.context)


class UpdateUserTestCase(TestCase):

    def setUp(self):
        self.client = Client()

        self.user = User.objects.create_user(
            username = 'test',
            email = 'testing@amadeus.com',
            is_staff = False,
            is_active = True,
            password = 'testing1'
        )

        assign_role(self.user, 'student')

        self.url = reverse('users:update_profile')

    def test_update_ok(self):
        #LOGGING USER TO TEST
        data = {'username': 'test', 'password': 'testing1'}
        response = self.client.post(reverse('core:home'), data)
        self.assertRedirects(response, reverse('app:index'))

        data={
                'username': 'test',
                'birth_date': '12/12/2000',
                'email': 'testing@amadeus.com',
                'cpf': '705.089.884-89',
                'name': 'Teste Amadeus',
                'city': 'Praia',
                'state': 'PE',
                'gender': 'F',
            }
        response = self.client.get(self.url)
        self.assertEqual(response.status_code, 200)
        response = self.client.post(self.url, data)
        self.assertRedirects(response, reverse('users:profile'))

    def test_update_error(self):
        
        #LOGING USER TO TEST
        data = {'username': 'test', 'password': 'testing1'}
        response = self.client.post(reverse('core:home'), data)
        self.assertRedirects(response, reverse('app:index'))

        response = self.client.get(self.url)
        self.assertEquals(response.status_code, 200)

        data = {
                'username': '',
                'email': 'teste@amadeus.com',
                'name': 'Teste Amadeus',
                'birth_date': '12/12/2000',
                'city': 'Praia',
                'state': 'PE',
                'gender': 'F',
            }
        response = self.client.post(self.url, data)
        self.assertFormError(response, 'form', 'username', _('This field is required.'))
        

class DeleteUserTestCase(TestCase):
    def setUp(self):
        self.client = Client()

        self.user = User.objects.create_user(
            username = 'test',
            email = 'testing@amadeus.com',
            is_staff = True,
            is_active = True,
            password = 'testing'
        )

        assign_role(self.user, 'student')
        self.url = reverse('core:home')

    def tearDown(test):
        User.objects.get(email='testing@amadeus.com').delete()

    def test_delete_ok(self):
        pass