forked from ctb/cse491-serverz
/
test_server.py
146 lines (130 loc) · 5.7 KB
/
test_server.py
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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
import server
class FakeConnection(object):
"""
A fake connection class that mimics a real TCP socket for the purpose
of testing socket I/O.
"""
def __init__(self, to_recv):
self.to_recv = to_recv
self.sent = ""
self.is_closed = False
def recv(self, n):
if n > len(self.to_recv):
r = self.to_recv
self.to_recv = ""
return r
r, self.to_recv = self.to_recv[:n], self.to_recv[n:]
return r
def send(self, s):
self.sent += s
def close(self):
self.is_closed = True
# Test a basic GET call.
def test_handle_index_get():
conn = FakeConnection("GET / HTTP/1.0\r\n\r\n")
header = 'HTTP/1.0 200 OK\r\n' + \
'Content-type: text/html\r\n' + \
'\r\n'
body = "<h1>Hello, world.</h1>This is hoffm386's Web server.\n" + \
" <ul>\n" + \
" <li><a href='/content'>Content</a></li>\n" + \
" <li><a href='/file'>File</a></li>\n" + \
" <li><a href='/image'>Image</a></li>\n" + \
" <li><a href='/form'>Form</a></li>\n" + \
" </ul>\n\n"
server.handle_connection(conn, 8500)
assert conn.sent.startswith(header), 'Got: %s' % (repr(conn.sent),)
assert conn.sent.find(body) != -1, 'Got %s' % (repr(conn.sent),)
def test_handle_content_get():
conn = FakeConnection("GET /content HTTP/1.0\r\n\r\n")
header = 'HTTP/1.0 200 OK\r\n' + \
'Content-type: text/html\r\n' + \
'\r\n'
body = '<h1>Content</h1>\n'
body += ' This page will contain "content"'
server.handle_connection(conn, 8500)
assert conn.sent.startswith(header), 'Got: %s' % (repr(conn.sent),)
assert conn.sent.find(body) != -1, 'Got %s' % (repr(conn.sent),)
def test_handle_file_get():
conn = FakeConnection("GET /file HTTP/1.0\r\n\r\n")
header = 'HTTP/1.0 200 OK\r\n' + \
'Content-type: text/plain\r\n' + \
'\r\n'
server.handle_connection(conn, 8500)
assert conn.sent.startswith(header), 'Got: %s' % (repr(conn.sent),)
def test_handle_image_get():
conn = FakeConnection("GET /image HTTP/1.0\r\n\r\n")
header = 'HTTP/1.0 200 OK\r\n' + \
'Content-type: image/jpeg\r\n' + \
'\r\n'
server.handle_connection(conn, 8500)
assert conn.sent.startswith(header), 'Got: a long string'
def test_handle_form_get():
conn = FakeConnection("GET /form HTTP/1.0\r\n\r\n")
header = 'HTTP/1.0 200 OK\r\n' + \
'Content-type: text/html\r\n' + \
'\r\n'
body = " <div>\n" + \
" <h1>GET form</h1>\n" + \
" <form action='/submit' method='GET'>\n" + \
" <input type='text' name='firstname'>\n" + \
" <input type='text' name='lastname'>\n" + \
" <input type='submit' value='Submit'>\n" + \
" </form>\n" + \
" </div>\n" + \
" <div>\n" + \
" <h1>POST form</h1>\n" + \
" <form action='/submit' method='POST'>\n" + \
" <input type='text' name='firstname'>\n" + \
" <input type='text' name='lastname'>\n" + \
" <input type='submit' value='Submit'>\n" + \
" </form>\n" + \
" </div>\n"
server.handle_connection(conn, 8500)
assert conn.sent.startswith(header), 'Got: %s' % (repr(conn.sent),)
assert conn.sent.find(body) != -1, 'Got %s' % (repr(conn.sent),)
def test_404_get():
conn = FakeConnection("GET /asdf HTTP/1.0\r\n\r\n")
header = 'HTTP/1.0 404 Not Found\r\n' + \
'Content-type: text/html\r\n' + \
'\r\n'
body = " <h1>404</h1>\n" + \
" <p>The page at /asdf could not be found</p>\n" + \
" <p>To return to the home page, click <a href='/'>here</a></p>\n\n"
server.handle_connection(conn, 8500)
assert conn.sent.startswith(header), 'Got: %s' % (repr(conn.sent),)
assert conn.sent.find(body) != -1, 'Got %s' % (repr(conn.sent),)
def test_post_urlencoded():
first = "Erin"
last = "Hoffman"
fake_str = "POST /submit HTTP/1.1\r\n" + \
"Content-Length: 31\r\n" + \
"Content-Type: application/x-www-form-urlencoded\r\n\r\n" + \
"firstname={0}&lastname={1}\r\n".format(first, last)
conn = FakeConnection(fake_str)
header = 'HTTP/1.0 200 OK\r\n' + \
'Content-type: text/html\r\n' + \
'\r\n'
body = " <h1>Form Submission</h1>\n" + \
" Hello Mr. Erin Hoffman.\n\n"
server.handle_connection(conn, 8500)
assert conn.sent.startswith(header), 'Got: %s' % (repr(conn.sent),)
assert conn.sent.find(body) != -1, 'Got %s' % (repr(conn.sent),)
def test_submit_post():
conn_string = "POST /submit HTTP/1.1\r\n" + \
"Host: arctic.cse.msu.edu:8987\r\n" + \
"Accept: text/html,application/xhtml+xml,application/xml;" + \
"q=0.9,*/*;q=0.8\r\n" + \
"Referer: http://arctic.cse.msu.edu:8987/form\r\n" + \
"Content-Type: application/x-www-form-urlencoded\r\n" + \
"Content-Length: 29\r\n\r\n" + \
"firstname=John&lastname=Smith"
conn = FakeConnection(conn_string)
header = 'HTTP/1.0 200 OK\r\n' + \
'Content-type: text/html\r\n' + \
'\r\n'
body = " <h1>Form Submission</h1>\n" + \
" Hello Mr. John Smith.\n\n"
server.handle_connection(conn, 8500)
assert conn.sent.startswith(header), 'Got: %s' % (repr(conn.sent),)
assert conn.sent.find(body) != -1, 'Got %s' % (repr(conn.sent),)