Ticket #1203: test_reponse.py

#!/usr/bin/env python
# -*- coding: UTF-8 -*-

import unittest

import cherrypy
import turbogears
from turbogears import controllers, redirect, testutil

class RedirectRoot(controllers.RootController):
    def index(self):
        raise redirect("foo")
    index = turbogears.expose()(index)

class TestControllers(unittest.TestCase):
    def setUp(self):
        cherrypy.root = None
        cherrypy.tree.mount_points = {}
        cherrypy.tree.mount(RedirectRoot(), "/")
        self.root = cherrypy.root

    def tearDown(self):
        cherrypy.root = None
        cherrypy.tree.mount_points = {}

    
    def test_index(self):
        "Test that call_with_request is usable if controller uses redirect"
        try:
            testutil.call(self.root.index)
            self.fail("no redirect exception raised")
        except cherrypy.HTTPRedirect, e:
            assert e.status in [302, 303]
            self.assertEquals(1, len(e.urls))
            assert e.urls[0].endswith("foo")

if __name__ == "__main__":
    unittest.main()