# -*- coding: utf-8 -*-
try:
    from unittest.mock import patch
except ImportError:
    from mock import patch

from django.contrib import messages
from django.contrib.auth import get_user_model
from django.contrib.messages.api import get_messages
from django.contrib.messages.storage.base import Message
from django.test import TestCase, override_settings

from .cas_clients import AcceptCASClient

User = get_user_model()


@patch('allauth_cas.views.cas.CASClient', AcceptCASClient)
def client_cas_login(client):
    """
    Sign in client through the example CAS provider.
    Returns the response of callbacK view.
    """
    r = client.get('/accounts/theid/login/callback/', {
        'ticket': 'fake-ticket',
    })
    return r


class LogoutFlowTests(TestCase):
    expected_msg_str = (
        "To logout of CAS, please close your browser, or visit this <a "
        "href=\"/accounts/theid/logout/?next=%2F\">link</a>."
    )

    def setUp(self):
        client_cas_login(self.client)

    def assertCASLogoutNotInMessages(self, response):
        r_messages = get_messages(response.wsgi_request)
        self.assertNotIn(
            self.expected_msg_str,
            (str(msg) for msg in r_messages),
        )
        self.assertTemplateNotUsed(
            response,
            'cas_account/messages/logged_out.txt',
        )

    @override_settings(SOCIALACCOUNT_PROVIDERS={
        'theid': {
            'MESSAGE_ON_LOGOUT': True,
            'MESSAGE_ON_LOGOUT_LEVEL': messages.WARNING,
        },
    })
    def test_message_on_logout(self):
        """
        Message is sent to propose user to logout of CAS.
        """
        r = self.client.post('/accounts/logout/')
        r_messages = get_messages(r.wsgi_request)

        expected_msg = Message(messages.WARNING, self.expected_msg_str)

        self.assertIn(expected_msg, r_messages)
        self.assertTemplateUsed(r, 'cas_account/messages/logged_out.txt')

    @override_settings(SOCIALACCOUNT_PROVIDERS={
        'theid': {
            'MESSAGE_ON_LOGOUT': False,
        },
    })
    def test_message_on_logout_disabled(self):
        """
        The logout message can be disabled in settings.
        """
        r = self.client.post('/accounts/logout/')
        self.assertCASLogoutNotInMessages(r)

    @override_settings(SOCIALACCOUNT_PROVIDERS={
        'theid': {'MESSAGE_ON_LOGOUT': True},
    })
    def test_default_logout(self):
        """
        The CAS logout message doesn't appear with other login methods.
        """
        User.objects.create_user('user', '', 'user')
        self.client.login(username='user', password='user')

        r = self.client.post('/accounts/logout/')
        self.assertCASLogoutNotInMessages(r)