def test_bad_calls(self): @self.callback def error_callback(result, error): self.assertIs(result, None) self.assertIsNot(error, None) random_object = object() auth_db.find_user_id(random_object, error_callback) self.wait()
def test_find_user_id(self): @self.callback def callback(result, error): self.assertIs(error, None) self.assertIs(result, None) hashed = "distincthash1" auth_db.find_user_id(hashed, callback) self.wait()
def wrapper(self, data): @mongo_callback(self) def find_callback(result): if not result: raise AuthError() func(self, data=data, userId=result["_id"]) phone = self.request.headers.get("phone") device = self.request.headers.get("device") if not phone or not device: raise AuthError() AuthDB.find_user_id(user_hash(phone, device), find_callback)
def login(self, phone, device): @mongo_callback(self) def create_callback(result): self.respond({ "_id": result, "phone": phone, }) @mongo_callback(self) def find_callback(result): if result: self.respond(result) return AuthDB.create_user(phone, hashed, create_callback) hashed = user_hash(phone, device) AuthDB.find_user_id(hashed, find_callback)
def test_create_user(self): phone = "fakephonenumber" hashed = "distincthash2" # Create @self.callback def callback(result, error): self.assertIs(error, None) self.assertIsInstance(result, ObjectId) auth_db.create_user(phone, hashed, callback) self.wait() # Check if created @self.callback def check_created(result, error): self.assertIs(error, None) self.assertEqual(result["phone"], phone) auth_db.find_user_id(hashed, check_created) self.wait()