| Index: chrome/common/extensions/docs/server2/render_servlet_test.py
|
| diff --git a/chrome/common/extensions/docs/server2/render_servlet_test.py b/chrome/common/extensions/docs/server2/render_servlet_test.py
|
| index 75c91feb01ac8363a4689fb368af216595148539..f1886716dcfc3f609816dc75706e807e2e72523a 100755
|
| --- a/chrome/common/extensions/docs/server2/render_servlet_test.py
|
| +++ b/chrome/common/extensions/docs/server2/render_servlet_test.py
|
| @@ -8,23 +8,25 @@ import unittest
|
| from local_file_system import LocalFileSystem
|
| from render_servlet import RenderServlet
|
| from server_instance import ServerInstance
|
| -from servlet import Request
|
| +from servlet import Request, Response
|
| from test_util import DisableLogging, ReadFile
|
|
|
| class _RenderServletDelegate(RenderServlet.Delegate):
|
| - def CreateServerInstanceForChannel(self, channel):
|
| + def CreateServerInstance(self):
|
| return ServerInstance.ForTest(LocalFileSystem.Create())
|
|
|
| class RenderServletTest(unittest.TestCase):
|
| def testExtensionAppRedirect(self):
|
| request = Request.ForTest('storage.html')
|
| - response = RenderServlet(request, _RenderServletDelegate()).Get()
|
| - self.assertEqual(302, response.status)
|
| + self.assertEqual(
|
| + Response.Redirect('/extensions/storage.html', permanent=False),
|
| + RenderServlet(request, _RenderServletDelegate()).Get())
|
|
|
| - def testDefaultChannel(self):
|
| + def testChannelRedirect(self):
|
| request = Request.ForTest('stable/extensions/storage.html')
|
| - response = RenderServlet(request, _RenderServletDelegate()).Get()
|
| - self.assertEqual(302, response.status)
|
| + self.assertEqual(
|
| + Response.Redirect('/extensions/storage.html', permanent=True),
|
| + RenderServlet(request, _RenderServletDelegate()).Get())
|
|
|
| @DisableLogging('warning')
|
| def testNotFound(self):
|
|
|