mirror of
https://github.com/open-jarvis/OpenJarvis.git
synced 2026-07-28 14:07:55 +00:00
Jon Saad-FalconandGitHub
f41cf420be
Consolidates NVIDIA vLLM, Apple Silicon, CPU Pearl mining support, CLI/docs, and live H100 validation.
137 lines
4.9 KiB
Python
137 lines
4.9 KiB
Python
"""Tests for openjarvis.mining._pearl_subprocess."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture
|
|
def fake_popen():
|
|
"""A MagicMock standing in for a live subprocess.Popen handle."""
|
|
p = MagicMock()
|
|
p.poll.return_value = None # still running
|
|
p.pid = 12345
|
|
return p
|
|
|
|
|
|
def _make_launcher(tmp_path):
|
|
"""Build a launcher with safe defaults that won't conflict with anything."""
|
|
from openjarvis.mining._pearl_subprocess import PearlSubprocessLauncher
|
|
|
|
return PearlSubprocessLauncher(
|
|
gateway_host="127.0.0.1",
|
|
gateway_port=18337,
|
|
metrics_port=18339,
|
|
pearld_rpc_url="http://localhost:44107",
|
|
pearld_rpc_user="rpcuser",
|
|
pearld_rpc_password="testpw",
|
|
wallet_address="prl1qtest",
|
|
log_dir=tmp_path,
|
|
)
|
|
|
|
|
|
def test_launcher_accepts_custom_provider_and_miner_module(fake_popen, tmp_path):
|
|
"""Providers can select a distinct miner-loop module and log file."""
|
|
from openjarvis.mining._pearl_subprocess import PearlSubprocessLauncher
|
|
|
|
with patch("subprocess.Popen", return_value=fake_popen) as mock_popen:
|
|
launcher = PearlSubprocessLauncher(
|
|
gateway_host="127.0.0.1",
|
|
gateway_port=18337,
|
|
metrics_port=18339,
|
|
pearld_rpc_url="http://localhost:44107",
|
|
pearld_rpc_user="rpcuser",
|
|
pearld_rpc_password="testpw",
|
|
wallet_address="prl1qtest",
|
|
log_dir=tmp_path,
|
|
provider_id="apple-mps-pearl",
|
|
miner_module="openjarvis.mining._mps_miner_loop_main",
|
|
)
|
|
launcher.start(m=128, n=128, k=1024, rank=64)
|
|
|
|
second_call = mock_popen.call_args_list[1]
|
|
assert "_mps_miner_loop_main" in " ".join(second_call.args[0])
|
|
|
|
|
|
def test_launcher_start_spawns_two_processes(fake_popen, tmp_path):
|
|
"""start() spawns gateway and miner-loop subprocesses."""
|
|
with patch("subprocess.Popen", return_value=fake_popen) as mock_popen:
|
|
launcher = _make_launcher(tmp_path)
|
|
launcher.start(m=256, n=128, k=1024, rank=32)
|
|
assert mock_popen.call_count == 2 # gateway + miner-loop
|
|
|
|
|
|
def test_launcher_stop_terminates_both(fake_popen, tmp_path):
|
|
"""stop() sends SIGTERM to both subprocesses."""
|
|
with patch("subprocess.Popen", return_value=fake_popen):
|
|
launcher = _make_launcher(tmp_path)
|
|
launcher.start(m=256, n=128, k=1024, rank=32)
|
|
launcher.stop()
|
|
assert fake_popen.terminate.call_count >= 2
|
|
|
|
|
|
def test_launcher_is_running_false_when_one_exited(tmp_path):
|
|
"""is_running() returns False if either subprocess has exited."""
|
|
fake_alive = MagicMock()
|
|
fake_alive.poll.return_value = None
|
|
fake_alive.pid = 11111
|
|
fake_dead = MagicMock()
|
|
fake_dead.poll.return_value = 1
|
|
fake_dead.pid = 22222
|
|
|
|
with patch("subprocess.Popen", side_effect=[fake_alive, fake_dead]):
|
|
launcher = _make_launcher(tmp_path)
|
|
launcher.start(m=256, n=128, k=1024, rank=32)
|
|
assert launcher.is_running() is False
|
|
|
|
|
|
def test_launcher_pids_returns_both(fake_popen, tmp_path):
|
|
"""pids() returns (gateway_pid, miner_loop_pid) after start."""
|
|
fake_a = MagicMock()
|
|
fake_a.poll.return_value = None
|
|
fake_a.pid = 11111
|
|
fake_b = MagicMock()
|
|
fake_b.poll.return_value = None
|
|
fake_b.pid = 22222
|
|
with patch("subprocess.Popen", side_effect=[fake_a, fake_b]):
|
|
launcher = _make_launcher(tmp_path)
|
|
launcher.start(m=256, n=128, k=1024, rank=32)
|
|
assert launcher.pids() == (11111, 22222)
|
|
|
|
|
|
def test_launcher_pids_returns_none_before_start(tmp_path):
|
|
"""pids() returns None before start()."""
|
|
launcher = _make_launcher(tmp_path)
|
|
assert launcher.pids() is None
|
|
|
|
|
|
def test_launcher_stop_idempotent(fake_popen, tmp_path):
|
|
"""stop() called twice is a no-op the second time."""
|
|
with patch("subprocess.Popen", return_value=fake_popen):
|
|
launcher = _make_launcher(tmp_path)
|
|
launcher.start(m=256, n=128, k=1024, rank=32)
|
|
launcher.stop()
|
|
launcher.stop() # should not raise
|
|
|
|
|
|
def test_launcher_spawn_order_is_gateway_then_miner_loop(fake_popen, tmp_path):
|
|
"""Gateway must be spawned before the miner-loop."""
|
|
with patch("subprocess.Popen", return_value=fake_popen) as mock_popen:
|
|
launcher = _make_launcher(tmp_path)
|
|
launcher.start(m=256, n=128, k=1024, rank=32)
|
|
first_call = mock_popen.call_args_list[0]
|
|
second_call = mock_popen.call_args_list[1]
|
|
assert first_call.args[0][0] == "pearl-gateway"
|
|
assert "_miner_loop_main" in " ".join(second_call.args[0])
|
|
|
|
|
|
def test_launcher_start_twice_raises(fake_popen, tmp_path):
|
|
"""Calling start() while already running raises RuntimeError."""
|
|
with patch("subprocess.Popen", return_value=fake_popen):
|
|
launcher = _make_launcher(tmp_path)
|
|
launcher.start(m=256, n=128, k=1024, rank=32)
|
|
with pytest.raises(RuntimeError, match="already started"):
|
|
launcher.start(m=256, n=128, k=1024, rank=32)
|