credentials_test.py

import unittest  # Importing the unittest module
from credentials import Credentials


class TestCredentials(unittest.TestCase):
    '''
    Test class that defines test cases for Credentials class behaviours

    Args:
        unittest.TestCase: TestCase class that  helps in creating test cases
    '''

    def setUp(self):
        '''
        Set up method to run before each test cases
        '''
        self.new_credentials = Credentials("Twitter", "Jimmy", "1234")

    def test_initialization(self):
        '''
        Test initialization test case to test if the object is initialized properly
        '''
        self.assertEqual(self.new_credentials.application_name, "Twitter")
        self.assertEqual(self.new_credentials.account_username, "Jimmy")
        self.assertEqual(self.new_credentials.account_password, "1234")

    def tearDown(self):
        '''
        tearDown method that does clean up after each test case has run
        '''
        Credentials.credentials_list = []

    def test_add_multiple_credentials(self):
        '''
        test_add_multiple_credentials to check if we can save multiple credential objects to our credentials list
        '''
        self.new_credentials.add_credentials()
        test_credentials = Credentials("Twitter", "Mike", "1234")
        test_credentials.add_credentials()
        self.assertEqual(len(Credentials.credentials_list), 2)

    def test_delete_credentials(self):
        '''
        test_delete_credentials to test if we can remove a credential from credentials list
        '''
        self.new_credentials.add_credentials()
        test_credentials = Credentials("Twitter", "Mike", "1234")
        test_credentials.add_credentials()

        test_credentials.delete_credentials()
        self.assertEqual(len(Credentials.credentials_list), 1)

    def test_display_all_credentials(self):
        '''
        test method that returns a list of all credentials saved
        '''
        self.assertEqual(Credentials.display_credentials(),
                         Credentials.credentials_list)

    def test_find_by_application_name(self):
        '''
        test to check if we can find a credentials by application name
        '''
        self.new_credentials.add_credentials()
        test_credentials = Credentials("Twitter", "Jimmy", "1234")
        test_credentials.add_credentials()

        found_credential = Credentials.find_by_application_name("Twitter")
        self.assertEqual(found_credential.account_username,
                         test_credentials.account_username)

    def test_credentials_exist(self):
        ''' 
        test to check if we can return boolean if credential exist or not
        '''
        self.new_credentials.add_credentials()
        test_credentials = Credentials("Twitter", "Mike", "1234")
        test_credentials.add_credentials()

        credentials_exist = Credentials.credentials_exist("Twitter")
        self.assertTrue(credentials_exist)


if __name__ == '__main__':
    unittest.main()