mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-19 13:31:43 +03:00
66beb02f7b
* Revert "add brotli shared lib to packager image (#2924)"
This reverts commit 79414cb225.
* remove unneeded dockerfile
* remove brotli traces in ci images and compression test
Co-authored-by: Rakesh Emmadi <12475069+rakeshkky@users.noreply.github.com>
61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import pytest
|
|
import ruamel.yaml as yaml
|
|
import jsondiff
|
|
|
|
from super_classes import DefaultTestSelectQueries
|
|
|
|
class TestCompression(DefaultTestSelectQueries):
|
|
|
|
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.safe_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'
|