add uv installation in becnhmark

Author: yungwine

mytonctrl/mytonctrl.py

@@ -406,8 +406,18 @@ def upgrade_btc_teleport(local, ton, reinstall=False, branch: str = 'master', us
 
 def run_benchmark(args: list):
 	if shutil.which("uv") is None:
-		color_print("{red}Error: uv is not installed. Install it: https://docs.astral.sh/uv/getting-started/installation/{endc}")
-		return
+		answer = input("uv is not installed. Install it? [y/n] ").strip().lower()
+		if answer == "y":
+			subprocess.run(["curl", "-LsSf", "https://astral.sh/uv/install.sh", "-o", "/tmp/uv_install.sh"], check=True)
+			subprocess.run(["sh", "/tmp/uv_install.sh"], check=True)
+			uv_local_bin = os.path.expanduser("~/.local/bin")
+			if uv_local_bin not in os.environ.get("PATH", ""):
+				os.environ["PATH"] = uv_local_bin + os.pathsep + os.environ.get("PATH", "")
+			if shutil.which("uv") is None:
+				color_print("{red}Error: uv installation failed{endc}")
+				return
+		else:
+			return
 
 	if get_service_status("validator"):
 		color_print("{red}Error: validator service is running. Stop it before running benchmark: `sudo systemctl stop validator`{endc}")

tests/integration/test_benchmark.py

@@ -4,10 +4,44 @@
 from unittest.mock import MagicMock
 
 
-def test_benchmark_uv_not_installed(cli, monkeypatch):
+def test_benchmark_uv_not_installed_decline(cli, monkeypatch):
     monkeypatch.setattr(shutil, "which", lambda name: None)
+    monkeypatch.setattr("builtins.input", lambda prompt: "n")
+
+    calls = []
+    monkeypatch.setattr(subprocess, "run", lambda args, **kw: calls.append(args))
+
+    cli.execute("benchmark", no_color=True)
+    assert len(calls) == 0
+
+
+def test_benchmark_uv_not_installed_install(cli, monkeypatch):
+    installed = {"uv": False}
+
+    def fake_which(name):
+        if name == "uv" and installed["uv"]:
+            return "/usr/bin/uv"
+        return None
+
+    monkeypatch.setattr(shutil, "which", fake_which)
+    monkeypatch.setattr("builtins.input", lambda prompt: "y")
+
+    calls = []
+
+    def fake_subprocess_run(args, **kwargs):
+        calls.append([str(a) for a in args])
+        if args[:2] == ["sh", "/tmp/uv_install.sh"]:
+            installed["uv"] = True
+        return subprocess.CompletedProcess(args, 0)
+
+    monkeypatch.setattr(subprocess, "run", fake_subprocess_run)
+
+    from mytonctrl import mytonctrl as mytonctrl_module
+    monkeypatch.setattr(mytonctrl_module, "get_service_status", lambda name: True)
+
     output = cli.execute("benchmark", no_color=True)
-    assert "uv is not installed" in output
+    assert calls[0] == ["curl", "-LsSf", "https://astral.sh/uv/install.sh", "-o", "/tmp/uv_install.sh"]
+    assert calls[1] == ["sh", "/tmp/uv_install.sh"]
 
 
 def test_benchmark_validator_running(cli, monkeypatch):