File: setup_sharded_server_state.py

package info (click to toggle)
firefox 149.0-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 4,767,760 kB
  • sloc: cpp: 7,416,064; javascript: 6,752,859; ansic: 3,774,850; python: 1,250,473; xml: 641,578; asm: 439,191; java: 186,617; sh: 56,634; makefile: 18,856; objc: 13,092; perl: 12,763; pascal: 5,960; yacc: 4,583; cs: 3,846; lex: 1,720; ruby: 1,002; php: 436; lisp: 258; awk: 105; sql: 66; sed: 53; csh: 10; exp: 6
file content (27 lines) | stat: -rw-r--r-- 1,083 bytes parent folder | download | duplicates (4) 
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
 
import json
import importlib
session_manager = importlib.import_module('device-bound-session-credentials.session_manager')

def main(request, response):
    request_body = json.loads(request.body.decode('utf-8'))

    test_id = request_body.get("testId")
    if test_id is None:
        test_id = session_manager.initialize_test()

    # subdomain-registration.https.html does registration on a
    # subdomain. Without the Domain attribute, the test_id cookie won't
    # be available on the subdomain.
    cookie_attributes = f"Domain={request.url_parts.hostname}"

    # Cross-site tests (e.g. allowed-refresh-initiators.https.html) require a
    # SameSite=None cookie, which must also be Secure. But
    # not-secure-connection.html cannot have a Secure cookie, so we need to make
    # the attributes conditional on the test.
    cross_site = request_body.get("crossSite")
    if cross_site is not None and cross_site:
        cookie_attributes += ";SameSite=None;Secure"

    headers = [("Set-Cookie", f"test_id={test_id};{cookie_attributes}")]

    return (200, headers, "")