• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• django-rest-framework-simplejwt-master
  • .circleci
    • merge_pr.sh
    • config.yml
  • Makefile
  • .coveragerc
  • licenses
    • LICENSE-django-rest-framework-jwt
    • LICENSE-django-rest-framework.md
  • codecov.yml
  • CHANGELOG.md
  • README.rst
  • setup.py
  • rest_framework_simplejwt
    • state.py
    • token_blacklist
      • apps.py
      • models.py
      • __init__.py
      • migrations
        • 0004_auto_20171017_2013.py
        • 0007_auto_20171017_2214.py
        • 0001_initial.py
        • 0006_auto_20171017_2113.py
        • 0002_outstandingtoken_jti_hex.py
        • __init__.py
        • 0005_remove_outstandingtoken_jti.py
        • 0003_auto_20171017_2007.py
      • admin.py
      • management
        • commands
          • flushexpiredtokens.py
          • __init__.py
        • __init__.py
    • exceptions.py
    • tokens.py
    • backends.py
    • authentication.py
    • serializers.py
    • models.py
    • views.py
    • locale
      • de_CH
        • LC_MESSAGES
          • django.mo
          • django.po
      • pl_PL
        • LC_MESSAGES
          • django.mo
          • django.po
      • pt_BR
        • LC_MESSAGES
          • django.mo
          • django.po
      • cs
        • LC_MESSAGES
          • django.mo
          • django.po
      • es_CL
        • LC_MESSAGES
          • django.mo
          • django.po
      • ru_RU
        • LC_MESSAGES
          • django.mo
          • django.po
      • fa_IR
        • LC_MESSAGES
          • django.mo
          • django.po
      • es_AR
        • LC_MESSAGES
          • django.mo
          • django.po
    • __init__.py
    • utils.py
    • settings.py
    • compat.py
  • setup.cfg
  • tests
    • test_utils.py
    • test_token_blacklist.py
    • test_serializers.py
    • models.py
    • views.py
    • test_models.py
    • urls.py
    • test_backends.py
    • test_integration.py
    • __init__.py
    • utils.py
    • test_authentication.py
    • test_tokens.py
    • test_views.py
    • conftest.py
  • pytest.ini
  • .isort.cfg
  • .gitignore
  • docs
    • Makefile
    • experimental_features.rst
    • creating_tokens_manually.rst
    • development_and_contributing.rst
    • getting_started.rst
    • customizing_token_claims.rst
    • rest_framework_simplejwt.rst
    • settings.rst
    • token_types.rst
    • index.rst
    • conf.py
    • blacklist_app.rst
  • LICENSE.txt
  • MANIFEST.in
  • tox.ini

from datetime import datetime, timedelta
from unittest.mock import patch

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

from rest_framework_simplejwt.utils import (
    aware_utcnow, datetime_from_epoch, datetime_to_epoch, format_lazy,
    make_utc,
)


class TestMakeUtc(TestCase):
    def test_it_should_return_the_correct_values(self):
        # It should make a naive datetime into an aware, utc datetime if django
        # is configured to use timezones and the datetime doesn't already have
        # a timezone

        # Naive datetime
        dt = datetime(year=1970, month=12, day=1)

        with self.settings(USE_TZ=False):
            dt = make_utc(dt)
            self.assertTrue(timezone.is_naive(dt))

        with self.settings(USE_TZ=True):
            dt = make_utc(dt)
            self.assertTrue(timezone.is_aware(dt))
            self.assertEqual(dt.utcoffset(), timedelta(seconds=0))


class TestAwareUtcnow(TestCase):
    def test_it_should_return_the_correct_value(self):
        now = datetime.utcnow()

        with patch('rest_framework_simplejwt.utils.datetime') as fake_datetime:
            fake_datetime.utcnow.return_value = now

            # Should return aware utcnow if USE_TZ == True
            with self.settings(USE_TZ=True):
                self.assertEqual(timezone.make_aware(now, timezone=timezone.utc), aware_utcnow())

            # Should return naive utcnow if USE_TZ == False
            with self.settings(USE_TZ=False):
                self.assertEqual(now, aware_utcnow())


class TestDatetimeToEpoch(TestCase):
    def test_it_should_return_the_correct_values(self):
        self.assertEqual(datetime_to_epoch(datetime(year=1970, month=1, day=1)), 0)
        self.assertEqual(datetime_to_epoch(datetime(year=1970, month=1, day=1, second=1)), 1)
        self.assertEqual(datetime_to_epoch(datetime(year=2000, month=1, day=1)), 946684800)


class TestDatetimeFromEpoch(TestCase):
    def test_it_should_return_the_correct_values(self):
        with self.settings(USE_TZ=False):
            self.assertEqual(datetime_from_epoch(0), datetime(year=1970, month=1, day=1))
            self.assertEqual(datetime_from_epoch(1), datetime(year=1970, month=1, day=1, second=1))
            self.assertEqual(datetime_from_epoch(946684800), datetime(year=2000, month=1, day=1), 946684800)

        with self.settings(USE_TZ=True):
            self.assertEqual(datetime_from_epoch(0), make_utc(datetime(year=1970, month=1, day=1)))
            self.assertEqual(datetime_from_epoch(1), make_utc(datetime(year=1970, month=1, day=1, second=1)))
            self.assertEqual(datetime_from_epoch(946684800), make_utc(datetime(year=2000, month=1, day=1)))


class TestFormatLazy(TestCase):
    def test_it_should_work(self):
        obj = format_lazy('{} {}', 'arst', 'zxcv')

        self.assertNotIsInstance(obj, str)
        self.assertEqual(str(obj), 'arst zxcv')