def _get_one(self): self.recipe = Recipe('carbo') self.m1 = Meal(datetime.date(2010, 01, 01), 0, [self.recipe]) self.m2 = Meal(datetime.date(2010, 01, 05), 1, [self.recipe]) Session.add_all([self.recipe, self.m1, self.m2]) fro = datetime.date(2010, 01, 01) to = datetime.date(2010, 01, 10) pw = PlanningWidget(fro, to) return pw
def test_we_got_the_right_meals(self): recipe = Recipe('carbo') m1 = Meal(datetime.date(2010, 01, 01), 0, [recipe]) m2 = Meal(datetime.date(2010, 01, 05), 0, [recipe]) m3 = Meal(datetime.date(2010, 01, 12), 1, [recipe]) Session.add_all((recipe, m1, m2, m3)) obj = mock.Mock(PlanningWidget) obj.fro = datetime.date(2010, 01, 01) obj.to = datetime.date(2010, 01, 10) self.assertEqual(PlanningWidget._meals_in_range(obj), [m1, m2])
def load_some_tags(): t1 = Tag(u'pâtes') t2 = Tag(u'tomate') r1 = Recipe(u'carbonara') r1.tags = [t1] r2 = Recipe(u'bolognaise') r2.tags = [t1, t2] r3 = Recipe(u'tomates farcies') r3.tags = [t2] r4 = Recipe(u'saucisses-purée') tags = [t1, t2] recipes = [r1, r2, r3, r4] Session.add_all(tags) Session.add_all(recipes) Session.flush() return tags, recipes
def setUp(self): super(TagsWidgetTestCase, self).setUp() Session.add_all(Tag(t) for t in [u'lourd', u'pâtes', u'hiver']) self.app = get_app()