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()