def setUp(self): drink_pref = Drink(1, "Coffee") self.maker = Person(1, "Chris", "Smith", drink_pref) self.round_duration = 30 person = Person(2, "Bob", "Bobson", drink_pref) drink_order = Drink(2, "Tea") order = Order(person, drink_order) self.orders = [order] self.round = Round(self.maker, self.round_duration, self.orders)
def get_drinks(self): result = queries.get_all_drinks() drinks = {} for drink in result: drink_obj = Drink(drink["drink_id"], drink["name"]) drinks[drink["drink_id"]] = drink_obj return drinks
def get_round_orders(self, round_id): orders_result = queries.get_orders_by_round_id(round_id) orders = {} for order in orders_result: preferred_drink = None preferred_drink_result = queries.get_drink_preference_by_person_id( order["person_id"]) if preferred_drink_result != None: preferred_drink = Drink(preferred_drink_result["drink_id"], preferred_drink_result["drink_name"]) person_obj = Person(order["person_id"], order["first_name"], order["surname"], preferred_drink) drink_obj = Drink(order["drink_id"], order["drink_name"]) order_obj = Order(person_obj, drink_obj, order["special_requests"]) orders[order["order_id"]] = order_obj return orders
def get_people(self): result = queries.get_all_people() people = {} for person in result: drink_obj = None if (person["drink_id"] != None): drink_obj = Drink(person["drink_id"], person["drink_name"]) person_obj = Person(person["person_id"], person["first_name"], person["surname"], drink_obj) people[person["person_id"]] = person_obj return people
def get_current_rounds(self): round_result = queries.get_current_rounds() if (round_result == None): return rounds = {} for round in round_result: maker_preferred_drink = None if (round["drink_id"] != None): maker_preferred_drink = Drink(round["drink_id"], round["drink_name"]) maker = Person(round["maker_id"], round["first_name"], round["surname"], maker_preferred_drink) orders_result = queries.get_orders_by_round_id(round["round_id"]) orders = [] for order in orders_result: ordered_drink = Drink(order["drink_id"], order["drink_name"]) preferred_drink_result = queries.get_drink_preference_by_person_id( order['person_id']) preferred_drink = None if preferred_drink_result != None: preferred_drink = Drink( preferred_drink_result["drink_id"], preferred_drink_result["drink_name"]) person = Person(order["person_id"], order["first_name"], order["surname"], preferred_drink) order_obj = Order(person, ordered_drink) orders.append(order_obj) round_obj = Round(maker, round["minutes_remaining"], orders) rounds[round["round_id"]] = round_obj return rounds
def test_person_integration_GET(self, get_people): payload = { "first_name": "John", "surname": "Smith", "preferred_drink_id": 1 } drink = Drink(1, "Coffee") person = Person(1, "John", "Smith", drink) get_people.return_value = [person] response = self.client.get("/people", data=json.dumps(payload), content_type=self.content_type) self.assertEqual(response.status_code, 200)
def setUp(self): drink_pref = Drink(1, "Coffee") self.person = Person(2, "Bob", "Bobson", drink_pref) self.drink_order = Drink(2, "Tea") self.order = Order(self.person, self.drink_order)
def setUp(self): self.drink = Drink(0, "Coffee")
def setUp(self): self.drink = Drink(0, "Coffee") self.person = Person(0, "Chris", "Smith", self.drink)