from django.core import mail

from hc.test import BaseTestCase
from hc.accounts.models import Profile


class LoginTestCase(BaseTestCase):

    def test_it_sends_set_password_link(self):
        self.client.login(username="alice@example.org", password="password")

        form = {"set_password": "1"}
        r = self.client.post("/accounts/profile/", form)
        assert r.status_code == 302

        # profile.token should be set now
        profile = Profile.objects.for_user(self.alice)
        self.assertTrue(len(profile.token) > 10)

        # And an email should have been sent
        self.assertEqual(len(mail.outbox), 1)
        expected_subject = 'Set password on healthchecks.io'
        self.assertEqual(mail.outbox[0].subject, expected_subject)

    def test_it_creates_api_key(self):
        self.client.login(username="alice@example.org", password="password")

        form = {"create_api_key": "1"}
        r = self.client.post("/accounts/profile/", form)
        assert r.status_code == 200

        profile = Profile.objects.for_user(self.alice)
        self.assertTrue(len(profile.api_key) > 10)

    def test_it_revokes_api_key(self):
        self.client.login(username="alice@example.org", password="password")

        form = {"revoke_api_key": "1"}
        r = self.client.post("/accounts/profile/", form)
        assert r.status_code == 200

        profile = Profile.objects.for_user(self.alice)
        self.assertEqual(profile.api_key, "")