def test_poke_failed(self, finish_code, mock_tableau_hook):
"""
Test poke failed
"""
mock_tableau_hook.return_value.__enter__ = Mock(return_value=mock_tableau_hook)
mock_get = mock_tableau_hook.server.jobs.get_by_id
mock_get.return_value.finish_code = finish_code
sensor = TableauJobStatusSensor(**self.kwargs)
with pytest.raises(TableauJobFailedException):
sensor.poke({})
mock_get.assert_called_once_with(sensor.job_id)
def test_poke(self, mock_tableau_hook):
"""
Test poke
"""
mock_tableau_hook.return_value.__enter__ = Mock(return_value=mock_tableau_hook)
mock_get = mock_tableau_hook.server.jobs.get_by_id
mock_get.return_value.finish_code = '0'
sensor = TableauJobStatusSensor(**self.kwargs)
job_finished = sensor.poke(context={})
assert job_finished
mock_get.assert_called_once_with(sensor.job_id)