import datetime

from django.test import TestCase
from django.utils import timezone
from django.utils import six

from oidc_provider.lib.utils.common import get_issuer, to_timestamp


class Request(object):
    """
    Mock request object.
    """
    scheme = 'http'

    def get_host(self):
        return 'host-from-request:8888'


class CommonTest(TestCase):
    """
    Test cases for common utils.
    """
    def test_get_issuer(self):
        request = Request()

        # from default settings
        self.assertEqual(get_issuer(),
                         'http://localhost:8000/openid')

        # from custom settings
        with self.settings(SITE_URL='http://otherhost:8000'):
            self.assertEqual(get_issuer(),
                             'http://otherhost:8000/openid')

        # `SITE_URL` not set, from `request`
        with self.settings(SITE_URL=''):
            self.assertEqual(get_issuer(request=request),
                             'http://host-from-request:8888/openid')

        # use settings first if both are provided
        self.assertEqual(get_issuer(request=request),
                         'http://localhost:8000/openid')

        # `site_url` can even be overridden manually
        self.assertEqual(get_issuer(site_url='http://127.0.0.1:9000',
                                    request=request),
                         'http://127.0.0.1:9000/openid')

    def test_to_timestamp(self):
        if not six.PY2:
            naive_dt = datetime.datetime.now()
            self.assertEqual(to_timestamp(naive_dt), int(naive_dt.timestamp()))

        aware_dt = datetime.datetime(2016, 3, 2, 14, 2, 6, 123, timezone.utc)
        self.assertEqual(to_timestamp(aware_dt), 1456927326)
        if not six.PY2:
            self.assertEqual(to_timestamp(aware_dt), int(aware_dt.timestamp()))