Extensions Docs Server: Integration testing

chrome/common/extensions/docs/server2/appengine_url_fetcher.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.
 
-from google.appengine.api import urlfetch
-
 from future import Future
 
 class _AsyncFetchDelegate(object):
   def __init__(self, rpc):
     self._rpc = rpc
 
   def Get(self):
     self._rpc.wait()
     return self._rpc.get_result()
 
 class AppEngineUrlFetcher(object):
   """A wrapper around the App Engine urlfetch module that allows for easy
   async fetches.
   """
-  def __init__(self, base_path):
+  def __init__(self, base_path, urlfetch):
     self._base_path = base_path
+    self._urlfetch = urlfetch
 
   def Fetch(self, url):
     """Fetches a file synchronously.
     """
-    return urlfetch.fetch(self._base_path + '/' + url)
+    return self._urlfetch.fetch(self._base_path + '/' + url)
 
   def FetchAsync(self, url):
     """Fetches a file asynchronously, and returns a Future with the result.
     """
-    rpc = urlfetch.create_rpc()
-    urlfetch.make_fetch_call(rpc, self._base_path + '/' + url)
+    rpc = self._urlfetch.create_rpc()
+    self._urlfetch.make_fetch_call(rpc, self._base_path + '/' + url)
     return Future(delegate=_AsyncFetchDelegate(rpc))