@@ -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