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 |
new file mode 100755 |
index 0000000000000000000000000000000000000000..46061d4869ccaf75370a9da5636c7e842edaa9bc |
--- /dev/null |
+++ b/chrome/common/extensions/docs/server2/render_servlet_test.py |
@@ -0,0 +1,71 @@ |
+#!/usr/bin/env python |
+# Copyright 2013 The Chromium Authors. All rights reserved. |
+# Use of this source code is governed by a BSD-style license that can be |
+# found in the LICENSE file. |
+ |
+import unittest |
+ |
+from local_file_system import LocalFileSystem |
+from render_servlet import RenderServlet |
+from server_instance import ServerInstance |
+from servlet import Request |
+from test_util import DisableLogging, ReadFile |
+ |
+def _CreateServerInstance(): |
+ return ServerInstance.CreateForTest(LocalFileSystem()) |
+ |
+class RenderServletTest(unittest.TestCase): |
+ def testExtensionAppRedirect(self): |
+ request = Request('storage.html', {}) |
+ response = RenderServlet(request).Get(_CreateServerInstance()) |
+ self.assertEqual(302, response.status) |
+ |
+ def testDefaultChannel(self): |
+ request = Request('stable/extensions/storage.html', {}) |
+ response = RenderServlet(request).Get(_CreateServerInstance()) |
+ self.assertEqual(302, response.status) |
+ |
+ @DisableLogging('warning') |
+ def testNotFound(self): |
+ request = Request('extensions/not_found.html', {}) |
+ response = RenderServlet(request).Get(_CreateServerInstance()) |
+ self.assertEqual(404, response.status) |
+ |
+ def testSampleFile(self): |
+ sample_file = 'extensions/talking_alarm_clock/background.js' |
+ request = Request('extensions/examples/%s' % sample_file, {}) |
+ response = RenderServlet(request).Get(_CreateServerInstance()) |
+ self.assertEqual(200, response.status) |
+ self.assertEqual('application/javascript', |
+ response.headers.get('content-type')) |
+ self.assertEqual(ReadFile('docs/examples/%s' % sample_file), |
+ response.content.ToString()) |
+ |
+ def testSampleZip(self): |
+ sample_dir = 'extensions/talking_alarm_clock' |
+ request = Request('extensions/examples/%s.zip' % sample_dir, {}) |
+ response = RenderServlet(request).Get(_CreateServerInstance()) |
+ self.assertEqual(200, response.status) |
+ self.assertEqual('application/zip', response.headers.get('content-type')) |
+ |
+ def testStaticFile(self): |
+ static_file = 'css/site.css' |
+ request = Request('static/%s' % static_file, {}) |
+ response = RenderServlet(request).Get(_CreateServerInstance()) |
+ self.assertEqual(200, response.status) |
+ self.assertEqual('text/css', response.headers.get('content-type')) |
+ self.assertEqual(ReadFile('docs/static/%s' % static_file), |
+ response.content.ToString()) |
+ |
+ def testHtmlTemplate(self): |
+ html_file = 'extensions/storage.html' |
+ request = Request(html_file, {}) |
+ response = RenderServlet(request).Get(_CreateServerInstance()) |
+ self.assertEqual(200, response.status) |
+ self.assertEqual('text/html', response.headers.get('content-type')) |
+ # Can't really test rendering all that well. |
+ self.assertTrue(len(response.content) > |
+ len(ReadFile('docs/templates/public/%s' % html_file))) |
+ |
+if __name__ == '__main__': |
+ unittest.main() |