def test_goal_set_input_calls_verify_input(self): g = Goal() g.verify_input = mock.Mock(return_value=True) g.set_input('kekeke') g.verify_input.assert_called_once_with('kekeke') self.assertEqual(g.input, 'kekeke')
def test_goal_invalid_input_raises_exception(self): g = Goal() g.verify_input = mock.Mock(return_value=False) self.assertRaises(BadInput, g.set_input, 'kokok')
def test_goal_can_verify_input(self): g = Goal() g.verify_input = lambda x: bool(x) self.assertTrue(g.verify_input('koko'))