Extensions Docs Server: Testing GithubFileSystem and cron jobs

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

 # 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 logging
 import os
 from StringIO import StringIO
 import sys
 
 from appengine_wrappers import webapp
@@ skipping 141 matching lines @@
     if real_path.strip('/') == 'apps':
       real_path = 'apps/index.html'
     if real_path.strip('/') == 'extensions':
       real_path = 'extensions/index.html'
     _CleanBranches()
     _GetInstanceForBranch(channel_name, self._local_path).Get(real_path,
                                                               self.request,
                                                               self.response)
 
   def _Render(self, files, branch):
+    original_response = self.response
     for f in files:
       path = branch + f.split(PUBLIC_TEMPLATE_PATH)[-1]
       self.request = _MockRequest(path)
       self.response = _MockResponse()
       self._HandleGet(path)
+    self.response = original_response
 
   def _HandleCron(self, path):
     branch = path.split('/')[-1]
     logging.info('Running cron job for %s.' % branch)
     branch_memcache = InMemoryObjectStore(branch)
     file_system = _CreateMemcacheFileSystem(branch, branch_memcache)
     builder = FileSystemCache.Builder(file_system, branch_memcache)
     render_cache = builder.build(lambda x: self._Render(x, branch),
                                  fs_cache.RENDER)
     render_cache.GetFromFileListing(PUBLIC_TEMPLATE_PATH)
+    self.response.out.write('Success')
 
   def get(self):
     path = self.request.path
     if path.startswith('/cron'):
       self._HandleCron(path)
     else:
       # Redirect paths like "directory" to "directory/". This is so relative
       # file paths will know to treat this as a directory.
       if os.path.splitext(path)[1] == '' and path[-1] != '/':
         self.redirect(path + '/')
       path = path.replace('/chrome/', '')
       path = path.strip('/')
       self._HandleGet(path)