|
13 | 13 | from cuda.pathfinder._dynamic_libs.load_dl_common import LoadedDL |
14 | 14 | from cuda.pathfinder._dynamic_libs.load_nvidia_dynamic_lib import ( |
15 | 15 | _load_lib_no_cache, |
| 16 | + _resolve_system_loaded_abs_path_in_subprocess, |
16 | 17 | _try_ctk_root_canary, |
17 | 18 | ) |
18 | 19 | from cuda.pathfinder._utils.platform_aware import IS_WINDOWS |
@@ -140,6 +141,58 @@ def test_try_via_ctk_root_regular_lib(tmp_path): |
140 | 141 | assert _FindNvidiaDynamicLib("cudart").try_via_ctk_root(str(ctk_root)) == str(cudart_lib) |
141 | 142 |
|
142 | 143 |
|
| 144 | +# --------------------------------------------------------------------------- |
| 145 | +# _resolve_system_loaded_abs_path_in_subprocess |
| 146 | +# --------------------------------------------------------------------------- |
| 147 | + |
| 148 | + |
| 149 | +def test_subprocess_probe_returns_abs_path_on_string_payload(mocker): |
| 150 | + result = mocker.Mock(stdout='"/usr/local/cuda/lib64/libcudart.so.13"\n') |
| 151 | + mocker.patch(f"{_MODULE}.run_in_spawned_child_process", return_value=result) |
| 152 | + |
| 153 | + assert _resolve_system_loaded_abs_path_in_subprocess("cudart") == "/usr/local/cuda/lib64/libcudart.so.13" |
| 154 | + |
| 155 | + |
| 156 | +def test_subprocess_probe_returns_none_on_null_payload(mocker): |
| 157 | + result = mocker.Mock(stdout="null\n") |
| 158 | + mocker.patch(f"{_MODULE}.run_in_spawned_child_process", return_value=result) |
| 159 | + |
| 160 | + assert _resolve_system_loaded_abs_path_in_subprocess("cudart") is None |
| 161 | + |
| 162 | + |
| 163 | +def test_subprocess_probe_raises_on_child_failure(mocker): |
| 164 | + mocker.patch( |
| 165 | + f"{_MODULE}.run_in_spawned_child_process", |
| 166 | + side_effect=ChildProcessError("child failed"), |
| 167 | + ) |
| 168 | + with pytest.raises(ChildProcessError, match="child failed"): |
| 169 | + _resolve_system_loaded_abs_path_in_subprocess("cudart") |
| 170 | + |
| 171 | + |
| 172 | +def test_subprocess_probe_raises_on_empty_stdout(mocker): |
| 173 | + result = mocker.Mock(stdout=" \n \n") |
| 174 | + mocker.patch(f"{_MODULE}.run_in_spawned_child_process", return_value=result) |
| 175 | + |
| 176 | + with pytest.raises(RuntimeError, match="produced no stdout payload"): |
| 177 | + _resolve_system_loaded_abs_path_in_subprocess("cudart") |
| 178 | + |
| 179 | + |
| 180 | +def test_subprocess_probe_raises_on_invalid_json_payload(mocker): |
| 181 | + result = mocker.Mock(stdout="not-json\n") |
| 182 | + mocker.patch(f"{_MODULE}.run_in_spawned_child_process", return_value=result) |
| 183 | + |
| 184 | + with pytest.raises(RuntimeError, match="invalid JSON payload"): |
| 185 | + _resolve_system_loaded_abs_path_in_subprocess("cudart") |
| 186 | + |
| 187 | + |
| 188 | +def test_subprocess_probe_raises_on_unexpected_json_payload(mocker): |
| 189 | + result = mocker.Mock(stdout='{"path": "/usr/local/cuda/lib64/libcudart.so.13"}\n') |
| 190 | + mocker.patch(f"{_MODULE}.run_in_spawned_child_process", return_value=result) |
| 191 | + |
| 192 | + with pytest.raises(RuntimeError, match="unexpected payload"): |
| 193 | + _resolve_system_loaded_abs_path_in_subprocess("cudart") |
| 194 | + |
| 195 | + |
143 | 196 | # --------------------------------------------------------------------------- |
144 | 197 | # _try_ctk_root_canary |
145 | 198 | # --------------------------------------------------------------------------- |
|
0 commit comments