def setUp(self):
db.create_all()
# Adding Users
user = User(name='user1', email="*****@*****.**")
user2 = User(name='user2', email="*****@*****.**")
user3 = User(name='user3', email="*****@*****.**")
db.session.add_all([user, user2, user3])
# Adding Families
family = Family(name='Family-name-1', members=[user, user2])
family2 = Family(name='Family-name-2', members=[user, user3])
db.session.add_all([family, family2, user3])
# Adding Fridges
fridge = Fridge()
fridge2 = Fridge()
db.session.add_all([fridge, fridge2])
family.fridge = fridge
family2.fridge = fridge2
# Adding Grocery Lists
grocery_list = GroceryList(name="listname1")
grocery_list2 = GroceryList(name="listname2")
grocery_list3 = GroceryList(name="listname3")
db.session.add_all([grocery_list, grocery_list2, grocery_list3])
family.grocery_lists.append(grocery_list)
family.grocery_lists.append(grocery_list2)
family2.grocery_lists.append(grocery_list3)
db.session.commit()
# Adding items
item1 = Item(name="item1", family_pk=family.pk)
item2 = Item(name="item2", family_pk=family.pk)
item3 = Item(name="item3", family_pk=family.pk)
item4 = Item(name="item4", family_pk=family2.pk)
item5 = Item(name="item5", family_pk=family2.pk)
db.session.add_all([item1, item2, item3, item4, item5])
# Adding items to fridge
fridge.items.append(item1)
fridge2.items.append(item2)
# Adding items to grocery lists
grocery_list.items.append(item3)
grocery_list3.items.append(item4)
grocery_list3.items.append(item5)
db.session.commit()
