Skip to content

Commit

Permalink
Show file tree
Hide file tree
Showing 2 changed files with 18 additions and 1 deletion.
2 changes: 1 addition & 1 deletion pigwig/request_response.py
Expand Up @@ -108,7 +108,7 @@ def __init__(self, body=None, code=200, content_type='text/plain', location=None
headers.extend(extra_headers)
self.headers = headers

def set_cookie(self, key, value, domain=None, path=None, expires=None, max_age=None, secure=False, http_only=False):
def set_cookie(self, key, value, domain=None, path='/', expires=None, max_age=None, secure=False, http_only=False):
'''
adds a Set-Cookie header
Expand Down
17 changes: 17 additions & 0 deletions pigwig/tests/test_request_response.py
Expand Up @@ -22,6 +22,23 @@ def test_json(self):
self.assertGreater(len(chunks), 1)
self.assertEqual(b''.join(chunks), json.dumps(big_obj).encode())

def test_cookie(self):
app = PigWig([])
r = Response()
r.set_cookie('cow', 'moo')
r.set_cookie('duck', 'quack', path='/pond')

cookies = http.cookies.SimpleCookie()
for header, value in r.headers:
if header == 'Set-Cookie':
cookies.load(value)

req = Request(app, None, None, None, None, None, cookies, None)
self.assertEqual(req.cookies['cow'].value, 'moo')
self.assertEqual(req.cookies['cow']['path'], '/')
self.assertEqual(req.cookies['duck'].value, 'quack')
self.assertEqual(req.cookies['duck']['path'], '/pond')

def test_secure_cookie(self):
app = PigWig([], cookie_secret=b'a|b')
req = Request(app, None, None, None, None, None, None, None)
Expand Down

0 comments on commit 2a23103

Please sign in to comment.