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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
from urllib.parse import parse_qs
from wptserve.utils import isomorphic_encode
def setNoCacheAndCORSHeaders(request, response):
"""Set Cache-Control, CORS and Content-Type headers appropriate for the cookie tests."""
headers = [(b"Content-Type", b"application/json"),
(b"Access-Control-Allow-Credentials", b"true")]
origin = b"*"
if b"origin" in request.headers:
origin = request.headers[b"origin"]
headers.append((b"Access-Control-Allow-Origin", origin))
#headers.append(("Access-Control-Allow-Credentials", "true"))
headers.append((b"Cache-Control", b"no-cache"))
headers.append((b"Expires", b"Fri, 01 Jan 1990 00:00:00 GMT"))
return headers
def makeCookieHeader(name, value, otherAttrs):
"""Make a Set-Cookie header for a cookie with the name, value and attributes provided."""
def makeAV(a, v):
if None == v or b"" == v:
return a
if isinstance(v, int):
return b"%s=%i" % (a, v)
else:
return b"%s=%s" % (a, v)
# ensure cookie name is always first
attrs = [b"%s=%s" % (name, value)]
attrs.extend(makeAV(a, v) for (a, v) in otherAttrs.items())
return (b"Set-Cookie", b"; ".join((attrs)))
def makeDropCookie(name, secure):
attrs = {b"max-age": 0, b"path": b"/"}
if secure:
attrs[b"secure"] = b""
return makeCookieHeader(name, b"", attrs)
def readParameter(request, paramName, requireValue):
"""Read a parameter from the request. Raise if requireValue is set and the
parameter has an empty value or is not present."""
params = parse_qs(request.url_parts.query)
param = params[paramName][0].strip()
if len(param) == 0:
raise Exception(u"Empty or missing name parameter.")
return isomorphic_encode(param)
def readCookies(request):
"""Read the cookies from the client present in the request."""
cookies = {}
for key in request.cookies:
for cookie in request.cookies.get_list(key):
# do we care we'll clobber cookies here? If so, do we
# need to modify the test to take cookie names and value lists?
cookies[key] = cookie.value
return cookies
|
