def handle_register(update):
user_id = dt.traverse(update, "message.from.id")
username = dt.traverse(update, "message.from.username")
existing = storage.get_user(user_id)
if existing is not None:
return telegram.send_message(
user_id,
"You are already registered.")
storage.register_user(user_id, username)
message = "Welcome. Thank you for registering."
if username is None:
message += "\nBe sure to set a username with /config."
else:
message += "\nYour username is: {}".format(username)
telegram.send_message(
user_id,
message)
def test_user_registered(self):
user_id = 111
username = "******"
storage.register_user(user_id, username)
user = db.get_user(user_id)
self.assertTrue(user is not None)
