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)