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)