def test_prompt_first_attempt_ok(self, m_vi, m_gi): p = Prompter() player = Mock(name='name') m_gi.return_value = 'get_input' m_vi.side_effect = [None, (0, 0)] self.assertEqual((0, 0), p.prompt(player)) self.assertEqual(2, len(m_vi.call_args_list))
def test_prompt_first_attempt_ok(self, m_vi, m_gi): p = Prompter() player = Mock(name='name') m_gi.return_value = 'a1' m_vi.return_value = (0, 0) self.assertEqual((0, 0), p.prompt(player)) m_gi.assert_called_once_with(player) m_vi.assert_called_once_with('a1')