Now some data to prevent HEAD parsing from reading the link near the end.
%snear the end """ % (("0123456789ABCDEF" * 4 + "\n") * 61) REFERER_TEST_HTML = """\
This page exists to test the Referer functionality of \ mechanize.
Here\ is a link to a page that displays the Referer header. """ BASIC_AUTH_PAGE = """
Hello, basic auth world.
""" DIGEST_AUTH_PAGE = """
Hello, digest auth world.
""" class TestHTTPUser(object): """ Test avatar implementation for http auth with cred """ isLeaf = True def render(self, request): return self.template.encode('utf-8') def __init__(self, template, username): self.template = template self.username = username @implementer(portal.IRealm) class TestAuthRealm(object): """ Test realm that supports the IHTTPUser interface """ def __init__(self, template=BASIC_AUTH_PAGE): self.template = template def requestAvatar(self, avatarId, mind, *interfaces): if IResource in interfaces: if avatarId == checkers.ANONYMOUS: return (IResource, TestHTTPUser(self.template, 'anonymous'), lambda: None) return (IResource, TestHTTPUser(self.template, avatarId), lambda: None) raise NotImplementedError("Only IResource interface is supported") class Page(resource.Resource): def __init__(self, text='', leaf=False, content_type='text/html'): self.isLeaf = leaf self.content_type = content_type self.text = text resource.Resource.__init__(self) def getChild(self, path, request): if not path: return self return resource.Resource.getChild(self, path, request) def render(self, request): request.setResponseCode(http.OK) request.setHeader('content-type', self.content_type) return self.text.encode('utf-8') class DynamicPage(resource.Resource): isLeaf = True def getChild(self, path, request): if not path: return self return resource.Resource.getChild(self, path, request) def render(self, request): request.setResponseCode(http.OK) request.setHeader('content-type', 'text/html') year_plus_one = time.localtime(time.time())[0] + 1 expires = "09-Nov-%d 23:12:40 GMT" % (year_plus_one,) request.addCookie('foo', 'bar', expires=expires) request.addCookie('sessioncookie', 'spam\n') html = ( "
Referer:
{}".format(
request.getHeader(b'referer'))
html += "Received cookies:
" html += ""
html += request.getHeader('cookie') or ''
html += ""
if request.getCookie(b'foo'):
html += "Your browser supports cookies!" if request.getCookie(b'sessioncookie'): html += "
Received session cookie" html += '' return html.encode('utf-8') def _make_page(parent, name, text, content_type, wrapper, leaf=False): page = Page(text=text, leaf=leaf, content_type=content_type) name = name.encode('utf-8') parent.putChild(name, wrapper(page)) return page def make_page(parent, name, text, content_type="text/html", wrapper=lambda page: page): return _make_page(parent, name, text, content_type, wrapper, leaf=False) def make_leaf_page(parent, name, text, content_type="text/html", wrapper=lambda page: page): return _make_page(parent, name, text, content_type, wrapper, leaf=True) def make_redirect(parent, name, location_relative_ref): redirect = Redirect(location_relative_ref) name = name.encode('utf-8') parent.putChild(name, redirect) return redirect def require_basic_auth(resource): p = portal.Portal(TestAuthRealm()) c = checkers.InMemoryUsernamePasswordDatabaseDontUse(john=b'john') p.registerChecker(c) cred_factory = BasicCredentialFactory("Basic Auth protected area") return HTTPAuthSessionWrapper(p, [cred_factory]) def require_digest_auth(resource): p = portal.Portal(TestAuthRealm(DIGEST_AUTH_PAGE)) c = checkers.InMemoryUsernamePasswordDatabaseDontUse( digestuser=b'digestuser') p.registerChecker(c) cred_factory = DigestCredentialFactory( "md5", b"Digest Auth protected area") return HTTPAuthSessionWrapper(p, [cred_factory]) def parse_options(args): parser = optparse.OptionParser() parser.add_option("--log", action="store_true") options, remaining_args = parser.parse_args(args) options.port = int(remaining_args[0]) if remaining_args else 8000 return options def main(argv): options = parse_options(argv[1:]) if options.log: log.startLogging(sys.stdout) # This is supposed to match the SF site so it's easy to run a functional # test over the internet and against Apache. # TODO: Remove bizarre structure and strings expected by functional tests. root = Page(text=ROOT_HTML) mechanize = make_page(root, "mechanize", MECHANIZE_HTML) make_leaf_page(root, "robots.txt", "User-Agent: *\nDisallow: /norobots", "text/plain") make_leaf_page(root, "robots", "Hello, robots.", "text/plain") make_leaf_page(root, "norobots", "Hello, non-robots.", "text/plain") root.putChild(b'dynamic', DynamicPage()) test_fixtures = make_page(root, "test_fixtures", # satisfy stupid assertions in functional tests html("Python bits", extra_content="GeneralFAQ.html")) make_leaf_page(test_fixtures, "cctest2.txt", "Hello ClientCookie functional test suite.", "text/plain") make_leaf_page(test_fixtures, "referertest.html", REFERER_TEST_HTML) make_leaf_page(test_fixtures, "mechanize_reload_test.html", RELOAD_TEST_HTML) make_redirect(root, "redirected", b"/doesnotexist") make_redirect(root, "redirected_good", b"/test_fixtures") example_html = open(os.path.join( "examples", "forms", "example.html")).read() make_leaf_page(mechanize, "example.html", example_html) make_page(root, "basic_auth", BASIC_AUTH_PAGE, wrapper=require_basic_auth) make_page(root, "digest_auth", DIGEST_AUTH_PAGE, wrapper=require_digest_auth) make_leaf_page( root, "gzip", open(__file__).read(), "text/plain", wrapper=gzip_wrapper) site = server.Site(root) reactor.listenTCP(options.port, site) reactor.run() if __name__ == "__main__": main(sys.argv)