CrOS: Load Chromite pins from JSON.

scripts/common/unittests/cros_chromite_test.py

 #!/usr/bin/env python2.7
 # Copyright 2015 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.
 
-"""Unit test suite for common.cros_chromite."""
+"""Unit test suite for common.cros.chromite."""
 
 import test_env
 
 import base64
 import json
 import unittest
 
 from common import cros_chromite
 
 
@@ skipping 158 matching lines @@
       base, suffix, category = cros_chromite.ChromiteTarget.Categorize(
           name,
           build_type=build_type)
       self.assertEqual(
           (base, suffix, category), (exp_base, exp_suffix, exp_category))
 
 
 class ChromitePinManagerTestCase(unittest.TestCase):
 
   def testGetPinnedBranch_PinnedBranchReturnsPinnedValue(self):
-    pm = cros_chromite.ChromitePinManager(pinned={'a': 'b'}, require=True)
+    pm = cros_chromite.ChromitePinManager(
+        'test',
+        pinned={'a': 'b'},
+        require=True)
     self.assertEqual(pm.GetPinnedBranch('a'), 'b')
 
   def testGetPinnedBranch_UnpinnedBranchReturnsBranch(self):
-    pm = cros_chromite.ChromitePinManager(pinned={'a': 'b'}, require=False)
+    pm = cros_chromite.ChromitePinManager(
+        'test',
+        pinned={'a': 'b'},
+        require=False)
     self.assertEqual(pm.GetPinnedBranch('foo'), 'foo')
 
   def testGetPinnedBranch_UnpinnedBranchReturnsErrorWithRequiredPinning(self):
-    pm = cros_chromite.ChromitePinManager(pinned={'a': 'b'}, require=True)
+    pm = cros_chromite.ChromitePinManager(
+        'test',
+        pinned={'a': 'b'},
+        require=True)
     self.assertRaises(cros_chromite.ChromiteError,
         pm.GetPinnedBranch, 'foo')
 
 
 class ChromiteConfigManagerTestCase(unittest.TestCase):
 
   def setUp(self):
     self.cache = MockConfigCache({
       ('test', 'v1'): '{}',
       ('test', 'v_invalid'): '{NOT JSON}',
     })
 
   def testGetConfig_ValidSucceeds(self):
     manager = cros_chromite.ChromiteConfigManager(self.cache,
-        cros_chromite.ChromitePinManager({'test': 'v1'}))
+        cros_chromite.ChromitePinManager(
+          'test',
+          {'test': 'v1'}))
     self.assertTrue(isinstance(manager.GetConfig('test'),
         cros_chromite.ChromiteConfig))
 
   def testGetConfig_InvalidJsonRaises(self):
     manager = cros_chromite.ChromiteConfigManager(self.cache,
-        cros_chromite.ChromitePinManager({'test': 'v_invalid'}))
+        cros_chromite.ChromitePinManager(
+          'test',
+          {'test': 'v_invalid'}))
     self.assertRaises(cros_chromite.ChromiteError, manager.GetConfig, 'test')
 
   def testGetConfig_MissingRaises(self):
     manager = cros_chromite.ChromiteConfigManager(self.cache)
     self.assertRaises(cros_chromite.ChromiteError, manager.GetConfig, 'foo')
 
 
 class ChromiteFetcherTestCase(unittest.TestCase):
 
   def setUp(self):
     self.fetcher = cros_chromite.ChromiteFetcher(
-        cros_chromite.ChromitePinManager({'test': 'v1'})
+        cros_chromite.ChromitePinManager(
+          'test',
+          {'test': 'v1'})
     )
 
   @staticmethod
   def _configUrlForBranch(branch):
     return '%s/+/%s/%s?format=text' % (
         cros_chromite.ChromiteFetcher.CHROMITE_GITILES_BASE,
         branch,
         cros_chromite.ChromiteFetcher.CHROMITE_CONFIG_PATH,
     )
 
@@ skipping 10 matching lines @@
 
   def testFetch_NotBase64(self):
     def _MockGetText(_url):
       return 'Not Valid Base64'
     self.fetcher._GetText = _MockGetText
     self.assertRaises(cros_chromite.GitilesError, self.fetcher, 'test', None)
 
 
 if __name__ == '__main__':
   unittest.main()