Function run_group_graphql_test

python/test_utils/utils.py:132–148

(queries_and_expected_outputs, graph, sort_output=False)

def run_group_graphql_test(queries_and_expected_outputs, graph, sort_output=False):
    tmp_work_dir = tempfile.mkdtemp()
    with GraphServer(tmp_work_dir).start() as server:
        client = server.get_client()
        client.send_graph(path="g", graph=graph)

        for query, expected_output in queries_and_expected_outputs:
            response = client.query(query)
            response_dict = (
                json.loads(response) if isinstance(response, str) else response
            )
            if sort_output:
                response_dict = sort_by_gql_name_or_id(response_dict)
                expected_output = sort_by_gql_name_or_id(expected_output)
            assert (
                response_dict == expected_output
            ), f"Expected:\n{sort_dict_recursive(expected_output)}\nGot:\n{sort_dict_recursive(response_dict)}"


def run_graphql_error_test(query, expected_error_message, graph):