pgadmin4/web/pgadmin/browser/tests/test_reset_password.py

# ##########################################################################
#
# #pgAdmin 4 - PostgreSQL Tools
#
# #Copyright (C) 2013 - 2016, The pgAdmin Development Team
# #This software is released under the PostgreSQL Licence
#
# ##########################################################################

import uuid

from pgadmin.utils.route import BaseTestGenerator
from regression.test_setup import config_data
from test_utils import login_tester_account, logout_tester_account


class ResetPasswordTestCase(BaseTestGenerator):
    """
    This class validates the reset password functionality by defining
    scenarios; Each dict parameter describe a scenario appended by
    test name.
    """

    scenarios = [
        # This test case validates the empty email field
        ('TestCase for Validating Empty Email', dict(
            email='', respdata='Email not provided')),

        # This test case validates the invalid/incorrect email field
        ('TestCase for Validating Invalid_Email', dict(
            email=str(uuid.uuid4())[1:6] + '@xyz.com',
            respdata='Specified user does not exist')),

        # This test case validates the valid email id
        ('TestCase for Validating Valid_Email', dict(
            email=config_data['pgAdmin4_login_credentials']
            ['test_login_username'], respdata='pgAdmin 4'))
    ]

    def setUp(self):
        logout_tester_account(self.tester)

    def runTest(self):
        """This function checks reset password functionality."""

        response = self.tester.get('/reset')
        self.assertIn('Recover pgAdmin 4 Password', response.data.decode(
            'utf-8'))
        response = self.tester.post(
            '/reset', data=dict(email=self.email),
            follow_redirects=True)
        self.assertIn(self.respdata, response.data.decode('utf-8'))

    def tearDown(self):
        login_tester_account(self.tester)