def setUp(self):
self.user1 = User.objects.create_user(
email="*****@*****.**",
username="******",
password="******",
)
self.user2 = User.objects.create_user(
email="*****@*****.**",
username="******",
password="******",
)
self.user3 = User.objects.create_user(
email="*****@*****.**",
username="******",
password="******",
)
self.party = Party(name="rust party",
location="rustland",
type=int(PartyType.InGroup),
leader=self.user1)
self.party.save()
self.state = self.party.get_state()
self.restaurant = Restaurant(name="Rustaurant")
self.restaurant.save()
self.menu1 = Menu(name="Trait", price=10000)
self.menu1.save()
self.menu2 = Menu(name="Struct", price=33000)
self.menu2.save()
self.menu3 = Menu(name="Macro", price=13130)
self.menu3.save()
self.state.member_ids = [self.user1.id, self.user2.id, self.user3.id]
self.state.member_ids_backup = [
self.user1.id, self.user2.id, self.user3.id
]
self.state.menu_entries.add(
self.menu1.id, 3, [self.user1.id, self.user2.id, self.user3.id])
self.state.menu_entries.add(
self.menu2.id, 1, [self.user1.id, self.user2.id, self.user3.id])
self.state.menu_entries.add(
self.menu3.id, 6, [self.user1.id, self.user2.id, self.user3.id])
self.state.phase = PartyPhase.PaymentAndCollection
self.state.paid_user_id = self.user2.id
self.state.restaurant_id = self.restaurant.id
self.state.save()
self.party.restaurant_id = self.restaurant.id
self.party.save()
