chore: import upstream snapshot with attribution

tests/claude_code/test_plugin.py

@@ -0,0 +1,49 @@
+import json
+import subprocess
+from unittest import mock
+
+import click
+import pytest
+
+from mlflow.claude_code.plugin import disable_tracing_plugin, ensure_plugin_installed
+
+
+def test_disable_tracing_plugin_removes_env_only(tmp_path):
+    settings_path = tmp_path / ".claude" / "settings.json"
+    settings_path.parent.mkdir(parents=True)
+    settings_path.write_text(
+        json.dumps({
+            "env": {
+                "MLFLOW_CLAUDE_TRACING_ENABLED": "true",
+                "MLFLOW_TRACKING_URI": "http://localhost:5000",
+                "MLFLOW_EXPERIMENT_ID": "123",
+            },
+            "other": "keep-me",
+        })
+    )
+
+    assert disable_tracing_plugin(settings_path) is True
+    config = json.loads(settings_path.read_text())
+    assert config == {"other": "keep-me"}
+
+
+def test_ensure_plugin_installed_runs_marketplace_add_and_install(tmp_path):
+    completed = subprocess.CompletedProcess(args=[], returncode=0, stdout="", stderr="")
+
+    with (
+        mock.patch("shutil.which", return_value="/usr/local/bin/claude"),
+        mock.patch("subprocess.run", return_value=completed) as mock_run,
+    ):
+        ensure_plugin_installed(tmp_path)
+
+    assert mock_run.call_count == 2
+    first_command = mock_run.call_args_list[0].args[0]
+    second_command = mock_run.call_args_list[1].args[0]
+    assert first_command[:5] == ["claude", "plugin", "marketplace", "add", "mlflow/mlflow"]
+    assert second_command[:4] == ["claude", "plugin", "install", "mlflow-tracing@mlflow-plugins"]
+
+
+def test_ensure_plugin_installed_requires_claude_binary(tmp_path):
+    with mock.patch("shutil.which", return_value=None):
+        with pytest.raises(click.ClickException, match="Claude Code CLI"):
+            ensure_plugin_installed(tmp_path)