summaryrefslogtreecommitdiffstats
path: root/testing/web-platform/tests/shared-storage/resources/credentials-test-helper.py
blob: 46fc0ea6fb2a99877dbb6b69cb3da25a19f490fd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
def main(request, response):
    """
    A handler that does either one of the following based on the provided
    "action" parameter:
    1) action="store-cookie": Stores the provided token and the request cookie
    to the stash, and returns a regular module script content.
    2) action="get-cookie": Retrieves and returns the content stored in the
    stash at the provided token.
    """
    token = request.GET[b"token"]
    action = request.GET[b"action"]

    response.status = 200
    response.headers.append(b"Content-Type", b"text/javascript")

    if b"access_control_allow_credentials_header" in request.GET:
      response.headers.append(b"Access-Control-Allow-Credentials", request.GET[b"access_control_allow_credentials_header"])

    if b"access_control_allow_origin_header" in request.GET:
      response.headers.append(b"Access-Control-Allow-Origin", request.GET[b"access_control_allow_origin_header"])

    if action == b"store-cookie":
      cookie = request.headers.get(b"Cookie", b"NO_COOKIE_HEADER")
      request.server.stash.put(token, cookie)
      return b""
    else:
      assert action == b"get-cookie"
      return request.server.stash.take(token)