import time
from unittest.mock import patch

from django.test import TestCase

from core_images.auth0 import Auth0, session


class TestAuth0(TestCase):
    def test_get_no_token(self):
        with patch.object(Auth0, "_execute_request") as mock:
            self.assertNotIn(Auth0.TOKEN_KEY, session)
            Auth0.get_token()
            self.assertTrue(mock.called)

    def test_get_token_expired(self):
        with patch.object(Auth0, "_execute_request") as mock:
            session[Auth0.TOKEN_KEY] = "test_token"
            session[Auth0.EXPIRE_AT] = time.time() + 60
            Auth0.get_token()
            self.assertTrue(mock.called)

    def test_get_token_not_expired(self):
        with patch.object(Auth0, "_execute_request") as mock:
            session[Auth0.TOKEN_KEY] = "test_token"
            session[Auth0.EXPIRE_AT] = time.time() + 61
            Auth0.get_token()
            self.assertFalse(mock.called)