graphql-engine/server/tests-py/test_inconsistent_meta.py
Samir Talwar 43779ff0c1 server/tests-py: Fix test_inconsistent_meta.py for split databases.
This test did not work when splitting the metadata and source backends. Fixed mostly by running the relevant SQL using `source_backend.engine`, but I also took the time to clean it up a little, and broke up _test.yaml_ into 3 files.

PR-URL: https://github.com/hasura/graphql-engine-mono/pull/6957
GitOrigin-RevId: bbca60a8906caba2d0cffd834b3b8595fca058fd
2022-11-23 14:14:01 +00:00

82 lines
2.6 KiB
Python

import json
import jsondiff
from ruamel.yaml import YAML
yaml=YAML(typ='safe', pure=True)
def load_yaml(path):
with open(path) as f:
return json.loads(json.dumps(yaml.load(f)))
class TestInconsistentObjects():
reload_metadata = {
"type": "reload_metadata",
"args": {},
}
drop_inconsistent_metadata = {
"type": "drop_inconsistent_metadata",
"args": {},
}
export_metadata = {
"type": "export_metadata",
"args": {},
}
def test_inconsistent_objects(self, hge_ctx, source_backend):
setup = load_yaml(self.dir() + "/setup.yaml")
expected_inconsistent_objects = load_yaml(self.dir() + "/expectation.yaml")
teardown = load_yaml(self.dir() + "/teardown.yaml")
hge_ctx.v1q(setup)
try:
# exec sql to cause inconsistentancy
# TODO: remove once parallelization work is completed
# `source_backend` will no longer be optional
if source_backend:
with source_backend.engine.connect() as connection:
connection.execute('drop table article')
else:
# this only works when the metadata database and the source database are the same
hge_ctx.sql('drop table article')
# reload metadata
resp = hge_ctx.v1q(q=self.reload_metadata)
# check inconsistent objects
actual_inconsistent_objects = resp['inconsistent_objects']
assert resp['is_consistent'] == False, resp
assert actual_inconsistent_objects == expected_inconsistent_objects, yaml.dump({
'response': actual_inconsistent_objects,
'expected': expected_inconsistent_objects,
'diff': jsondiff.diff(expected_inconsistent_objects, actual_inconsistent_objects)
})
# export metadata
export = hge_ctx.v1q(q=self.export_metadata)
# apply metadata
hge_ctx.v1q(
q={
"type": "replace_metadata",
"args": export,
},
expected_status_code = 400,
)
finally:
# drop inconsistent objects
hge_ctx.v1q(q=self.drop_inconsistent_metadata)
# reload metadata
resp = hge_ctx.v1q(q=self.reload_metadata)
# check inconsistent objects
assert resp['is_consistent'] == True, resp
hge_ctx.v1q(teardown)
@classmethod
def dir(cls):
return 'queries/inconsistent_objects'