class TestCredentials(unittest.TestCase): ''' Test class that defines test cases for the credential class behaviours. Args: unittest.TestCase: TestCase class that helps in creating test cases ''' def tearDown(self): ''' tearDown method that does clean up after each test case has run. ''' Credentials.credential_list = [] def setUp(self): ''' Set up method to run before each test cases. ''' self.new_credentials = Credentials( "Twitter", "Chiri", "pass123") # create credential object def test_init(self): ''' test_init test case to test if the object is initialized properly ''' self.assertEqual(self.new_credentials.account_platform, "Twitter") self.assertEqual(self.new_credentials.account_name, "Chiri") self.assertEqual(self.new_credentials.account_password, "pass123") def test_save_credentials(self): ''' test_save_credentials test case to test if the credentials object is saved into the credentials list ''' self.new_credentials.save_credentials() # saving the new credentials self.assertEqual(len(Credentials.credentials_list), 3) def test_create_credentials(self): ''' test_create_credentials test case to test if the credentials object is added into the credentials list ''' self.new_credentials.create_credentials() # create new credentials self.assertEqual(len(Credentials.credentials_list), 1) def test_save_multiple_credentials(self): ''' test_save_multiple_credentials to check if we can save multiple credentials objects to our credentials_list ''' self.new_credentials.save_credentials() test_credentials = Credentials("Twitter", "Chiri", "pass123") # new credentials test_credentials.save_credentials() self.assertEqual(len(Credentials.credentials_list), 5) def test_delete_credentials(self): ''' test_delete_credentials to test if we can remove a credentials from our credentials list ''' self.new_credentials.save_credentials() test_credentials = Credentials("Twitter", "Chiri", "pass123") # new credentials test_credentials.save_credentials() self.new_credentials.delete_credentials( ) # Deleting a credentials object self.assertEqual(len(Credentials.credentials_list), 2) def test_display_all_credentials(self): ''' method that returns a list of all credentials saved ''' self.assertEqual(Credentials.display_credentials(), Credentials.credentials_list)
def create_credentials(save_credentials): ''' function to create new user ''' Credentials.create_credentials("account_name", "login_detail", "Password")