def test_bad_calls(self):
@self.callback
def error_callback(result, error):
self.assertIs(result, None)
self.assertIsNot(error, None)
random_object = object()
user_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"
user_db.find_user_id(hashed, callback)
self.wait()
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({
"_id": result["_id"],
"phone": result["phone"]
})
return
UserDB.create_user(phone, hashed, create_callback)
hashed = user_hash(phone, device)
UserDB.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)
user_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)
user_db.find_user_id(hashed, check_created)
self.wait()
