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

chrome/common/extensions/docs/server2/branch_utility.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 json
 import logging
 import operator
 
 from appengine_wrappers import GetAppVersion
-from object_store_creator import ObjectStoreCreator
 
 class BranchUtility(object):
-  def __init__(self, fetch_url, fetcher, object_store=None):
+  def __init__(self, fetch_url, fetcher, object_store_creator_factory):
     self._fetch_url = fetch_url
     self._fetcher = fetcher
-    if object_store is None:
-      object_store = (ObjectStoreCreator.SharedFactory(GetAppVersion())
-          .Create(BranchUtility).Create())
-    self._object_store = object_store
+    self._object_store = (object_store_creator_factory.Create(BranchUtility)
+        .Create())
 
   @staticmethod
   def GetAllBranchNames():
     return ['stable', 'beta', 'dev', 'trunk']
 
   def GetAllBranchNumbers(self):
     return [(branch, self.GetBranchNumberForChannelName(branch))
             for branch in BranchUtility.GetAllBranchNames()]
 
   @staticmethod
@@ skipping 42 matching lines @@
         else:
           branch_numbers[branch] += 1
 
     sorted_branches = sorted(branch_numbers.iteritems(),
                              None,
                              operator.itemgetter(1),
                              True)
     self._object_store.Set(channel_name, sorted_branches[0][0])
 
     return sorted_branches[0][0]