def test_add_food_user(self):
cmndr = Commander()
foodstore.registerFoodStoreCommands(cmndr)
request_body = {'message': {'from': {u'id': 123}}}
result = myutils.handle_message(cmndr, u'/add_food', request_body, [u'\ud83c\udf6a',u'100'])
assert result == [u'Got it ! \ud83c\udf6a=100']
assert ndb.Key('FoodCalorieValues', u'\ud83c\udf6a:123').get().calories == 100
def test_handle_foods_no_max_calories(self):
cmndr = Commander()
foodstore.registerFoodStoreCommands(cmndr)
# foodstore.register_handle_foods(cmndr)
request_body = {'message': {'date': 1542830400, 'from': {u'id': 123}}}
myutils.handle_message(cmndr, u'/add_food_default', request_body, [u'\ud83c\udf6a', u'100'])
result = myutils.handle_message(cmndr, u'\ud83c\udf6a', request_body, [])
assert result == ['Sorry, you didn\'t set max calories. Please use /set_max_calories command.\nCalories counted 100/0']
def test_handle_foods_got_max_calories(self):
cmndr = Commander()
foodstore.registerFoodStoreCommands(cmndr)
# foodstore.register_handle_foods(cmndr)
userstore.register_user_commands(cmndr)
request_body = {'message': {'date': 1542830400, 'from': {u'id': 123}}}
myutils.handle_message(cmndr, u'/set_max_calories', request_body, ['2010'])
result = myutils.handle_message(cmndr, u'\ud83c\udf6a', request_body, [])
assert result == ['I got your message! (but I do not know how to answer)']
def test_show_foods(self):
cmndr = Commander()
foodstore.registerFoodStoreCommands(cmndr)
request_body = {'message': {'from': {u'id': 123}}}
myutils.handle_message(cmndr, u'/add_food_default', request_body, [u'\ud83c\udf6a', u'100'])
myutils.handle_message(cmndr, u'/add_food_default', request_body, [u'\ud83c\udf6b', u'100'])
myutils.handle_message(cmndr, u'/add_food', request_body, [u'\ud83c\udf6b', u'20'])
result = myutils.handle_message(cmndr, u'/show_foods', request_body, [])
assert result == [u'Default foods: \n\U0001f36a = 100\n\U0001f36b = 100\nCustom foods: \n\U0001f36b = 20']
def test_handle_foods_multiple_foods(self):
cmndr = Commander()
foodstore.registerFoodStoreCommands(cmndr)
# foodstore.register_handle_foods(cmndr)
userstore.register_user_commands(cmndr)
request_body = {'message': {'date': 1542830400, 'from': {u'id': 123}}}
myutils.handle_message(cmndr, u'/add_food_default', request_body, [u'\U0001f36b', u'100'])
myutils.handle_message(cmndr, u'/set_max_calories', request_body, ['2010'])
result = myutils.handle_message(cmndr, u'\U0001f36b\U0001f36b', request_body, [])
assert result == ['Calories counted 200/2010']
def test_show_food(self):
cmndr = Commander()
foodstore.registerFoodStoreCommands(cmndr)
request_body = {'message': {'from': {u'id': 123}}}
myutils.handle_message(cmndr, u'/add_food', request_body, [u'\ud83c\udf6a', u'100'])
result = myutils.handle_message(cmndr, u'/show_food', request_body, [u'\ud83c\udf6a'])
assert result == [u'Custom \ud83c\udf6a = 100']
myutils.handle_message(cmndr, u'/add_food_default', request_body, [u'\ud83c\udf6a', u'35'])
result = myutils.handle_message(cmndr, u'/show_food', request_body, [u'\ud83c\udf6a'])
assert result == [u'Default \ud83c\udf6a = 35\nCustom \ud83c\udf6a = 100']
def test_handle_foods_default_food(self):
cmndr = Commander()
foodstore.registerFoodStoreCommands(cmndr)
# foodstore.register_handle_foods(cmndr)
userstore.register_user_commands(cmndr)
request_body = {'message': {'date': 1542830400, 'from': {u'id': 123}}}
myutils.handle_message(cmndr, u'/add_food_default', request_body, [u'\ud83c\udf6a', u'100'])
myutils.handle_message(cmndr, u'/add_food', request_body, [u'\ud83c\udf6a', u'50'])
myutils.handle_message(cmndr, u'/add_food', request_body, [u'\ud83c\udf6b', u'50'])
myutils.handle_message(cmndr, u'/add_food_default', request_body, [u'\ud83c\udf6c', u'100'])
myutils.handle_message(cmndr, u'/set_max_calories', request_body, ['2010'])
result = myutils.handle_message(cmndr, u'here is what I ate: \ud83c\udf6a\ud83c\udf6c', request_body, [])
result = myutils.handle_message(cmndr, u'and again: \ud83c\udf6c', request_body, [])
assert result == ['Calories counted 250/2010']
import bottoken
import foodstore
import myutils
import botenabler
import coffeestore
import userstore
import waterstore
from commander import Commander
TOKEN = bottoken.get_token()
BASE_URL = 'https://api.telegram.org/bot' + TOKEN + '/'
commander = Commander()
botenabler.register_user_commands(commander)
foodstore.registerFoodStoreCommands(commander)
coffeestore.registerCoffeeCommands(commander)
userstore.register_user_commands(commander)
waterstore.registerWaterCommands(commander)
# ================================
class MeHandler(webapp2.RequestHandler):
def get(self):
urlfetch.set_default_fetch_deadline(60)
self.response.write(
json.dumps(json.load(urllib2.urlopen(BASE_URL + 'getMe'))))
class GetUpdatesHandler(webapp2.RequestHandler):
def test_show_food_not_recognized(self):
cmndr = Commander()
foodstore.registerFoodStoreCommands(cmndr)
request_body = {'message': {'from': {u'id': 123}}}
result = myutils.handle_message(cmndr, u'/show_food', request_body, [u'\ud83c\udf6a'])
assert result == [u'Sorry, I don\'t recognize this food: \ud83c\udf6a']
def test_add_default_food(self):
cmndr = Commander()
foodstore.registerFoodStoreCommands(cmndr)
result = cmndr.execute('/add_food_default', {}, ['\ud83c\udf6a','100'])
assert result == 'Got it ! \ud83c\udf6a=100'