mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-24 07:52:14 +03:00
c2cb07f7e8
We use a helper service to start a webhook-based authentication service for some tests. This moves the initialization of the service out of _test-server.sh_ and into the Python test harness, as a fixture. In order to do this, I had to make a few changes. The main deviation is that we no longer run _all_ tests against an HGE with this authentication service, just a few (those in _test_webhook.py_). Because this reduced coverage, I have added some more tests there, which actually cover some areas not exacerbated elsewhere (mainly trying to use webhook credentials to talk to an admin-only endpoint). The webhook service can run both with and without TLS, and decide whether it's necessary to skip one of these based on the arguments passed and how HGE is started, according to the following logic: * If a TLS CA certificate is passed in, it will run with TLS, otherwise it will skip it. * If HGE was started externally and a TLS certificate is provided, it will skip running without TLS, as it will assume that HGE was configured to talk to a webhook over HTTPS. * Some tests should only be run with TLS; this is marked with a `tls_webhook_server` marker. * Some tests should only be run _without_ TLS; this is marked with a `no_tls_webhook_server` marker. The actual parameterization of the webhook service configuration is done through test subclasses, because normal pytest parameterization doesn't work with the `hge_fixture_env` hack that we use. Because `hge_fixture_env` is not a sanctioned way of conveying data between fixtures (and, unfortunately, there isn't a sanctioned way of doing this when the fixtures in question may not know about each other directly), parameterizing the `webhook_server` fixture doesn't actually parameterize `hge_server` properly. Subclassing forces this to work correctly. The certificate generation is moved to a Python fixture, so that we don't have to revoke the CA certificate for _test_webhook_insecure.py_; we can just generate a bogus certificate instead. The CA certificate is still generated in the _test-server.sh_ script, as it needs to be installed into the OS certificate store. Interestingly, the CA certificate installation wasn't actually working, because the certificates were written to the wrong location. This didn't cause any failures, as we weren't actually testing this behavior. This is now fixed with the other changes. PR-URL: https://github.com/hasura/graphql-engine-mono/pull/6363 GitOrigin-RevId: 0f277d374daa64f657257ed2a4c2057c74b911db
69 lines
2.1 KiB
Python
69 lines
2.1 KiB
Python
import pytest
|
|
|
|
@pytest.mark.usefixtures('auth_hook', 'per_class_tests_db_state')
|
|
@pytest.mark.admin_secret
|
|
class TestWebhookHeaderCookie(object):
|
|
@classmethod
|
|
def dir(cls):
|
|
return 'webhook/insecure'
|
|
|
|
def test_single_set_cookie_header_in_response(self, hge_ctx):
|
|
query = """
|
|
query allUsers {
|
|
author {
|
|
id
|
|
name
|
|
}
|
|
}
|
|
"""
|
|
|
|
query_obj = {
|
|
"query": query,
|
|
"operationName": "allUsers"
|
|
}
|
|
|
|
headers = {}
|
|
|
|
headers['cookie'] = "Test"
|
|
headers['response-set-cookie-1'] = "__Host-id=1; Secure; Path=/; Domain=example.com"
|
|
|
|
code, resp, respHeaders = hge_ctx.anyq('/v1/graphql', query_obj, headers)
|
|
print("Status Code: ", code)
|
|
print("Response: ", resp)
|
|
print("Headers: ", respHeaders)
|
|
|
|
assert 'Set-Cookie' in respHeaders
|
|
assert respHeaders['Set-Cookie'] == "__Host-id=1; Secure; Path=/; Domain=example.com"
|
|
|
|
def test_duplicate_set_cookie_header_in_response(self, hge_ctx):
|
|
query = """
|
|
query allUsers {
|
|
author {
|
|
id
|
|
name
|
|
}
|
|
}
|
|
"""
|
|
|
|
query_obj = {
|
|
"query": query,
|
|
"operationName": "allUsers"
|
|
}
|
|
|
|
headers = {}
|
|
|
|
headers['cookie'] = "Test"
|
|
headers['response-set-cookie-1'] = "__Host-id=1; Secure; Path=/; Domain=example1.com"
|
|
headers['response-set-cookie-2'] = "__Host-id=2; Secure; Path=/; Domain=example2.com"
|
|
|
|
code, resp, respHeaders = hge_ctx.anyq('/v1/graphql', query_obj, headers)
|
|
print("Status Code: ", code)
|
|
print("Response: ", resp)
|
|
print("Headers: ", respHeaders)
|
|
|
|
assert 'Set-Cookie' in respHeaders
|
|
|
|
# In python, multiple headers with the same key are concatenated with a comma and
|
|
# then sent back in the response. Refer to: https://github.com/psf/requests/issues/4520
|
|
assert respHeaders['Set-Cookie'] == "__Host-id=2; Secure; Path=/; Domain=example2.com, __Host-id=1; Secure; Path=/; Domain=example1.com"
|