def test_input_retry_with__bad_departure_and_bad_arrival_value(self, inputs, print):
inputs.side_effect = ["HAS -- SAX", "DUB -- LAX"]
dep, arr = airport_distance.read_in_airports(["DUB", "LAX"])
self.assertEqual(2, inputs.call_count)
self.assertEqual(1, print.call_count)
self.assertTrue("destination airport" in print.call_args.args[0])
self.assertTrue("departure airport" in print.call_args.args[0])
def test_input_retry_with_bad_input_value(self, inputs, print):
inputs.side_effect = [" NOTAGOODVAL", "DUB -- LAX"]
dep, arr = airport_distance.read_in_airports(["DUB", "LAX"])
self.assertEqual(2, inputs.call_count)
self.assertEqual(1, print.call_count)
self.assertTrue("Unable to parse" in print.call_args.args[0])
def test_getting_input_with_good_input(self, good_input):
good_input.return_value = "DUB -- LAX"
dep, arr = airport_distance.read_in_airports(["DUB", "LAX"])
self.assertEqual("DUB", dep)
self.assertEqual("LAX", arr)