graphql-engine/server/tests-py/test_horizontal_scale.py

import pytest
import time
import json
import jsondiff
from context import PytestConf
from ruamel.yaml import YAML

yaml=YAML(typ='safe', pure=True)


if not PytestConf.config.getoption("--test-hge-scale-url"):
    pytest.skip("--test-hge-scale-url flag is missing, skipping tests", allow_module_level=True)


class TestHorizontalScaleBasic():
    servers = {}

    @pytest.fixture(autouse=True, scope='class')
    def transact(self, hge_ctx):
        self.servers['1'] = hge_ctx.hge_url
        self.servers['2'] = hge_ctx.hge_scale_url
        yield
        # teardown
        hge_ctx.v1q_f(self.dir() + '/teardown.yaml')

    def test_horizontal_scale_basic(self, hge_ctx):
        with open(self.dir() + "/steps.yaml") as c:
            conf = yaml.load(c)

        assert isinstance(conf, list) == True, 'Not a list'
        for _, step in enumerate(conf):
            # execute operation
            response = hge_ctx.http.post(
                self.servers[step['operation']['server']] + "/v1/query",
                json=step['operation']['query']
            )
            resp = response.json()
            assert \
                response.status_code == 200, \
                f'Expected {response.status_code} to be 200. Response:\n{json.dumps(resp, indent=2)}'

            # wait for 20 sec
            time.sleep(20)
            # validate data
            response = hge_ctx.http.post(
                self.servers[step['validate']['server']] + "/v1alpha1/graphql",
                json=step['validate']['query']
            )
            resp = response.json()
            assert \
                response.status_code == 200, \
                f'Expected {response.status_code} to be 200. Response:\n{json.dumps(resp, indent=2)}'

            if 'response' in step['validate']:
                assert resp == step['validate']['response'], yaml.dump({
                    'response': resp,
                    'expected': step['validate']['response'],
                    'diff': jsondiff.diff(step['validate']['response'], resp)
                })

    @classmethod
    def dir(cls):
        return 'queries/horizontal_scale/basic'