darkhttpd/devel/test_forward_all.py
2016-01-23 20:05:02 +11:00

28 lines
979 B
Python
Executable File

#!/usr/bin/env python
# This is run by the "run-tests" script.
import unittest
from test import TestHelper, Conn, parse
class TestForwardAll(TestHelper):
def test_forward_root(self):
resp = Conn().get("/", req_hdrs = { "Host": "not-example.com" })
status, hdrs, body = parse(resp)
self.assertContains(status, "301 Moved Permanently")
expect = "http://catchall.example.com/"
self.assertEquals(hdrs["Location"], expect)
self.assertContains(body, expect)
def test_forward_relative(self):
resp = Conn().get("/foo/bar",
req_hdrs = { "Host": "still-not.example.com" })
status, hdrs, body = parse(resp)
self.assertContains(status, "301 Moved Permanently")
expect = "http://catchall.example.com/foo/bar"
self.assertEquals(hdrs["Location"], expect)
self.assertContains(body, expect)
if __name__ == '__main__':
unittest.main()
# vim:set ts=4 sw=4 et: