Added `cros_pin` CrOS pin-bump tool.

infra/libs/gitiles/test/gitiles_test.py

 # 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.
 
 import base64
 import httplib2
 import os
 import time
 
 from infra.libs.gitiles import gitiles
 from testing_support import auto_stub
 
+import mock
 
 DATA_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'data')
 
 
 class TestGitiles(auto_stub.TestCase):
   def setUp(self):
     super(TestGitiles, self).setUp()
 
-    self.response = {'status': '200'}
+    self.response = {'status': 200}
     self.content = None
 
     class FakeHttp(object):
       def __init__(self, response, content):
         self.response = response
         self.content = content
 
       def request(self, *_args, **_kwargs):
-        return self.response, self.content
+        r = mock.Mock()
+        r.status = self.response['status']
+        return r, self.content
 
     def http_maker():
       return FakeHttp(self.response, self.content)
 
     self.mock(httplib2, 'Http', http_maker)
     self.mock(time, 'sleep', lambda _x: None)
 
   def testJson(self):
     with open(os.path.join(DATA_DIR, 'scripts_slave')) as f:
       self.content = f.read()
 
     result = gitiles.call_gitiles(
         'http://bananas',
         'json',
         os.path.join(DATA_DIR, 'netrc'))
     self.assertEqual(result['id'], '24a7f79e278700fab6dfd3866b1b8508c44ddb55')
 
   def testText(self):
     with open(os.path.join(DATA_DIR, 'init_py')) as f:
       self.content = f.read()
 
     result = gitiles.call_gitiles(
         'http://bananas',
         'text',
         os.path.join(DATA_DIR, 'netrc'))
 
     self.assertTrue(result.startswith('# Copyright 2015'))
 
   def testInvalidResponse(self):
-    self.response = {'status': '500'}
+    self.response = {'status': 500}
     with self.assertRaises(gitiles.GitilesError):
       gitiles.call_gitiles(
           'http://bananas',
+          'text',
+          os.path.join(DATA_DIR, 'netrc'))
+
+  def testErrorResponse(self):
+    self.response = {'status': 400}
+    with self.assertRaises(gitiles.GitilesError):
+      gitiles.call_gitiles(
+          'http://bananas',
           'text',
           os.path.join(DATA_DIR, 'netrc'))
 
   def testInvalidJsonResponse(self):
     self.content = 'blerg\ndefinitely does not start with )]}\''
     with self.assertRaises(gitiles.GitilesError):
       gitiles.call_gitiles(
           'http://bananas',
           'json',
           os.path.join(DATA_DIR, 'netrc'))
@@ skipping 20 matching lines @@
           os.path.join(DATA_DIR, 'netrc'))
 
   def testNoAuth(self):
     with open(os.path.join(DATA_DIR, 'scripts_slave')) as f:
       self.content = f.read()
     result = gitiles.call_gitiles(
         'http://twirly',
         'json'
     )
     self.assertEqual(result['id'], '24a7f79e278700fab6dfd3866b1b8508c44ddb55')
+
+  def testRepositoryCall(self):
+    r = gitiles.Repository('http://example.com')
+    self.content = ')]}\'\n{"commit": "foo"}'
+    v = r.ref_info('master')
+    self.assertEqual(v, {'commit': 'foo'})
+
+  def testRepositorySubpathCall(self):
+    r = gitiles.Repository('http://example.com')
+    self.content = ')]}\'\n{"commit": "foo"}'
+    v = r(subpath='master/foo/bar/baz.py')
+    self.assertEqual(v, {'commit': 'foo'})