def get(self, id): query_set = User.objects(id=id) try: u = query_set.first() return loads(u.public_json()) except: return ErrorHandler.not_found()
def _get_timer(self, id): query_set = User.objects(id=id) try: u = query_set.first() user_timer = UserTimer(u, db) return user_timer.timer_get() except: return None
def verify_password(username, password): global current_user user = User.verify_auth_token(password) # Todo review -- Is verifying user part of auth token sufficient? Seems to me we should also be verifying the generated token? if not user: return False current_user = user return True
def test_can_create_john_timer(self): query_result = User.objects(email="*****@*****.**") u = query_result.first() user_timer = UserTimer(u, db) timer = user_timer.timer_create() timer.set_seconds_today(300) timer.notes = "John's perpetual timer" timer.start() timer.save()
def test_login(self): # Create the user email = test_data["TEST_USER_EMAIL"] password = test_data.get("TEST_USER_PASSWORD") userOriginal = User(email=email, accountId=test_data["DEMO"], password=password) userOriginal.save() try: credentials = dict(email = test_data["TEST_USER_EMAIL"], password = test_data["TEST_USER_PASSWORD"]) login_payload = dumps(credentials) response = requests.post(url=test_server + "/login", data=login_payload, headers={'content-type' : 'application/json'}) assert(response.status_code == 200) response_object = response.json() user = response_object["response"]["user"] assert(user["id"] is not None) assert(user["authentication_token"] is not None) finally: #Cleanup userOriginal.delete()
def test_login_and_use_resource(self): # Create User email = test_data["TEST_USER_EMAIL"] password = test_data.get("TEST_USER_PASSWORD") userOriginal = User(email=email, accountId=test_data["DEMO"], password=password) userOriginal.save() try: credentials = dict(email = test_data["TEST_USER_EMAIL"], password = password) login_payload = dumps(credentials) response = requests.post(url=test_server + "/login", data=login_payload, headers={'content-type' : 'application/json'}) assert(response.status_code == 200) response_object = response.json() user = response_object["response"]["user"] basic_auth_credentials = HTTPBasicAuth(email, user["authentication_token"]) response = requests.get(url=test_server + "/auth/api/resource", headers={'content-type' : 'application/json'}, auth=basic_auth_credentials) assert(response.status_code == 200) assert(email in str(response.json()["data"])) finally: #Cleanup userOriginal.delete()