47659ddf70
Refactor the LAVA farm detection to use a simpler environment variable-based approach: - Remove the complex regex-based farm detection - Replace LavaFarm enum with a simple string-based farm identification - Update related tests and job definition logic - Remove hypothesis testing dependency Signed-off-by: Guilherme Gallo <guilherme.gallo@collabora.com> Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/33888>
75 lines
2.1 KiB
Python
75 lines
2.1 KiB
Python
from collections import defaultdict
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
import yaml
|
|
from freezegun import freeze_time
|
|
|
|
from .lava.helpers import generate_testsuite_result, jobs_logs_response
|
|
|
|
|
|
def pytest_configure(config):
|
|
config.addinivalue_line(
|
|
"markers", "slow: marks tests as slow (deselect with '-m \"not slow\"')"
|
|
)
|
|
|
|
@pytest.fixture
|
|
def mock_sleep():
|
|
"""Mock time.sleep to make test faster"""
|
|
with patch("time.sleep", return_value=None):
|
|
yield
|
|
|
|
|
|
@pytest.fixture
|
|
def frozen_time(mock_sleep):
|
|
with freeze_time("2024-01-01") as frozen_time:
|
|
yield frozen_time
|
|
|
|
|
|
RESULT_GET_TESTJOB_RESULTS = [{"metadata": {"result": "test"}}]
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_proxy(frozen_time):
|
|
def mock_job_state(jid) -> dict[str, str]:
|
|
frozen_time.tick(1)
|
|
return {"job_state": "Running"}
|
|
|
|
def create_proxy_mock(
|
|
job_results=RESULT_GET_TESTJOB_RESULTS,
|
|
testsuite_results=[generate_testsuite_result()],
|
|
**kwargs
|
|
):
|
|
proxy_mock = MagicMock()
|
|
proxy_submit_mock = proxy_mock.scheduler.jobs.submit
|
|
proxy_submit_mock.return_value = "1234"
|
|
|
|
proxy_results_mock = proxy_mock.results.get_testjob_results_yaml
|
|
proxy_results_mock.return_value = yaml.safe_dump(job_results)
|
|
|
|
proxy_test_suites_mock = proxy_mock.results.get_testsuite_results_yaml
|
|
proxy_test_suites_mock.return_value = yaml.safe_dump(testsuite_results)
|
|
|
|
proxy_logs_mock = proxy_mock.scheduler.jobs.logs
|
|
proxy_logs_mock.return_value = jobs_logs_response()
|
|
|
|
proxy_job_state = proxy_mock.scheduler.job_state
|
|
proxy_job_state.side_effect = mock_job_state
|
|
|
|
proxy_show_mock = proxy_mock.scheduler.jobs.show
|
|
proxy_show_mock.return_value = defaultdict(
|
|
str,
|
|
{
|
|
"device_type": "test_device",
|
|
"device": "test_device-cbg-1",
|
|
"state": "created",
|
|
},
|
|
)
|
|
|
|
for key, value in kwargs.items():
|
|
setattr(proxy_logs_mock, key, value)
|
|
|
|
return proxy_mock
|
|
|
|
yield create_proxy_mock
|