blob: 71f7bc9115daf01f07f0fba98edeade39c2198d8 (
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
44
45
46
47
48
49
50
|
#! /usr/bin/env python
# -*- coding: utf-8 -*-
# vi:ts=4:et
from . import localhost
import pycurl
import unittest
import time as _time
from . import appmanager
from . import util
setup_module, teardown_module = appmanager.setup(('app', 8380))
class HeaderCbTest(unittest.TestCase):
def setUp(self):
self.curl = util.DefaultCurl()
self.header_lines = []
def tearDown(self):
self.curl.close()
def header_function(self, line):
self.header_lines.append(line.decode())
def test_get(self):
self.curl.setopt(pycurl.URL, 'http://%s:8380/success' % localhost)
sio = util.BytesIO()
self.curl.setopt(pycurl.WRITEFUNCTION, sio.write)
self.curl.setopt(pycurl.HEADERFUNCTION, self.header_function)
self.curl.perform()
self.assertEqual('success', sio.getvalue().decode())
assert len(self.header_lines) > 0
self.assertEqual("HTTP/1.0 200 OK\r\n", self.header_lines[0])
# day of week
# important: must be in utc
todays_day = _time.strftime('%a', _time.gmtime())
# Date: Sun, 03 Mar 2013 05:38:12 GMT\r\n
self.check('Date: %s' % todays_day)
# Server: WSGIServer/0.1 Python/2.7.3\r\n
self.check('Server: WSGIServer')
self.check('Content-Length: 7')
self.check('Content-Type: text/html')
def check(self, wanted_text):
for line in self.header_lines:
if wanted_text in line:
return
assert False, "%s not found in header lines" % wanted_text
|
