Extensions Docs Server: BranchUtility not fetching branch numbers correctly (fixed)

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 = [
 ]
 
+def _FakeGetBranchNumberForChannelName(channel_name):
+  return 'local'
+handler.BRANCH_UTILITY.GetBranchNumberForChannelName = (
+    _FakeGetBranchNumberForChannelName)

[not at google - send to devlin, 2012/08/13 00:53:00]

I alluded to something like this in a bug report I just replied to.

A more integration-test kind of thing to do would be to set up a fake server
which knows about the protocol. I.e. a fake omaha server kind of thing (you can
just make it return canned responses that you get from omaha).

The trick is getting the server to actually point to that "server" instead of
the real server.

I'd try doing this by putting the knowledge in the fake urlfetch implementation
(in appengine_wrappers). Rather than that returning a FakeUrlFetch, allow it to
be configured. I'm imagining an interface like


--- integration_test.py

import appengine_wrappers
import url_constants

class FakeOmahaProxy(object):
  def fetch(self, url):
    return "insert_canned_response_here"

class FakeSubversionServer(object):
  _base_pattern = re.compile(r"chrome/common/extensions/(.*)$")
  def fetch(self, url):
    # Only the interesting part of the path.
    # ... whatever the local filesystem fetch is.
    return local_fetch('chrome/common/extensions/%s' %
                       _base_pattern.match(url).groups()[0])

appengine_wrappers.ConfigureFakeUrlFetch({
  url_constants.OMAHA_PROXY_URL:  FakeOmahaProxy(),
  "%s/*" % url_constants.SVN_URL: FakeSubversionServer(),
})


--- appengine_wrappers.py

def ConfigureFakeUrlFetch(configuration):
  FAKE_URL_FETCHER_CONFIGURATION = configuration

try:
  import ...
except ImportError:
  class FakeUrlFetch(object):
    def fetch(self, url):
      # TODO: match globs.
      return FAKE_URL_FETCHER_CONFIGURATION[url].fetch(url)
    def create_rpc(self):
      # Return object that eventually just calls fetch()
      # so that the fake fetch implementations can handle it.
    def make_fetch_call(self):
      # ditto

[cduvall, 2012/08/13 21:45:04]

Done.

+
 class _MockResponse(object):
   def __init__(self):
     self.status = 200
     self.out = StringIO()
 
   def set_status(self, status):
     self.status = status
 
 class _MockRequest(object):
   def __init__(self, path):
@@ skipping 40 matching lines @@
     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()