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)