sts-sponsors team mailing list archive

[Björn Tillenius <mp+433941@xxxxxxxxxxxxxxxxxx>]

Diff comments:

> diff --git a/src/maasperf/tests/maasspike/test_listing.py b/src/maasperf/tests/maasspike/test_listing.py
> new file mode 100644
> index 0000000..6b29bd1
> --- /dev/null
> +++ b/src/maasperf/tests/maasspike/test_listing.py
> @@ -0,0 +1,121 @@
> +"""Performance tests for the machine listing spike.
> +
> +All tests that test the performance of the different implementations for the
> +machine listing spike should go in this file.
> +
> +For each implemenation, we should have a test that lists all machines, and
> +a test that list 50 machines.
> +
> +Each test should measure how long it takes to produce the list, and then
> +assert that the listing is the same as the original websocket handler.
> +"""
> +
> +import pytest
> +
> +from maasserver.models import Machine
> +from maasserver.websockets.handlers.machine import MachineHandler
> +from maasspike import baseline
> +
> +
> +class ExpectedMachines:
> +    def __init__(self, expected_list):
> +        self._expected_list = expected_list
> +
> +    def assert_list(self, machine_list, limit=None):
> +        """Assert that the passed in machine list is correct.
> +
> +        Compare the number of machines in each list, as well as making
> +        sure that the first and last machine of each list are equal.
> +
> +        Comparing the full list takes too long if you have a list of
> +        1000 machines.
> +        """
> +        assert isinstance(machine_list, list)
> +        expected_list = (
> +            self._expected_list[:limit] if limit else self._expected_list
> +        )
> +        assert len(machine_list) == len(expected_list)

Yes, that's a good idea.

> +        assert machine_list[0] == expected_list[0]
> +        assert machine_list[-1] == expected_list[-1]
> +
> +
> +def get_expected_machines(admin):
> +    """Get the list of machines that the normal websocket handler returns."""
> +    machine_count = Machine.objects.all().count()
> +    ws_handler = MachineHandler(admin, {}, None)
> +    params = {
> +        "filter": {},
> +        "page_number": 1,
> +        "page_size": machine_count + 1,
> +        "sort_direction": "ascending",
> +        "sort_key": "hostname",
> +    }
> +    result = ws_handler.list(params)
> +    return ExpectedMachines(result["groups"][0]["items"])
> +
> +
> +@pytest.fixture(scope="session")
> +def _expected():

We could, but i'd like to avoid global variables if possible.

> +    """Helper fixture to store the expected machine list for the session.
> +
> +    A session fixture doesn't have access to the DB, so we make use of a
> +    function fixture, expected_machines, to get the machine listing
> +    for the first test and store it here.
> +
> +    The fixture shouldn't be used by any test.
> +    """
> +    return {}
> +
> +
> +@pytest.fixture
> +def expected_machines(admin, _expected):
> +    if "machines" not in _expected:
> +        _expected["machines"] = get_expected_machines(admin)
> +    return _expected["machines"]
> +
> +
> +def test_populate_expected_machines(expected_machines):
> +    """A blank test to populate the expected_machines fixture.
> +
> +    This should be the first test to be run, so that other test
> +    don't get any advantage by having the query to the machine
> +    listing cached.
> +    """
> +
> +
> +@pytest.mark.parametrize("limit", [None, 50])
> +class TestListing:
> +    """Collection of tests for spike machine listing implementations.
> +
> +    A class is used to group the tests together to ensure they measure
> +    and assert the same thing.
> +
> +    Each implementation should have a test that does the required setup
> +    and then call self.run_listing_test.
> +
> +    Each test is run once using the full listing, and once using the first
> +    50 machines.
> +    """
> +
> +    @pytest.fixture(autouse=True)
> +    def set_up(self, perf, admin, expected_machines):
> +        self._expected_machines = expected_machines
> +        self._admin = admin
> +        self._perf = perf
> +        yield
> +        self._expected_machines = None
> +        self._admin = None
> +        self._perf = None
> +
> +    def run_listing_test(self, name, func, limit):
> +        record_name = name
> +        if limit:
> +            record_name += f"_{limit}"
> +        else:
> +            record_name += "_all"
> +        with self._perf.record(record_name):
> +            machines = func(self._admin, limit)
> +        self._expected_machines.assert_list(machines, limit)
> +
> +    def test_baseline(self, limit):
> +        self.run_listing_test("baseline", baseline.list_machines, limit)


-- 
https://code.launchpad.net/~bjornt/maas/+git/maas/+merge/433941
Your team MAAS Committers is subscribed to branch maas:machine-list-spike.