You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
285 lines
11 KiB
Python
285 lines
11 KiB
Python
#!/usr/bin/env vpython3
|
|
# 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.
|
|
|
|
"""Unit tests for scm.py."""
|
|
|
|
import logging
|
|
import os
|
|
import sys
|
|
import unittest
|
|
|
|
if sys.version_info.major == 2:
|
|
import mock
|
|
else:
|
|
from unittest import mock
|
|
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
from testing_support import fake_repos
|
|
|
|
import scm
|
|
import subprocess2
|
|
|
|
|
|
def callError(code=1, cmd='', cwd='', stdout=b'', stderr=b''):
|
|
return subprocess2.CalledProcessError(code, cmd, cwd, stdout, stderr)
|
|
|
|
|
|
class GitWrapperTestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
super(GitWrapperTestCase, self).setUp()
|
|
self.root_dir = '/foo/bar'
|
|
|
|
@mock.patch('scm.GIT.Capture')
|
|
def testGetEmail(self, mockCapture):
|
|
mockCapture.return_value = 'mini@me.com'
|
|
self.assertEqual(scm.GIT.GetEmail(self.root_dir), 'mini@me.com')
|
|
mockCapture.assert_called_with(['config', 'user.email'], cwd=self.root_dir)
|
|
|
|
@mock.patch('scm.GIT.Capture')
|
|
def testAssertVersion(self, mockCapture):
|
|
cases = [
|
|
('1.7', True),
|
|
('1.7.9', True),
|
|
('1.7.9.foo-bar-baz', True),
|
|
('1.8', True),
|
|
('1.6.9', False),
|
|
]
|
|
for expected_version, expected_ok in cases:
|
|
class GIT(scm.GIT):
|
|
pass
|
|
mockCapture.return_value = 'git version ' + expected_version
|
|
ok, version = GIT.AssertVersion('1.7')
|
|
self.assertEqual(expected_ok, ok)
|
|
self.assertEqual(expected_version, version)
|
|
|
|
def testRefToRemoteRef(self):
|
|
remote = 'origin'
|
|
refs = {
|
|
'refs/branch-heads/1234': ('refs/remotes/branch-heads/', '1234'),
|
|
# local refs for upstream branch
|
|
'refs/remotes/%s/foobar' % remote: ('refs/remotes/%s/' % remote,
|
|
'foobar'),
|
|
'%s/foobar' % remote: ('refs/remotes/%s/' % remote, 'foobar'),
|
|
# upstream ref for branch
|
|
'refs/heads/foobar': ('refs/remotes/%s/' % remote, 'foobar'),
|
|
# could be either local or upstream ref, assumed to refer to
|
|
# upstream, but probably don't want to encourage refs like this.
|
|
'heads/foobar': ('refs/remotes/%s/' % remote, 'foobar'),
|
|
# underspecified, probably intended to refer to a local branch
|
|
'foobar': None,
|
|
# tags and other refs
|
|
'refs/tags/TAG': None,
|
|
'refs/changes/34/1234': None,
|
|
}
|
|
for k, v in refs.items():
|
|
r = scm.GIT.RefToRemoteRef(k, remote)
|
|
self.assertEqual(r, v, msg='%s -> %s, expected %s' % (k, r, v))
|
|
|
|
def testRemoteRefToRef(self):
|
|
remote = 'origin'
|
|
refs = {
|
|
'refs/remotes/branch-heads/1234': 'refs/branch-heads/1234',
|
|
# local refs for upstream branch
|
|
'refs/remotes/origin/foobar': 'refs/heads/foobar',
|
|
# tags and other refs
|
|
'refs/tags/TAG': 'refs/tags/TAG',
|
|
'refs/changes/34/1234': 'refs/changes/34/1234',
|
|
# different remote
|
|
'refs/remotes/other-remote/foobar': None,
|
|
# underspecified, probably intended to refer to a local branch
|
|
'heads/foobar': None,
|
|
'origin/foobar': None,
|
|
'foobar': None,
|
|
None: None,
|
|
}
|
|
for k, v in refs.items():
|
|
r = scm.GIT.RemoteRefToRef(k, remote)
|
|
self.assertEqual(r, v, msg='%s -> %s, expected %s' % (k, r, v))
|
|
|
|
@mock.patch('scm.GIT.Capture')
|
|
@mock.patch('os.path.exists', lambda _:True)
|
|
def testGetRemoteHeadRefLocal(self, mockCapture):
|
|
mockCapture.side_effect = ['refs/remotes/origin/main']
|
|
self.assertEqual('refs/remotes/origin/main',
|
|
scm.GIT.GetRemoteHeadRef('foo', 'proto://url', 'origin'))
|
|
self.assertEqual(mockCapture.call_count, 1)
|
|
|
|
@mock.patch('scm.GIT.Capture')
|
|
@mock.patch('os.path.exists', lambda _: True)
|
|
def testGetRemoteHeadRefLocalUpdateHead(self, mockCapture):
|
|
mockCapture.side_effect = [
|
|
'refs/remotes/origin/master', # first symbolic-ref call
|
|
'foo', # set-head call
|
|
'refs/remotes/origin/main', # second symbolic-ref call
|
|
]
|
|
self.assertEqual('refs/remotes/origin/main',
|
|
scm.GIT.GetRemoteHeadRef('foo', 'proto://url', 'origin'))
|
|
self.assertEqual(mockCapture.call_count, 3)
|
|
|
|
@mock.patch('scm.GIT.Capture')
|
|
@mock.patch('os.path.exists', lambda _:True)
|
|
def testGetRemoteHeadRefRemote(self, mockCapture):
|
|
mockCapture.side_effect = [
|
|
subprocess2.CalledProcessError(1, '', '', '', ''),
|
|
'ref: refs/heads/main\tHEAD\n' +
|
|
'0000000000000000000000000000000000000000\tHEAD',
|
|
]
|
|
self.assertEqual('refs/remotes/origin/main',
|
|
scm.GIT.GetRemoteHeadRef('foo', 'proto://url', 'origin'))
|
|
self.assertEqual(mockCapture.call_count, 2)
|
|
|
|
|
|
class RealGitTest(fake_repos.FakeReposTestBase):
|
|
def setUp(self):
|
|
super(RealGitTest, self).setUp()
|
|
self.enabled = self.FAKE_REPOS.set_up_git()
|
|
if self.enabled:
|
|
self.cwd = scm.os.path.join(self.FAKE_REPOS.git_base, 'repo_1')
|
|
else:
|
|
self.skipTest('git fake repos not available')
|
|
|
|
def testResolveCommit(self):
|
|
self.assertIsNone(scm.GIT.ResolveCommit(self.cwd, 'zebra'))
|
|
self.assertIsNone(scm.GIT.ResolveCommit(self.cwd, 'r123456'))
|
|
first_rev = self.githash('repo_1', 1)
|
|
self.assertEqual(first_rev, scm.GIT.ResolveCommit(self.cwd, first_rev))
|
|
self.assertEqual(
|
|
self.githash('repo_1', 2), scm.GIT.ResolveCommit(self.cwd, 'HEAD'))
|
|
|
|
def testIsValidRevision(self):
|
|
# Sha1's are [0-9a-z]{32}, so starting with a 'z' or 'r' should always fail.
|
|
self.assertFalse(scm.GIT.IsValidRevision(cwd=self.cwd, rev='zebra'))
|
|
self.assertFalse(scm.GIT.IsValidRevision(cwd=self.cwd, rev='r123456'))
|
|
# Valid cases
|
|
first_rev = self.githash('repo_1', 1)
|
|
self.assertTrue(scm.GIT.IsValidRevision(cwd=self.cwd, rev=first_rev))
|
|
self.assertTrue(scm.GIT.IsValidRevision(cwd=self.cwd, rev='HEAD'))
|
|
|
|
def testIsAncestor(self):
|
|
self.assertTrue(scm.GIT.IsAncestor(
|
|
self.cwd, self.githash('repo_1', 1), self.githash('repo_1', 2)))
|
|
self.assertFalse(scm.GIT.IsAncestor(
|
|
self.cwd, self.githash('repo_1', 2), self.githash('repo_1', 1)))
|
|
self.assertFalse(scm.GIT.IsAncestor(
|
|
self.cwd, self.githash('repo_1', 1), 'zebra'))
|
|
|
|
def testGetAllFiles(self):
|
|
self.assertEqual(['DEPS','origin'], scm.GIT.GetAllFiles(self.cwd))
|
|
|
|
def testGetSetConfig(self):
|
|
key = 'scm.test-key'
|
|
|
|
self.assertIsNone(scm.GIT.GetConfig(self.cwd, key))
|
|
self.assertEqual(
|
|
'default-value', scm.GIT.GetConfig(self.cwd, key, 'default-value'))
|
|
|
|
scm.GIT.SetConfig(self.cwd, key, 'set-value')
|
|
self.assertEqual('set-value', scm.GIT.GetConfig(self.cwd, key))
|
|
self.assertEqual(
|
|
'set-value', scm.GIT.GetConfig(self.cwd, key, 'default-value'))
|
|
|
|
scm.GIT.SetConfig(self.cwd, key)
|
|
self.assertIsNone(scm.GIT.GetConfig(self.cwd, key))
|
|
self.assertEqual(
|
|
'default-value', scm.GIT.GetConfig(self.cwd, key, 'default-value'))
|
|
|
|
def testGetSetBranchConfig(self):
|
|
branch = scm.GIT.GetBranch(self.cwd)
|
|
key = 'scm.test-key'
|
|
|
|
self.assertIsNone(scm.GIT.GetBranchConfig(self.cwd, branch, key))
|
|
self.assertEqual(
|
|
'default-value',
|
|
scm.GIT.GetBranchConfig(self.cwd, branch, key, 'default-value'))
|
|
|
|
scm.GIT.SetBranchConfig(self.cwd, branch, key, 'set-value')
|
|
self.assertEqual(
|
|
'set-value', scm.GIT.GetBranchConfig(self.cwd, branch, key))
|
|
self.assertEqual(
|
|
'set-value',
|
|
scm.GIT.GetBranchConfig(self.cwd, branch, key, 'default-value'))
|
|
self.assertEqual(
|
|
'set-value',
|
|
scm.GIT.GetConfig(self.cwd, 'branch.%s.%s' % (branch, key)))
|
|
|
|
scm.GIT.SetBranchConfig(self.cwd, branch, key)
|
|
self.assertIsNone(scm.GIT.GetBranchConfig(self.cwd, branch, key))
|
|
|
|
def testFetchUpstreamTuple_NoUpstreamFound(self):
|
|
self.assertEqual(
|
|
(None, None), scm.GIT.FetchUpstreamTuple(self.cwd))
|
|
|
|
@mock.patch('scm.GIT.GetRemoteBranches', return_value=['origin/main'])
|
|
def testFetchUpstreamTuple_GuessOriginMaster(self, _mockGetRemoteBranches):
|
|
self.assertEqual(('origin', 'refs/heads/main'),
|
|
scm.GIT.FetchUpstreamTuple(self.cwd))
|
|
|
|
@mock.patch('scm.GIT.GetRemoteBranches',
|
|
return_value=['origin/master', 'origin/main'])
|
|
def testFetchUpstreamTuple_GuessOriginMain(self, _mockGetRemoteBranches):
|
|
self.assertEqual(('origin', 'refs/heads/main'),
|
|
scm.GIT.FetchUpstreamTuple(self.cwd))
|
|
|
|
def testFetchUpstreamTuple_RietveldUpstreamConfig(self):
|
|
scm.GIT.SetConfig(self.cwd, 'rietveld.upstream-branch', 'rietveld-upstream')
|
|
scm.GIT.SetConfig(self.cwd, 'rietveld.upstream-remote', 'rietveld-remote')
|
|
self.assertEqual(
|
|
('rietveld-remote', 'rietveld-upstream'),
|
|
scm.GIT.FetchUpstreamTuple(self.cwd))
|
|
scm.GIT.SetConfig(self.cwd, 'rietveld.upstream-branch')
|
|
scm.GIT.SetConfig(self.cwd, 'rietveld.upstream-remote')
|
|
|
|
@mock.patch('scm.GIT.GetBranch', side_effect=callError())
|
|
def testFetchUpstreamTuple_NotOnBranch(self, _mockGetBranch):
|
|
scm.GIT.SetConfig(self.cwd, 'rietveld.upstream-branch', 'rietveld-upstream')
|
|
scm.GIT.SetConfig(self.cwd, 'rietveld.upstream-remote', 'rietveld-remote')
|
|
self.assertEqual(
|
|
('rietveld-remote', 'rietveld-upstream'),
|
|
scm.GIT.FetchUpstreamTuple(self.cwd))
|
|
scm.GIT.SetConfig(self.cwd, 'rietveld.upstream-branch')
|
|
scm.GIT.SetConfig(self.cwd, 'rietveld.upstream-remote')
|
|
|
|
def testFetchUpstreamTuple_BranchConfig(self):
|
|
branch = scm.GIT.GetBranch(self.cwd)
|
|
scm.GIT.SetBranchConfig(self.cwd, branch, 'merge', 'branch-merge')
|
|
scm.GIT.SetBranchConfig(self.cwd, branch, 'remote', 'branch-remote')
|
|
self.assertEqual(
|
|
('branch-remote', 'branch-merge'), scm.GIT.FetchUpstreamTuple(self.cwd))
|
|
scm.GIT.SetBranchConfig(self.cwd, branch, 'merge')
|
|
scm.GIT.SetBranchConfig(self.cwd, branch, 'remote')
|
|
|
|
def testFetchUpstreamTuple_AnotherBranchConfig(self):
|
|
branch = 'scm-test-branch'
|
|
scm.GIT.SetBranchConfig(self.cwd, branch, 'merge', 'other-merge')
|
|
scm.GIT.SetBranchConfig(self.cwd, branch, 'remote', 'other-remote')
|
|
self.assertEqual(
|
|
('other-remote', 'other-merge'),
|
|
scm.GIT.FetchUpstreamTuple(self.cwd, branch))
|
|
scm.GIT.SetBranchConfig(self.cwd, branch, 'merge')
|
|
scm.GIT.SetBranchConfig(self.cwd, branch, 'remote')
|
|
|
|
def testGetBranchRef(self):
|
|
self.assertEqual('refs/heads/main', scm.GIT.GetBranchRef(self.cwd))
|
|
HEAD = scm.GIT.Capture(['rev-parse', 'HEAD'], cwd=self.cwd)
|
|
scm.GIT.Capture(['checkout', HEAD], cwd=self.cwd)
|
|
self.assertIsNone(scm.GIT.GetBranchRef(self.cwd))
|
|
scm.GIT.Capture(['checkout', 'main'], cwd=self.cwd)
|
|
|
|
def testGetBranch(self):
|
|
self.assertEqual('main', scm.GIT.GetBranch(self.cwd))
|
|
HEAD = scm.GIT.Capture(['rev-parse', 'HEAD'], cwd=self.cwd)
|
|
scm.GIT.Capture(['checkout', HEAD], cwd=self.cwd)
|
|
self.assertIsNone(scm.GIT.GetBranchRef(self.cwd))
|
|
scm.GIT.Capture(['checkout', 'main'], cwd=self.cwd)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
if '-v' in sys.argv:
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
unittest.main()
|
|
|
|
# vim: ts=2:sw=2:tw=80:et:
|