1
0
mirror of https://github.com/hasura/graphql-engine.git synced 2024-12-17 20:41:49 +03:00
graphql-engine/server/tests-py/test_compression.py
Brandon Simmons 3b0ad30757 server: don't compress small responses
PR-URL: https://github.com/hasura/graphql-engine-mono/pull/7216
GitOrigin-RevId: 20a94267eca42edf8e79470711766a3fecac0f50
2022-12-13 17:50:01 +00:00

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'