def test_find_by_service(self): """ test display of credentials according to service :return: """ self.new_credential.save_credentials() test_credential = Credentials("Eugene znm", "whatsapp", "123ert") test_credential.save_credentials() found_credential = Credentials.find_by_service('whatsapp') self.assertEqual(found_credential.service, test_credential.service)
def find_credential(service): """ function to find credential by service """ return Credentials.find_by_service(service)