pfunk.tests.test_jwt

View Source 
from faunadb.errors import PermissionDenied

from pfunk.contrib.auth.collections import Key
from pfunk.tests import User, Group
from pfunk.exceptions import LoginFailed
from pfunk.testcase import APITestCase
from pfunk.contrib.auth.collections import Key


class AuthToken(APITestCase):
    collections = [User, Group]

    def setUp(self) -> None:
        super(AuthToken, self).setUp()
        self.group = Group.create(name='Power Users', slug='power-users')
        self.user = User.create(username='test', email='tlasso@example.org', first_name='Mr. Auth',
                                last_name='Broski', _credentials='abc123', account_status='ACTIVE',
                                groups=[self.group])

    def test_generate_token(self):
        token = Key.create_jwt('secret')
        api_login_token = User.api_login('test', 'abc123')

        self.assertIsNotNone(token)
        self.assertIsNotNone(api_login_token)

    def test_decrypt_token(self):
        claims = {
            'token': 'token',
            'permissions': ['read'],
            'user': {'username': 'test'}
        }
        # `create_jwt` returns both jwt and exp, only get the jwt
        token = Key.create_jwt(claims)[0]
        decrypted_token = Key.decrypt_jwt(encoded=token)

        # the value should be the same
        self.assertEqual(claims, decrypted_token)

    def test_api_login(self):
        token = User.api_login('test', 'abc123')
        self.assertIsNotNone(token)

        with self.assertRaises(LoginFailed):
            User.login('test', 'wrongpass')
#   class AuthToken(pfunk.testcase.APITestCase):
View Source 
class AuthToken(APITestCase):
    collections = [User, Group]

    def setUp(self) -> None:
        super(AuthToken, self).setUp()
        self.group = Group.create(name='Power Users', slug='power-users')
        self.user = User.create(username='test', email='tlasso@example.org', first_name='Mr. Auth',
                                last_name='Broski', _credentials='abc123', account_status='ACTIVE',
                                groups=[self.group])

    def test_generate_token(self):
        token = Key.create_jwt('secret')
        api_login_token = User.api_login('test', 'abc123')

        self.assertIsNotNone(token)
        self.assertIsNotNone(api_login_token)

    def test_decrypt_token(self):
        claims = {
            'token': 'token',
            'permissions': ['read'],
            'user': {'username': 'test'}
        }
        # `create_jwt` returns both jwt and exp, only get the jwt
        token = Key.create_jwt(claims)[0]
        decrypted_token = Key.decrypt_jwt(encoded=token)

        # the value should be the same
        self.assertEqual(claims, decrypted_token)

    def test_api_login(self):
        token = User.api_login('test', 'abc123')
        self.assertIsNotNone(token)

        with self.assertRaises(LoginFailed):
            User.login('test', 'wrongpass')

A class whose instances are single test cases.

By default, the test code itself should be placed in a method named 'runTest'.

If the fixture may be used for many test cases, create as many test methods as are needed. When instantiating such a TestCase subclass, specify in the constructor arguments the name of the test method that the instance is to execute.

Test authors should subclass TestCase for their own tests. Construction and deconstruction of the test's environment ('fixture') can be implemented by overriding the 'setUp' and 'tearDown' methods respectively.

If it is necessary to override the __init__ method, the base class __init__ method must always be called. It is important that subclasses should not change the signature of their __init__ method, since instances of the classes are instantiated automatically by parts of the framework in order to be run.

When subclassing TestCase, you can set these attributes:

  • failureException: determines which exception will be raised when the instance's assertion methods fail; test methods raising this exception will be deemed to have 'failed' rather than 'errored'.
  • longMessage: determines whether long messages (including repr of objects used in assert methods) will be printed on failure in addition to any explicit message passed.
  • maxDiff: sets the maximum length of a diff in failure messages by assert methods using difflib. It is looked up as an instance attribute so can be configured by individual tests if required.
#   def setUp(self) -> None:
View Source 
    def setUp(self) -> None:
        super(AuthToken, self).setUp()
        self.group = Group.create(name='Power Users', slug='power-users')
        self.user = User.create(username='test', email='tlasso@example.org', first_name='Mr. Auth',
                                last_name='Broski', _credentials='abc123', account_status='ACTIVE',
                                groups=[self.group])

Hook method for setting up the test fixture before exercising it.

#   def test_generate_token(self):
View Source 
    def test_generate_token(self):
        token = Key.create_jwt('secret')
        api_login_token = User.api_login('test', 'abc123')

        self.assertIsNotNone(token)
        self.assertIsNotNone(api_login_token)
#   def test_decrypt_token(self):
View Source 
    def test_decrypt_token(self):
        claims = {
            'token': 'token',
            'permissions': ['read'],
            'user': {'username': 'test'}
        }
        # `create_jwt` returns both jwt and exp, only get the jwt
        token = Key.create_jwt(claims)[0]
        decrypted_token = Key.decrypt_jwt(encoded=token)

        # the value should be the same
        self.assertEqual(claims, decrypted_token)
#   def test_api_login(self):
View Source 
    def test_api_login(self):
        token = User.api_login('test', 'abc123')
        self.assertIsNotNone(token)

        with self.assertRaises(LoginFailed):
            User.login('test', 'wrongpass')
Inherited Members
unittest.case.TestCase
TestCase
failureException
longMessage
maxDiff
addTypeEqualityFunc
addCleanup
addClassCleanup
setUpClass
tearDownClass
countTestCases
defaultTestResult
shortDescription
id
subTest
run
doCleanups
doClassCleanups
debug
skipTest
fail
assertFalse
assertTrue
assertRaises
assertWarns
assertLogs
assertEqual
assertNotEqual
assertAlmostEqual
assertNotAlmostEqual
assertSequenceEqual
assertListEqual
assertTupleEqual
assertSetEqual
assertIn
assertNotIn
assertIs
assertIsNot
assertDictEqual
assertDictContainsSubset
assertCountEqual
assertMultiLineEqual
assertLess
assertLessEqual
assertGreater
assertGreaterEqual
assertIsNone
assertIsNotNone
assertIsInstance
assertNotIsInstance
assertRaisesRegex
assertWarnsRegex
assertRegex
assertNotRegex
failUnlessRaises
failIf
assertRaisesRegexp
assertRegexpMatches
assertNotRegexpMatches
failUnlessEqual
assertEquals
failIfEqual
assertNotEquals
failUnlessAlmostEqual
assertAlmostEquals
failIfAlmostEqual
assertNotAlmostEquals
failUnless
assert_
pfunk.testcase.APITestCase
tearDown