def test_sugar(self, input_details):
# arrange
input_details.side_effect = [2]
expected = "with sugar"
# act
actual = d.milk_sugar()
# assert
self.assertEqual(expected, actual)
def test_straight_up(self, input_details):
# arrange
input_details.side_effect = [3]
expected = "-"
# act
actual = d.milk_sugar()
# assert
self.assertEqual(expected, actual)
def test_milk(self, input_details):
# arrange
input_details.side_effect = [1]
expected = "extra milk"
# act
actual = d.milk_sugar()
# assert
self.assertEqual(expected, actual)