mirror of
https://github.com/mozilla-services/syncstorage-rs.git
synced 2026-05-05 04:06:16 +02:00
835bc58555
Some checks failed
Glean probe-scraper / glean-probe-scraper (push) Has been cancelled
test: refactor of tokenserver python tests to support postgres
47 lines
1.5 KiB
Python
Executable File
47 lines
1.5 KiB
Python
Executable File
from wsgiref.simple_server import make_server as _make_server
|
|
from pyramid.config import Configurator
|
|
from pyramid.response import Response
|
|
from pyramid.view import view_config
|
|
import json
|
|
import os
|
|
|
|
|
|
@view_config(route_name="mock_oauth_verify", renderer="json")
|
|
def _mock_oauth_verify(request):
|
|
body = json.loads(request.json_body["token"])
|
|
|
|
return Response(
|
|
json=body["body"], content_type="application/json", status=body["status"]
|
|
)
|
|
|
|
|
|
# The PyFxA OAuth client makes a request to the FxA OAuth server for its
|
|
# current public RSA key. While the client allows us to pass in a JWK to
|
|
# prevent this request from happening, mocking the endpoint is simpler.
|
|
@view_config(route_name="mock_oauth_jwk", renderer="json")
|
|
def _mock_oauth_jwk(request):
|
|
return {"keys": [{"fake": "RSA key"}]}
|
|
|
|
|
|
def make_server(host, port):
|
|
with Configurator() as config:
|
|
config.add_route("mock_oauth_verify", "/v1/verify")
|
|
config.add_view(
|
|
_mock_oauth_verify, route_name="mock_oauth_verify", renderer="json"
|
|
)
|
|
|
|
config.add_route("mock_oauth_jwk", "/v1/jwks")
|
|
config.add_view(_mock_oauth_jwk, route_name="mock_oauth_jwk", renderer="json")
|
|
app = config.make_wsgi_app()
|
|
|
|
return _make_server(host, port, app)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
host = os.environ.get("MOCK_FXA_SERVER_HOST", "localhost")
|
|
port = os.environ.get("MOCK_FXA_SERVER_PORT", 6000)
|
|
|
|
with make_server(host, int(port)) as httpd:
|
|
print(f"Running mock FxA server on {host}:{port}")
|
|
httpd.serve_forever()
|