c85f3fbbb7
LavaFarm is a class created to handle the different types of LAVA farms
and their tags in Mesa CI. Since specific jobs may require different
types of LAVA farms to run on, it is essential to determine which farm
the runner is running on to configure the job correctly.
LavaFarm provides an easy-to-use interface for checking the runner tag
and returning the corresponding LAVA farm, making it simple for Mesa CI
to configure jobs appropriately. By adding tests for LavaFarm, the team
can ensure that this class is functioning as expected, allowing for the
smooth execution of Mesa CI jobs on the correct LAVA farm.
The tests ensure that get_lava_farm returns the correct LavaFarm value
when given invalid or valid tags and that it returns LavaFarm.UNKNOWN
when no tag is provided. The tests use Hypothesis strategies to generate
various labels and farms for testing.
Example of use:
```
from lava.utils.lava_farm import LavaFarm, get_lava_farm
lava_farm = get_lava_farm()
if lava_farm == LavaFarm.DUMMY:
# Configure the job for the DUMMY farm
...
elif lava_farm == LavaFarm.COLLABORA:
# Configure the job for the COLLABORA farm
...
elif lava_farm == LavaFarm.KERNELCI:
# Configure the job for the KERNELCI farm
...
else:
# Handle the case where the LAVA farm is unknown
...
```
Signed-off-by: Guilherme Gallo <guilherme.gallo@collabora.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/21325>
42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
import re
|
|
|
|
import pytest
|
|
from hypothesis import given
|
|
from hypothesis import strategies as st
|
|
from lava.utils.lava_farm import LAVA_FARM_RUNNER_PATTERNS, LavaFarm, get_lava_farm
|
|
|
|
|
|
@given(
|
|
runner_tag=st.text(
|
|
alphabet=st.characters(
|
|
min_codepoint=1, max_codepoint=127, blacklist_categories=("C",)
|
|
),
|
|
min_size=1,
|
|
)
|
|
)
|
|
def test_get_lava_farm_invalid_tags(runner_tag):
|
|
with pytest.MonkeyPatch().context() as mp:
|
|
mp.setenv("RUNNER_TAG", runner_tag)
|
|
assert get_lava_farm() == LavaFarm.UNKNOWN
|
|
|
|
|
|
def test_get_lava_farm_no_tag(monkeypatch):
|
|
monkeypatch.delenv("RUNNER_TAG", raising=False)
|
|
assert get_lava_farm() == LavaFarm.UNKNOWN
|
|
|
|
|
|
@given(
|
|
st.fixed_dictionaries(
|
|
{k: st.from_regex(v) for k, v in LAVA_FARM_RUNNER_PATTERNS.items()}
|
|
)
|
|
)
|
|
def test_get_lava_farm_valid_tags(runner_farm_tag: dict):
|
|
with pytest.MonkeyPatch().context() as mp:
|
|
for farm, tag in runner_farm_tag.items():
|
|
try:
|
|
mp.setenv("RUNNER_TAG", tag)
|
|
except ValueError:
|
|
# hypothesis may generate null bytes in the string
|
|
continue
|
|
assert get_lava_farm() == farm
|