Aaryan123456679 commented on code in PR #69264:
URL: https://github.com/apache/airflow/pull/69264#discussion_r3610633479


##########
dev/breeze/tests/test_docker_command_utils.py:
##########
@@ -389,3 +391,76 @@ def 
test_bring_all_compose_projects_down_preserve_volumes(mock_discover, mock_ru
     down_call = next(c for c in mock_run_command.call_args_list if 
c.args[0][:2] == ["docker", "compose"])
     assert "--volumes" not in down_call.args[0]
     assert "--remove-orphans" in down_call.args[0]
+
+
+def _shell_params_for_openlineage(backend: str, postgres_version: str) -> 
mock.MagicMock:
+    shell_params = mock.MagicMock()
+    shell_params.use_airflow_version = None
+    shell_params.restart = False
+    shell_params.include_mypy_volume = False
+    shell_params.quiet = True
+    shell_params.project_name = None
+    shell_params.tty = "disabled"
+    shell_params.command_passed = None
+    shell_params.integration = ("openlineage",)

Review Comment:
   Good catch, thanks! Added `("all",)` as a parametrized case alongside 
`("openlineage",)` for all three tests in bb38659afd.



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: [email protected]

For queries about this service, please contact Infrastructure at:
[email protected]

Reply via email to