Extensions Docs Server: Preview server and more integration tests

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

 #!/usr/bin/env python
 # Copyright (c) 2012 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 os
 from StringIO import StringIO
 import unittest
 
+import appengine_memcache as memcache
+import handler
 from handler import Handler
 
 KNOWN_FAILURES = [
   'webstore.html',
 ]
 
-class _MockResponse:
+class _MockResponse(object):
   def __init__(self):
     self.status = 200
     self.out = StringIO()
 
   def set_status(self, status):
     self.status = status
 
-class _MockRequest:
+class _MockRequest(object):
   def __init__(self, path):
     self.headers = {}
     self.path = path
 
 class IntegrationTest(unittest.TestCase):
   def testAll(self):
     for filename in os.listdir(os.path.join('templates', 'public')):
       if filename in KNOWN_FAILURES or filename.startswith('.'):
         continue
       request = _MockRequest(filename)
       response = _MockResponse()
       Handler(request, response, local_path='../..').get()
       self.assertEqual(200, response.status)
       self.assertTrue(response.out.getvalue())
 
   def test404(self):
     request = _MockRequest('junk.html')
     bad_response = _MockResponse()
     Handler(request, bad_response, local_path='../..').get()
     self.assertEqual(404, bad_response.status)
     self.assertTrue(bad_response.out.getvalue())
 
+  def testWarmupRequest(self):
+    for branch in ['dev', 'trunk', 'beta', 'stable']:
+      handler.BRANCH_UTILITY_MEMCACHE.Set(
+          branch + '.' +  handler.OMAHA_PROXY_URL,
+          'local',
+          memcache.MEMCACHE_BRANCH_UTILITY)
+    request = _MockRequest('_ah/warmup')
+    response = _MockResponse()
+    Handler(request, response, local_path='../..').get()
+    self.assertEqual(200, response.status)
+    # Test that the pages were rendered by checking the size of the output.
+    # In python 2.6 there is no 'assertGreater' method.
+    self.assertTrue(len(response.out.getvalue()) > 500000)
+
 if __name__ == '__main__':
   unittest.main()