pfunk.tests.test_web_forgot_password

View Source
from werkzeug.test import Client

from pfunk.tests import User, Group
from pfunk.testcase import APITestCase


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

    def setUp(self) -> None:
        super(TestWebForgotPassword, self).setUp()
        self.group = Group.create(name='Power Users', slug='power-users')
        self.user = User.create(username='test', email='tlasso@example.org', first_name='Ted',
                                last_name='Lasso', _credentials='abc123', account_status='ACTIVE',
                                groups=[self.group])
        self.app = self.project.wsgi_app
        self.c = Client(self.app)
        self.token, self.exp = User.api_login("test", "abc123")
        self.user.attach_forgot_verification_key()
        self.key = self.user.forgot_password_key

    def test_send_forgot_req(self):
        res = self.c.post(f'/user/forgot-password/',
                          json={"email": "tlasso@example.org"},
                          headers={
                              "Content-Type": "application/json"})

        self.assertTrue(res.json['success'])
    
    def test_submit_key_for_forgot_pass(self):
        """ Submits the key from the forgot password email to initiate password reset """
        
        res = self.c.put(f'/user/forgot-password/',
                          json={
                              "verification_key": self.key,
                              "password": "new_updated_pass"},
                          headers={
                              "Content-Type": "application/json"})

        new_login = User.api_login("test", "new_updated_pass")
        self.assertTrue(res.json['success'])
        self.assertIsNotNone(new_login)

    def test_submit_wrong_key_for_forgot_pass(self):
        """ Submit a wrong key for verification of reset password. Should return `Not Found` """
        key = 'wrong-key'
        res = self.c.put(f'/user/forgot-password/',
                          json={
                              "verification_key": key,
                              "password": "forgotten_password"},
                          headers={
                              "Content-Type": "application/json"})
        expected = {'data': 'Not Found', 'success': False}

        self.assertFalse(res.json['success'])
        self.assertDictEqual(res.json, expected)
#   class TestWebForgotPassword(pfunk.testcase.APITestCase):
View Source
class TestWebForgotPassword(APITestCase):
    collections = [User, Group]

    def setUp(self) -> None:
        super(TestWebForgotPassword, self).setUp()
        self.group = Group.create(name='Power Users', slug='power-users')
        self.user = User.create(username='test', email='tlasso@example.org', first_name='Ted',
                                last_name='Lasso', _credentials='abc123', account_status='ACTIVE',
                                groups=[self.group])
        self.app = self.project.wsgi_app
        self.c = Client(self.app)
        self.token, self.exp = User.api_login("test", "abc123")
        self.user.attach_forgot_verification_key()
        self.key = self.user.forgot_password_key

    def test_send_forgot_req(self):
        res = self.c.post(f'/user/forgot-password/',
                          json={"email": "tlasso@example.org"},
                          headers={
                              "Content-Type": "application/json"})

        self.assertTrue(res.json['success'])
    
    def test_submit_key_for_forgot_pass(self):
        """ Submits the key from the forgot password email to initiate password reset """
        
        res = self.c.put(f'/user/forgot-password/',
                          json={
                              "verification_key": self.key,
                              "password": "new_updated_pass"},
                          headers={
                              "Content-Type": "application/json"})

        new_login = User.api_login("test", "new_updated_pass")
        self.assertTrue(res.json['success'])
        self.assertIsNotNone(new_login)

    def test_submit_wrong_key_for_forgot_pass(self):
        """ Submit a wrong key for verification of reset password. Should return `Not Found` """
        key = 'wrong-key'
        res = self.c.put(f'/user/forgot-password/',
                          json={
                              "verification_key": key,
                              "password": "forgotten_password"},
                          headers={
                              "Content-Type": "application/json"})
        expected = {'data': 'Not Found', 'success': False}

        self.assertFalse(res.json['success'])
        self.assertDictEqual(res.json, expected)

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(TestWebForgotPassword, self).setUp()
        self.group = Group.create(name='Power Users', slug='power-users')
        self.user = User.create(username='test', email='tlasso@example.org', first_name='Ted',
                                last_name='Lasso', _credentials='abc123', account_status='ACTIVE',
                                groups=[self.group])
        self.app = self.project.wsgi_app
        self.c = Client(self.app)
        self.token, self.exp = User.api_login("test", "abc123")
        self.user.attach_forgot_verification_key()
        self.key = self.user.forgot_password_key

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

#   def test_send_forgot_req(self):
View Source
    def test_send_forgot_req(self):
        res = self.c.post(f'/user/forgot-password/',
                          json={"email": "tlasso@example.org"},
                          headers={
                              "Content-Type": "application/json"})

        self.assertTrue(res.json['success'])
#   def test_submit_key_for_forgot_pass(self):
View Source
    def test_submit_key_for_forgot_pass(self):
        """ Submits the key from the forgot password email to initiate password reset """
        
        res = self.c.put(f'/user/forgot-password/',
                          json={
                              "verification_key": self.key,
                              "password": "new_updated_pass"},
                          headers={
                              "Content-Type": "application/json"})

        new_login = User.api_login("test", "new_updated_pass")
        self.assertTrue(res.json['success'])
        self.assertIsNotNone(new_login)

Submits the key from the forgot password email to initiate password reset

#   def test_submit_wrong_key_for_forgot_pass(self):
View Source
    def test_submit_wrong_key_for_forgot_pass(self):
        """ Submit a wrong key for verification of reset password. Should return `Not Found` """
        key = 'wrong-key'
        res = self.c.put(f'/user/forgot-password/',
                          json={
                              "verification_key": key,
                              "password": "forgotten_password"},
                          headers={
                              "Content-Type": "application/json"})
        expected = {'data': 'Not Found', 'success': False}

        self.assertFalse(res.json['success'])
        self.assertDictEqual(res.json, expected)

Submit a wrong key for verification of reset password. Should return Not Found

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