wehub-resource-sync
46 lines
1.0 KiB
Python
46 lines
1.0 KiB
Python
import sys
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from ray._private.runtime_env import uv
|
|
|
|
|
|
class TestRuntimeEnv:
|
|
def uv_config(self):
|
|
return {"packages": ["requests"]}
|
|
|
|
def env_vars(self):
|
|
return {}
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_install_uv():
|
|
with patch(
|
|
"ray._private.runtime_env.uv.UvProcessor._install_uv"
|
|
) as mock_install_uv:
|
|
mock_install_uv.return_value = None
|
|
yield mock_install_uv
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_install_uv_packages():
|
|
with patch(
|
|
"ray._private.runtime_env.uv.UvProcessor._install_uv_packages"
|
|
) as mock_install_uv_packages:
|
|
mock_install_uv_packages.return_value = None
|
|
yield mock_install_uv_packages
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run(mock_install_uv, mock_install_uv_packages):
|
|
target_dir = "/tmp"
|
|
runtime_env = TestRuntimeEnv()
|
|
|
|
uv_processor = uv.UvProcessor(target_dir=target_dir, runtime_env=runtime_env)
|
|
await uv_processor._run()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(pytest.main(["-vv", __file__]))
|