def test_other_general(self):
cafe = Cafe(1, [{
"order_id": 1,
"order_time": 0,
"type": "affogato"
}, {
"order_id": 2,
"order_time": 1,
"type": "tea"
}, {
"order_id": 3,
"order_time": 2,
"type": "latte"
}, {
"order_id": 4,
"order_time": 2,
"type": "tea"
}], 'test_output.json')
# So the input data doesn't have to be huge
cafe.open_time = 5
cafe.open_for_business()
self.assertEqual(cafe.output_data, [{
"order_id": 2,
"start_time": 1,
"barista_id": 1
}, {
"order_id": 3,
"start_time": 2,
"barista_id": 2
}, {
"order_id": 4,
"start_time": 4,
"barista_id": 1
}])
def test_fifo_general(self):
# test general case with given same input file
with open('sample_input.json', 'r') as input_file:
decoder = json.JSONDecoder()
input_data = decoder.decode(input_file.read())
cafe = Cafe(0, input_data, 'test_output.json')
cafe.open_for_business()
# use given sample output file as "rubric"
with open('sample_output_fifo.json', 'r') as input_file:
decoder = json.JSONDecoder()
sample_output = decoder.decode(input_file.read())
self.assertEqual(cafe.output_data, sample_output)
def test_no_orders(self):
cafe = Cafe(0, [], 'test_output.json')
cafe.open_for_business()
self.assertEqual(cafe.output_data, [])
cafe = Cafe(1, [], 'test_output.json')
cafe.open_for_business()
self.assertEqual(cafe.output_data, [])
def salesman(name):
salesman = Salesman(name)
cafe = Cafe()
pricing_service = cafe.pricing_service
if name not in cafe.get_salesman_list():
raise Exception(f"There is no {name} in list of salesmans")
answers = prompt(questions=coffee_questions(cafe.menu),
style=custom_style_1) #how to add some(2) latte?
# pprint(answers)
pricing_service.update_summary_table_by_name("total", name, answers)
pricing_service.update_summary_table_by_name("number_of_sales", name,
answers)
if answers[const.BILL] == const.YES:
pricing_service.show_price(answers, salesman)
def test_invalid_order_time(self):
cafe = Cafe(0, [{
"order_time": -1,
"order_id": 1,
"type": "affogato"
}], 'test_output.json')
self.assertRaises(InvalidInputDataException, cafe.open_for_business)
def test_one_order(self):
cafe = Cafe(0, [{
"order_time": 4,
"order_id": 1,
"type": "tea"
}], 'test_output.json')
cafe.open_for_business()
self.assertEqual(cafe.output_data, [{
"barista_id": 1,
"start_time": 4,
"order_id": 1
}])
cafe = Cafe(1, [{
"order_time": 4,
"order_id": 1,
"type": "tea"
}], 'test_output.json')
cafe.open_for_business()
self.assertEqual(cafe.output_data, [{
"barista_id": 1,
"start_time": 4,
"order_id": 1
}])