def test_getMultipleFilesToPlay(self):
logger = IOTest()
logger.setInputList(["1", "0", "2"])
getFileToPlay(self.fileList, logger)
getFileToPlay(self.fileList, logger)
self.assertEqual("track_3.mp3", getFileToPlay(self.fileList, logger))
def test_getValidFileToPlay(self):
logger = IOTest()
logger.setInputList(["1"])
self.assertEqual("track_2.mp3", getFileToPlay(self.fileList, logger))
def test_getEmptyFileToPlay(self):
logger = IOTest()
logger.setInputList(["", "0"])
getFileToPlay(self.fileList, logger)
self.assertEqual("That is an invalid track number",
logger.outputList[-1])