Docserver: achieve online vs offline (cron vs instance) behaviour at the object

chrome/common/extensions/docs/server2/render_servlet_test.py

+#!/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()