def test_find_by_site(self): ''' test to find out if user can find credentials using site name ''' self.new_credential.save_credentials() test_credential = Credentials("Jane", "Eyre", "facebook", "char257") test_credential.save_credentials() found_credential = Credentials.find_by_site("facebook") self.assertEqual(found_credential.password, test_credential.password)
def get_credential(site_name): ''' method to look for credential using site_name ''' return Credentials.find_by_site(site_name)