mirror of
https://github.com/vale981/ray
synced 2025-03-13 14:46:38 -04:00
f1faa79a04
* Basic event module * Fix comments * Set the SCAN_EVENT_DIR_INTERVAL_SECONDS defaults to 2 * Fix lint * Fix lint * Clean code * Try to fix flaky * Fix test * Disable event module by default * Make monitor events task cancellable * Fix error Co-authored-by: 刘宝 <po.lb@antfin.com>
54 lines
1.2 KiB
Python
54 lines
1.2 KiB
Python
import os
|
|
import pytest
|
|
from ray.tests.conftest import * # noqa
|
|
|
|
|
|
@pytest.fixture
|
|
def enable_test_module():
|
|
os.environ["RAY_DASHBOARD_MODULE_TEST"] = "true"
|
|
yield
|
|
os.environ.pop("RAY_DASHBOARD_MODULE_TEST", None)
|
|
|
|
|
|
@pytest.fixture
|
|
def enable_event_module():
|
|
os.environ["RAY_DASHBOARD_MODULE_EVENT"] = "true"
|
|
yield
|
|
os.environ.pop("RAY_DASHBOARD_MODULE_EVENT", None)
|
|
|
|
|
|
@pytest.fixture
|
|
def disable_aiohttp_cache():
|
|
os.environ["RAY_DASHBOARD_NO_CACHE"] = "true"
|
|
yield
|
|
os.environ.pop("RAY_DASHBOARD_NO_CACHE", None)
|
|
|
|
|
|
@pytest.fixture
|
|
def set_http_proxy():
|
|
http_proxy = os.environ.get("http_proxy", None)
|
|
https_proxy = os.environ.get("https_proxy", None)
|
|
|
|
# set http proxy
|
|
os.environ["http_proxy"] = "www.example.com:990"
|
|
os.environ["https_proxy"] = "www.example.com:990"
|
|
|
|
yield
|
|
|
|
# reset http proxy
|
|
if http_proxy:
|
|
os.environ["http_proxy"] = http_proxy
|
|
else:
|
|
del os.environ["http_proxy"]
|
|
|
|
if https_proxy:
|
|
os.environ["https_proxy"] = https_proxy
|
|
else:
|
|
del os.environ["https_proxy"]
|
|
|
|
|
|
@pytest.fixture
|
|
def small_event_line_limit():
|
|
os.environ["EVENT_READ_LINE_LENGTH_LIMIT"] = "1024"
|
|
yield 1024
|
|
os.environ.pop("EVENT_READ_LINE_LENGTH_LIMIT", None)
|