class TestHandler(TestCase):
def setUp(self):
self.receiver = Receiver()
self.handler = HTTPServer(None, ('0.0.0.0', 1234), None, self.receiver.callback)
def test_empty_get(self):
self.handler.inject(RequestSample.empty_get)
self.assertEqual(self.receiver.request, {'headers': None, 'params': None, 'postdata': None, 'path': '/'})
def test_query_get(self):
self.handler.inject(RequestSample.query_get)
self.assertEqual(self.receiver.request, {'headers': None, 'params': {'a': ['3'], 'b': [' ']}, 'postdata': None, 'path': '/index'})
def test_post(self):
self.handler.inject(RequestSample.post)
self.assertEqual(self.receiver.request, {'headers': {'Host': 'www.mysite.com', 'Content-Type': 'application/x-www-form-urlencoded', 'Content-Length': '27', 'User-Agent': 'Mozilla/4.0'}, 'params': None, 'postdata': {'password': ['guessme'], 'userid': ['joe']}, 'path': '/login.jsp'})
def test_post_octet(self):
self.handler.inject(RequestSample.post_octet)
self.assertEqual(self.receiver.request, {'headers': {'Host': 'www.mysite.com', 'Content-Type': 'application/octet-stream', 'Content-Length': '27', 'User-Agent': 'Mozilla/4.0'}, 'params': None, 'postdata': 'Arbitrary Content', 'path': '/login.jsp'})
def setUp(self):
self.receiver = Receiver()
self.handler = HTTPServer(None, ('0.0.0.0', 1234), None, self.receiver.callback)
