summaryrefslogtreecommitdiffstats
path: root/testing/web-platform/tests/common/security-features/scope/util.py
blob: da5aacf35e99a4bd9de9cb5c3e96e8b1804da054 (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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import os

from wptserve.utils import isomorphic_decode

def get_template(template_basename):
  script_directory = os.path.dirname(os.path.abspath(isomorphic_decode(__file__)))
  template_directory = os.path.abspath(
      os.path.join(script_directory, u"template"))
  template_filename = os.path.join(template_directory, template_basename)

  with open(template_filename, "r") as f:
    return f.read()


def __noop(request, response):
  return u""


def respond(request,
            response,
            status_code=200,
            content_type=b"text/html",
            payload_generator=__noop,
            cache_control=b"no-cache; must-revalidate",
            access_control_allow_origin=b"*",
            maybe_additional_headers=None):
  response.add_required_headers = False
  response.writer.write_status(status_code)

  if access_control_allow_origin != None:
    response.writer.write_header(b"access-control-allow-origin",
                                 access_control_allow_origin)
  response.writer.write_header(b"content-type", content_type)
  response.writer.write_header(b"cache-control", cache_control)

  additional_headers = maybe_additional_headers or {}
  for header, value in additional_headers.items():
    response.writer.write_header(header, value)

  response.writer.end_headers()

  payload = payload_generator()
  response.writer.write(payload)