def test_find_by_site_name(self): ''' search credential by site_name and return the right details. ''' self.new_credential.save_credential() gmail = Credentials('fennis', 'Gmail', 'mephism', 'naks222') gmail.save_credential() credential_exists = Credentials.find_by_site_name('Gmail') self.assertEqual(credential_exists, gmail)
def test_copy_credential(self): ''' Test if the copy credential method copies the correct credential from credit list. ''' self.new_credential.save_credential() twitter = Credentials("Dennis", "Twitter", "dennism", "nakuru@91") twitter.save_credential() find_credential = None for credential in Credentials.users_credentials_list: find_credential = Credentials.find_by_site_name( credential.site_name) return pyperclip.copy(find_credential.password) Credentials.copy_credential(self.new_credential.site_name) self.assertEqual('nakuru@91', pyperclip.paste()) print(pyperclip.paste())
def find_credentials(site_name): ''' Function that finds a creddential account by sitename and returns the credential account. ''' return Credentials.find_by_site_name(site_name)