Skip to content

Commit 2868322

Browse files
committed
test: add more test for game_flow
1 parent f1c3975 commit 2868322

File tree

1 file changed

+14
-4
lines changed

1 file changed

+14
-4
lines changed

tests/game/test_game_flow.py

Lines changed: 14 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -37,13 +37,23 @@ def test_resume_game_started(self):
3737

3838
@patch("mastermind.game.player_logic.PlayerLogic.process_player_guessing")
3939
@patch.object(GameFlow, "output_result")
40-
def test_play_game(self, mock_output_result, mock_process_player_guessing):
41-
mock_process_player_guessing.return_value = "q"
42-
self.assertEqual(self.game_flow._play_game(), "q")
40+
def test_play_game_quit(self, mock_output_result, mock_process_player_guessing):
41+
self._test_play_game("q", mock_output_result, mock_process_player_guessing)
42+
43+
@patch("mastermind.game.player_logic.PlayerLogic.process_player_guessing")
44+
@patch.object(GameFlow, "output_result")
45+
def test_play_game_discard(self, mock_output_result, mock_process_player_guessing):
46+
self._test_play_game("d", mock_output_result, mock_process_player_guessing)
47+
48+
def _test_play_game(
49+
self, return_value, mock_output_result, mock_process_player_guessing
50+
):
51+
mock_process_player_guessing.return_value = return_value
52+
self.assertEqual(self.game_flow._play_game(), return_value)
4353
mock_process_player_guessing.assert_called()
4454
mock_output_result.assert_called()
4555

46-
def test_output_win(self): # sourcery skip: extract-duplicate-method
56+
def test_output_win(self): # sourcery skip: class-extract-method, extract-duplicate-method
4757
self.game._player_logic.initialize_players()
4858

4959
self.game._player_logic.PLAYER_CRACKER.win_message = MagicMock()

0 commit comments

Comments
 (0)