| Index: appengine/findit/common/test/git_repository_test.py
|
| diff --git a/appengine/findit/common/test/git_repository_test.py b/appengine/findit/common/test/git_repository_test.py
|
| index a0991742638da9b6fac6acdb7840e21c327aed83..d870f81d4aceabbd1a7fe3eca63a56aaf0adc567 100644
|
| --- a/appengine/findit/common/test/git_repository_test.py
|
| +++ b/appengine/findit/common/test/git_repository_test.py
|
| @@ -599,3 +599,28 @@ class GitRepositoryTest(testing.AppengineTestCase):
|
| self.assertRaisesRegexp(
|
| Exception, 'Failed to pull changelog for revision 2',
|
| self.git_repo.GetChangeLogs, '0', '2')
|
| +
|
| + def testGetRevisionForChromeVersion(self):
|
| +
|
| + def _MockGetChangeLog(*_):
|
| + changelog = ChangeLog.FromDict(DUMMY_CHANGELOG_JSON)
|
| + changelog.revision = '123a'
|
| + return changelog
|
| +
|
| + self.mock(git_repository.GitRepository, 'GetChangeLog', _MockGetChangeLog)
|
| +
|
| + git_repo = git_repository.GitRepository(self.repo_url,
|
| + self.http_client_for_git)
|
| + self.assertEqual(git_repo.GetRevisionForChromeVersion('50.0.1234.0'),
|
| + '123a')
|
| +
|
| + def testGetRevisionForChromeVersionFailToGetChangeLog(self):
|
| +
|
| + def _MockGetChangeLog(*_):
|
| + return None
|
| +
|
| + self.mock(git_repository.GitRepository, 'GetChangeLog', _MockGetChangeLog)
|
| +
|
| + git_repo = git_repository.GitRepository(self.repo_url,
|
| + self.http_client_for_git)
|
| + self.assertIsNone(git_repo.GetRevisionForChromeVersion('50.0.1234.0'))
|
|
|