Skip to content

Commit 07b2076

Browse files
authored
Fix: Log MWAA CLI stderr instead of raising an exception (#1527)
1 parent 7e74ed8 commit 07b2076

2 files changed

Lines changed: 4 additions & 25 deletions

File tree

sqlmesh/core/plan/evaluator.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
Refer to `sqlmesh.core.plan`.
1515
"""
1616
import abc
17+
import logging
1718
import typing as t
1819

1920
from sqlmesh.core.console import Console, get_console
@@ -38,6 +39,8 @@
3839
from sqlmesh.utils.date import now
3940
from sqlmesh.utils.errors import SQLMeshError
4041

42+
logger = logging.getLogger(__name__)
43+
4144

4245
class PlanEvaluator(abc.ABC):
4346
@abc.abstractmethod
@@ -370,7 +373,7 @@ def _apply_plan(self, plan: Plan, plan_request_id: str) -> None:
370373
)
371374

372375
if stderr:
373-
raise SQLMeshError(f"Failed to submit a plan application request:\n{stderr}")
376+
logger.warning("MWAA CLI stderr:\n%s", stderr)
374377

375378

376379
def can_evaluate_before_promote(

tests/core/test_plan_evaluator.py

Lines changed: 0 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -142,30 +142,6 @@ def test_mwaa_evaluator(sushi_plan: Plan, mocker: MockerFixture):
142142
mwaa_client_mock.wait_for_first_dag_run.assert_called_once()
143143

144144

145-
def test_mwaa_evaluator_error_from_cli(sushi_plan: Plan, mocker: MockerFixture):
146-
mwaa_client_mock = mocker.Mock()
147-
mwaa_client_mock.wait_for_dag_run_completion.return_value = True
148-
mwaa_client_mock.wait_for_first_dag_run.return_value = "test_plan_application_dag_run_id"
149-
mwaa_client_mock.set_variable.return_value = "", "Error"
150-
151-
state_sync_mock = mocker.Mock()
152-
153-
plan_dag_spec_json = """{"request_id": "test_request_id"}"""
154-
155-
plan_dag_spec_mock = mocker.Mock()
156-
plan_dag_spec_mock.json.return_value = plan_dag_spec_json
157-
158-
create_plan_dag_spec_mock = mocker.patch("sqlmesh.schedulers.airflow.plan.create_plan_dag_spec")
159-
create_plan_dag_spec_mock.return_value = plan_dag_spec_mock
160-
161-
evaluator = MWAAPlanEvaluator(mwaa_client_mock, state_sync_mock)
162-
163-
with pytest.raises(SQLMeshError, match=r"Failed to submit a plan application request:\nError"):
164-
evaluator.evaluate(sushi_plan)
165-
166-
mwaa_client_mock.set_variable.assert_called_once_with(mocker.ANY, plan_dag_spec_json)
167-
168-
169145
def test_can_evaluate_before_promote(sushi_context: Context):
170146
parent_model_a = SqlModel(
171147
name="sushi.new_test_model_a",

0 commit comments

Comments
 (0)