diff --git a/tests/dags/example_dags/__pycache__/anyscale_dag.cpython-311.pyc b/tests/dags/example_dags/__pycache__/anyscale_dag.cpython-311.pyc new file mode 100644 index 0000000..5e8baef Binary files /dev/null and b/tests/dags/example_dags/__pycache__/anyscale_dag.cpython-311.pyc differ diff --git a/tests/dags/example_dags/__pycache__/anyscale_service.cpython-311.pyc b/tests/dags/example_dags/__pycache__/anyscale_service.cpython-311.pyc new file mode 100644 index 0000000..242ec85 Binary files /dev/null and b/tests/dags/example_dags/__pycache__/anyscale_service.cpython-311.pyc differ diff --git a/tests/hooks/test_anyscale_hook.py b/tests/hooks/test_anyscale_hook.py index 3bef42c..50e9088 100644 --- a/tests/hooks/test_anyscale_hook.py +++ b/tests/hooks/test_anyscale_hook.py @@ -158,6 +158,24 @@ def test_get_logs(self, mock_get_logs): mock_get_logs.assert_called_once_with("test_job_id") assert result == "job logs" + @patch('anyscale_provider.hooks.anyscale.AnyscaleHook.get_logs') + def test_get_logs_empty(self, mock_get_logs): + mock_get_logs.return_value = "" + + result = self.hook.get_logs("test_job_id") + + mock_get_logs.assert_called_once_with("test_job_id") + assert result == "" + + @patch('anyscale_provider.hooks.anyscale.AnyscaleHook.get_logs') + def test_get_logs_error(self, mock_get_logs): + mock_get_logs.side_effect = AirflowException("Failed to get logs") + + with pytest.raises(AirflowException) as exc: + self.hook.get_logs("test_job_id") + + assert str(exc.value) == "Failed to get logs" + @patch('anyscale_provider.hooks.anyscale.AnyscaleHook.get_service_status') def test_get_service_status(self, mock_get_service_status): mock_service_status = ServiceStatus(id="test_service_id", name="test_service", query_url="http://example.com", state=ServiceState.RUNNING) @@ -205,29 +223,11 @@ def test_terminate_job_with_delay(self, mock_sleep): mock_terminate.assert_called_once_with(name="test_job_id") mock_sleep.assert_called_once_with(1) assert result is True - + @patch("anyscale_provider.hooks.anyscale.time.sleep", return_value=None) def test_terminate_service_with_delay(self, mock_sleep): with patch.object(self.hook.sdk.service, 'terminate', return_value=None) as mock_terminate: result = self.hook.terminate_service("test_service_id", time_delay=1) mock_terminate.assert_called_once_with(name="test_service_id") mock_sleep.assert_called_once_with(1) - assert result is True - - @patch('anyscale_provider.hooks.anyscale.AnyscaleHook.get_logs') - def test_get_logs_empty(self, mock_get_logs): - mock_get_logs.return_value = "" - - result = self.hook.get_logs("test_job_id") - - mock_get_logs.assert_called_once_with("test_job_id") - assert result == "" - - @patch('anyscale_provider.hooks.anyscale.AnyscaleHook.get_logs') - def test_get_logs_error(self, mock_get_logs): - mock_get_logs.side_effect = AirflowException("Failed to get logs") - - with pytest.raises(AirflowException) as exc: - self.hook.get_logs("test_job_id") - - assert str(exc.value) == "Failed to get logs" \ No newline at end of file + assert result is True \ No newline at end of file