def age_handler(update: Update, context: CallbackContext):
# Получить пол
gender = validate_gender(text=update.message.text)
if gender is None:
update.message.reply_text('Пожалуйста, укажите корректный пол!')
return GENDER
context.user_data[GENDER] = gender
logger.info('user_data: %s', context.user_data)
# Спросить возраст
update.message.reply_text('''
Введите свой возраст:
''')
return AGE
def test_validate_gender_male(self):
r = validate_gender('1')
self.assertEqual(r, 1)
def test_validate_gender_not_int(self):
r = validate_gender('random')
self.assertIsNone(r)