Method assert_graphql_query_result

edb/testbase/http.py:425–485

(
        self,
        query,
        result,
        *,
        msg=None,
        sort=None,
        operation_name=None,
        use_http_post=True,
        native_variables=None,
        variables=None,
        globals=None,
        deprecated_globals=None,
        config=None,
    )

            await con.aclose()

    def assert_graphql_query_result(
        self,
        query,
        result,
        *,
        msg=None,
        sort=None,
        operation_name=None,
        use_http_post=True,
        native_variables=None,
        variables=None,
        globals=None,
        deprecated_globals=None,
        config=None,
    ):
        # Try to use the native protocol first!
        if operation_name is None and config is None:
            try:
                res = asyncio.run(self._native_graphql_query(
                    query,
                    variables=(
                        native_variables if native_variables is not None
                        else variables
                    ),
                    globals=globals or deprecated_globals,
                ))

                if sort is not None:
                    # GQL will always have a single object
                    # returned. The data is in the top-level fields,
                    # so that's what needs to be sorted.
                    for r in res.values():
                        assert_data_shape.sort_results(r, sort)

                assert_data_shape.assert_data_shape(
                    res, result, self.fail, message=msg)
            except gel.UnsupportedFeatureError as e:
                if 'Default variables are not supported' in str(e):
                    # Whatever.
                    pass
                else:
                    raise

        res = self.graphql_query(
            query,
            operation_name=operation_name,
            use_http_post=use_http_post,
            variables=variables,
            globals=globals,
            deprecated_globals=deprecated_globals,
            config=config,
        )

        if sort is not None:
            # GQL will always have a single object returned. The data is
            # in the top-level fields, so that's what needs to be sorted.
            for r in res.values():
                assert_data_shape.sort_results(r, sort)