import ruamel.yaml as yaml
import re

class TestConfigAPI():

    def test_config_api(self, hge_ctx):
        admin_secret = hge_ctx.hge_key
        auth_hook = hge_ctx.hge_webhook
        jwt_conf = hge_ctx.hge_jwt_conf

        headers = {}
        if admin_secret is not None:
            headers['x-hasura-admin-secret'] = admin_secret

        resp = hge_ctx.http.get(hge_ctx.hge_url + '/v1alpha1/config', headers=headers)
        body = resp.json()

        # The tree may be dirty because we're developing tests locally while
        # graphql-engine was built previously when tree was clean. If we're
        # modifying graphql-engine too then both of these will be tagged dirty,
        # since a rebuild would necessarily be forced:
        assert body['version'] in (hge_ctx.version, re.sub('-dirty$', '', hge_ctx.version))
        assert body['is_admin_secret_set'] == (admin_secret is not None)
        assert body['is_auth_hook_set'] == (auth_hook is not None)
        assert body['is_jwt_set'] == (jwt_conf is not None)

        if jwt_conf is not None:
            claims_namespace = "https://hasura.io/jwt/claims"
            if 'claims_namespace' in jwt_conf:
                claims_namespace = jwt_conf['claims_namespace']
            claims_format = "json"
            if 'claims_format' in jwt_conf:
                claims_format = jwt_conf['claims_format']
            assert body['jwt']['claims_namespace'] == claims_namespace
            assert body['jwt']['claims_format'] == claims_format
        else:
            assert body['jwt'] == None

        # test if the request fails without auth headers if admin secret is set
        if admin_secret is not None:
            resp = hge_ctx.http.get(hge_ctx.hge_url + '/v1alpha1/config')
            body = resp.json()
            assert ((resp.status_code == 401) or (resp.status_code == 400))