class TestCli(unittest.TestCase):
def setUp(self):
self.cli = Cli(IoMock())
def testCliPrintsBoard(self):
self.cli.prompt_board(BoardMock())
output = self.cli.io.output
self.assertIn('1', output)
def testCliAsksForMove(self):
expected = 'Please enter your move: '
self.cli.ask_move()
output = self.cli.io.output
self.assertEqual(expected, output)
def testCliPromptsGoodBye(self):
expected = 'Good bye!'
self.cli.prompt_good_bye()
output = self.cli.io.output
self.assertEqual(expected, output)
def testCliPromptsWelcome(self):
expected = 'Welcome!'
self.cli.prompt_welcome()
output = self.cli.io.output
self.assertEqual(expected, output)
def testReturnsUserInput(self):
expected = '5'
user_input = self.cli.ask_move()
self.assertEqual(expected, user_input)
