mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-14 17:02:49 +03:00
3b0ad30757
PR-URL: https://github.com/hasura/graphql-engine-mono/pull/7216 GitOrigin-RevId: 20a94267eca42edf8e79470711766a3fecac0f50
65 lines
2.0 KiB
Python
65 lines
2.0 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import pytest
|
|
import jsondiff
|
|
from ruamel.yaml import YAML
|
|
|
|
yaml=YAML(typ='safe', pure=True)
|
|
|
|
usefixtures = pytest.mark.usefixtures
|
|
|
|
# A very basic test that compression seems to be working
|
|
@usefixtures('per_class_tests_db_state')
|
|
class TestCompression:
|
|
|
|
gzip_header = {'Accept-Encoding': 'gzip'}
|
|
|
|
def _make_post(self, hge_ctx, u, q, h):
|
|
if hge_ctx.hge_key is not None:
|
|
h['X-Hasura-Admin-Secret'] = hge_ctx.hge_key
|
|
resp = hge_ctx.http.post(
|
|
hge_ctx.hge_url + u,
|
|
json=q,
|
|
headers=h
|
|
)
|
|
return resp
|
|
|
|
def _get_config(self, f):
|
|
with open(f) as c:
|
|
conf = yaml.load(c)
|
|
return conf['url'], conf['query'], conf['response']
|
|
|
|
def _assert_status_code_200(self, resp):
|
|
assert resp.status_code == 200, resp.json()
|
|
|
|
def _assert_encoding(self, headers, encoding):
|
|
assert 'Content-Encoding' in headers, headers
|
|
assert headers['Content-Encoding'] == encoding, headers
|
|
|
|
def _assert_resp(self, resp, exp_resp):
|
|
json_resp = resp.json()
|
|
assert json_resp == exp_resp, yaml.dump({
|
|
'response': json_resp,
|
|
'expected': exp_resp,
|
|
'diff': jsondiff.diff(exp_resp, json_resp)
|
|
})
|
|
|
|
def _assert_gzip(self, resp, exp_resp):
|
|
self._assert_status_code_200(resp)
|
|
self._assert_encoding(resp.headers, 'gzip')
|
|
self._assert_resp(resp, exp_resp)
|
|
|
|
def test_gzip_compression_graphql(self, hge_ctx):
|
|
url, q, exp_resp = self._get_config(self.dir() + '/graphql_query.yaml')
|
|
resp = self._make_post(hge_ctx, url, q, self.gzip_header)
|
|
self._assert_gzip(resp, exp_resp)
|
|
|
|
def test_gzip_compression_v1_query(self, hge_ctx):
|
|
url, q, exp_resp = self._get_config(self.dir() + '/v1_query.yaml')
|
|
resp = self._make_post(hge_ctx, url, q, self.gzip_header)
|
|
self._assert_gzip(resp, exp_resp)
|
|
|
|
@classmethod
|
|
def dir(cls):
|
|
return 'queries/compression'
|