def test_on_kill(self, mock_kill, mock_rul_cli): operator = PigOperator(pig=PIG, task_id=TEST_TASK_ID) operator.execute(context=TEST_CONTEXT_ID) operator.on_kill() mock_rul_cli.assert_called() mock_kill.assert_called()
def test_execute(self, mock_run_cli): pig_opts = "-x mapreduce" operator = PigOperator(pig=PIG, pig_opts=pig_opts, task_id=TEST_TASK_ID) operator.execute(context=TEST_CONTEXT_ID) mock_run_cli.assert_called_once_with(pig=PIG, pig_opts=pig_opts)
def test_execute_default_pig_opts_to_none(self, mock_run_cli): operator = PigOperator(pig=PIG, task_id=TEST_TASK_ID) operator.execute(context=TEST_CONTEXT_ID) mock_run_cli.assert_called_once_with(pig=PIG, pig_opts=None)