def test_chat_get_remaining_nutrition(self):
request = {
'session_id': 'abc.123',
'text': '',
'context': {},
'entities': {}
}
e = eater.Eater(parent=ndb.Key('Account', 9), id='abc', fb_id='abc')
e.nutrition_plans = [
eater.NutritionPlan(protein=2, carb=4, fat=6, cals=78)
] * 7
def get_remaining_nutrition_patch():
return {
'cals': 0,
'protein': 0,
'carb': 0,
'fat': 0,
'fiber': 0,
'sodium': 0
}
e.get_remaining_nutrition = get_remaining_nutrition_patch
e.put()
context = chat.get_remaining_nutrition(request)
self.assertEqual(0, context['cals'])
self.assertTrue(context['complete'])
def test_nutrition_plan_insert(self):
p = eater.NutritionPlan()
p.put()
self.assertEqual(0, p.protein)
self.assertEqual(25, p.fiber_min)
self.assertEqual(2000, p.sodium_max)
# Check sat fat calc
p.fat = 100
p.put()
self.assertEqual(30, p.satfat_max)
def test_eater_get_remaining_nutrition(self):
e = eater.Eater()
e.nutrition_plans = [
eater.NutritionPlan(protein=2, carb=4, fat=6, cals=78)
] * 7
e.put()
rec = eater.NutritionRecord(parent=e.key, protein=0, carb=1, fat=2)
rec.put()
nut = e.get_remaining_nutrition()
self.assertEqual(2, nut['protein'])
self.assertEqual(3, nut['carb'])
self.assertEqual(4, nut['fat'])
self.assertEqual(56, nut['cals'])
def test_eater_get_nutrition_plan_today(self):
e = eater.Eater()
e.nutrition_plans = [eater.NutritionPlan(protein=i) for i in range(7)]
e.put()
today = datetime.datetime.now().weekday()
self.assertEqual(today, e.get_nutrition_plan_today().protein)