| #!/usr/bin/python |
| # |
| # Copyright 2008-2009 Google Inc. All Rights Reserved. |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); |
| # you may not use this file except in compliance with the License. |
| # You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| |
| """Unit tests for gclient_scm.py.""" |
| |
| import shutil |
| # Import it before super_mox to keep a valid reference. |
| from subprocess import Popen, PIPE, STDOUT |
| import tempfile |
| |
| import gclient_scm |
| from gclient_test import BaseTestCase as GCBaseTestCase |
| from super_mox import mox, SuperMoxBaseTestBase |
| |
| |
| class BaseTestCase(GCBaseTestCase): |
| def setUp(self): |
| GCBaseTestCase.setUp(self) |
| self.mox.StubOutWithMock(gclient_scm.gclient_utils, 'FileRead') |
| self.mox.StubOutWithMock(gclient_scm.gclient_utils, 'FileWrite') |
| self.mox.StubOutWithMock(gclient_scm.gclient_utils, 'SubprocessCall') |
| self.mox.StubOutWithMock(gclient_scm.gclient_utils, 'RemoveDirectory') |
| self._CaptureSVNInfo = gclient_scm.scm.SVN.CaptureInfo |
| self.mox.StubOutWithMock(gclient_scm.scm.SVN, 'Capture') |
| self.mox.StubOutWithMock(gclient_scm.scm.SVN, 'CaptureInfo') |
| self.mox.StubOutWithMock(gclient_scm.scm.SVN, 'CaptureStatus') |
| self.mox.StubOutWithMock(gclient_scm.scm.SVN, 'Run') |
| self.mox.StubOutWithMock(gclient_scm.scm.SVN, 'RunAndGetFileList') |
| self._scm_wrapper = gclient_scm.CreateSCM |
| |
| |
| class SVNWrapperTestCase(BaseTestCase): |
| class OptionsObject(object): |
| def __init__(self, test_case, verbose=False, revision=None): |
| self.verbose = verbose |
| self.revision = revision |
| self.manually_grab_svn_rev = True |
| self.deps_os = None |
| self.force = False |
| self.nohooks = False |
| |
| def Options(self, *args, **kwargs): |
| return self.OptionsObject(self, *args, **kwargs) |
| |
| def setUp(self): |
| BaseTestCase.setUp(self) |
| self.root_dir = self.Dir() |
| self.args = self.Args() |
| self.url = self.Url() |
| self.relpath = 'asf' |
| |
| def testDir(self): |
| members = [ |
| 'COMMAND', 'Capture', 'CaptureHeadRevision', 'CaptureInfo', |
| 'CaptureStatus', 'DiffItem', 'GetEmail', |
| 'GetFileProperty', 'IsMoved', 'ReadSimpleAuth', 'Run', |
| 'RunAndFilterOutput', 'RunAndGetFileList', |
| 'RunCommand', 'cleanup', 'diff', 'export', 'pack', 'relpath', 'revert', |
| 'revinfo', 'runhooks', 'scm_name', 'status', 'update', 'url', |
| ] |
| |
| # If you add a member, be sure to add the relevant test! |
| self.compareMembers(self._scm_wrapper(), members) |
| |
| def testUnsupportedSCM(self): |
| args = [self.url, self.root_dir, self.relpath] |
| kwargs = {'scm_name' : 'foo'} |
| exception_msg = 'Unsupported scm %(scm_name)s' % kwargs |
| self.assertRaisesError(exception_msg, self._scm_wrapper, *args, **kwargs) |
| |
| def testRunCommandException(self): |
| options = self.Options(verbose=False) |
| file_path = gclient_scm.os.path.join(self.root_dir, self.relpath, '.git') |
| gclient_scm.os.path.exists(file_path).AndReturn(False) |
| |
| self.mox.ReplayAll() |
| scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir, |
| relpath=self.relpath) |
| exception = "Unsupported argument(s): %s" % ','.join(self.args) |
| self.assertRaisesError(exception, scm.RunCommand, |
| 'update', options, self.args) |
| |
| def testRunCommandUnknown(self): |
| # TODO(maruel): if ever used. |
| pass |
| |
| def testRevertMissing(self): |
| options = self.Options(verbose=True) |
| base_path = gclient_scm.os.path.join(self.root_dir, self.relpath) |
| gclient_scm.os.path.isdir(base_path).AndReturn(False) |
| # It'll to a checkout instead. |
| gclient_scm.os.path.exists(gclient_scm.os.path.join(base_path, '.git') |
| ).AndReturn(False) |
| print("\n_____ %s is missing, synching instead" % self.relpath) |
| # Checkout. |
| gclient_scm.os.path.exists(base_path).AndReturn(False) |
| files_list = self.mox.CreateMockAnything() |
| gclient_scm.scm.SVN.RunAndGetFileList(options, |
| ['checkout', self.url, base_path], |
| self.root_dir, files_list) |
| |
| self.mox.ReplayAll() |
| scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir, |
| relpath=self.relpath) |
| scm.revert(options, self.args, files_list) |
| |
| def testRevertNone(self): |
| options = self.Options(verbose=True) |
| base_path = gclient_scm.os.path.join(self.root_dir, self.relpath) |
| gclient_scm.os.path.isdir(base_path).AndReturn(True) |
| gclient_scm.scm.SVN.CaptureStatus(base_path).AndReturn([]) |
| gclient_scm.scm.SVN.RunAndGetFileList(options, |
| ['update', '--revision', 'BASE'], |
| base_path, mox.IgnoreArg()) |
| |
| self.mox.ReplayAll() |
| scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir, |
| relpath=self.relpath) |
| file_list = [] |
| scm.revert(options, self.args, file_list) |
| |
| def testRevert2Files(self): |
| options = self.Options(verbose=True) |
| base_path = gclient_scm.os.path.join(self.root_dir, self.relpath) |
| gclient_scm.os.path.isdir(base_path).AndReturn(True) |
| items = [ |
| ('M ', 'a'), |
| ('A ', 'b'), |
| ] |
| file_path1 = gclient_scm.os.path.join(base_path, 'a') |
| file_path2 = gclient_scm.os.path.join(base_path, 'b') |
| gclient_scm.scm.SVN.CaptureStatus(base_path).AndReturn(items) |
| gclient_scm.os.path.exists(file_path1).AndReturn(True) |
| gclient_scm.os.path.isfile(file_path1).AndReturn(True) |
| gclient_scm.os.remove(file_path1) |
| gclient_scm.os.path.exists(file_path2).AndReturn(True) |
| gclient_scm.os.path.isfile(file_path2).AndReturn(True) |
| gclient_scm.os.remove(file_path2) |
| gclient_scm.scm.SVN.RunAndGetFileList(options, |
| ['update', '--revision', 'BASE'], |
| base_path, mox.IgnoreArg()) |
| print(gclient_scm.os.path.join(base_path, 'a')) |
| print(gclient_scm.os.path.join(base_path, 'b')) |
| |
| self.mox.ReplayAll() |
| scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir, |
| relpath=self.relpath) |
| file_list = [] |
| scm.revert(options, self.args, file_list) |
| |
| def testRevertDirectory(self): |
| options = self.Options(verbose=True) |
| base_path = gclient_scm.os.path.join(self.root_dir, self.relpath) |
| gclient_scm.os.path.isdir(base_path).AndReturn(True) |
| items = [ |
| ('~ ', 'a'), |
| ] |
| gclient_scm.scm.SVN.CaptureStatus(base_path).AndReturn(items) |
| file_path = gclient_scm.os.path.join(base_path, 'a') |
| print(file_path) |
| gclient_scm.os.path.exists(file_path).AndReturn(True) |
| gclient_scm.os.path.isfile(file_path).AndReturn(False) |
| gclient_scm.os.path.isdir(file_path).AndReturn(True) |
| gclient_scm.gclient_utils.RemoveDirectory(file_path) |
| file_list1 = [] |
| gclient_scm.scm.SVN.RunAndGetFileList(options, |
| ['update', '--revision', 'BASE'], |
| base_path, mox.IgnoreArg()) |
| |
| self.mox.ReplayAll() |
| scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir, |
| relpath=self.relpath) |
| file_list2 = [] |
| scm.revert(options, self.args, file_list2) |
| |
| def testStatus(self): |
| options = self.Options(verbose=True) |
| base_path = gclient_scm.os.path.join(self.root_dir, self.relpath) |
| gclient_scm.os.path.isdir(base_path).AndReturn(True) |
| gclient_scm.scm.SVN.RunAndGetFileList(options, |
| ['status'] + self.args, |
| base_path, []).AndReturn(None) |
| |
| self.mox.ReplayAll() |
| scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir, |
| relpath=self.relpath) |
| file_list = [] |
| self.assertEqual(scm.status(options, self.args, file_list), None) |
| |
| |
| # TODO(maruel): TEST REVISIONS!!! |
| # TODO(maruel): TEST RELOCATE!!! |
| def testUpdateCheckout(self): |
| options = self.Options(verbose=True) |
| base_path = gclient_scm.os.path.join(self.root_dir, self.relpath) |
| file_info = gclient_scm.gclient_utils.PrintableObject() |
| file_info.root = 'blah' |
| file_info.url = self.url |
| file_info.uuid = 'ABC' |
| file_info.revision = 42 |
| gclient_scm.os.path.exists(gclient_scm.os.path.join(base_path, '.git') |
| ).AndReturn(False) |
| # Checkout. |
| gclient_scm.os.path.exists(base_path).AndReturn(False) |
| files_list = self.mox.CreateMockAnything() |
| gclient_scm.scm.SVN.RunAndGetFileList(options, |
| ['checkout', self.url, base_path], |
| self.root_dir, files_list) |
| self.mox.ReplayAll() |
| scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir, |
| relpath=self.relpath) |
| scm.update(options, (), files_list) |
| |
| def testUpdateUpdate(self): |
| options = self.Options(verbose=True) |
| base_path = gclient_scm.os.path.join(self.root_dir, self.relpath) |
| options.force = True |
| options.nohooks = False |
| file_info = { |
| 'Repository Root': 'blah', |
| 'URL': self.url, |
| 'UUID': 'ABC', |
| 'Revision': 42, |
| } |
| gclient_scm.os.path.exists(gclient_scm.os.path.join(base_path, '.git') |
| ).AndReturn(False) |
| # Checkout or update. |
| gclient_scm.os.path.exists(base_path).AndReturn(True) |
| gclient_scm.scm.SVN.CaptureInfo( |
| gclient_scm.os.path.join(base_path, "."), '.' |
| ).AndReturn(file_info) |
| # Cheat a bit here. |
| gclient_scm.scm.SVN.CaptureInfo(file_info['URL'], '.').AndReturn(file_info) |
| additional_args = [] |
| if options.manually_grab_svn_rev: |
| additional_args = ['--revision', str(file_info['Revision'])] |
| files_list = [] |
| gclient_scm.scm.SVN.RunAndGetFileList( |
| options, |
| ['update', base_path] + additional_args, |
| self.root_dir, files_list) |
| |
| self.mox.ReplayAll() |
| scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir, |
| relpath=self.relpath) |
| scm.update(options, (), files_list) |
| |
| def testUpdateGit(self): |
| options = self.Options(verbose=True) |
| file_path = gclient_scm.os.path.join(self.root_dir, self.relpath, '.git') |
| gclient_scm.os.path.exists(file_path).AndReturn(True) |
| print("________ found .git directory; skipping %s" % self.relpath) |
| |
| self.mox.ReplayAll() |
| scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir, |
| relpath=self.relpath) |
| file_list = [] |
| scm.update(options, self.args, file_list) |
| |
| |
| class GitWrapperTestCase(BaseTestCase): |
| """This class doesn't use pymox.""" |
| class OptionsObject(object): |
| def __init__(self, test_case, verbose=False, revision=None): |
| self.verbose = verbose |
| self.revision = revision |
| self.manually_grab_svn_rev = True |
| self.deps_os = None |
| self.force = False |
| self.nohooks = False |
| |
| sample_git_import = """blob |
| mark :1 |
| data 6 |
| Hello |
| |
| blob |
| mark :2 |
| data 4 |
| Bye |
| |
| reset refs/heads/master |
| commit refs/heads/master |
| mark :3 |
| author Bob <[email protected]> 1253744361 -0700 |
| committer Bob <[email protected]> 1253744361 -0700 |
| data 8 |
| A and B |
| M 100644 :1 a |
| M 100644 :2 b |
| |
| blob |
| mark :4 |
| data 10 |
| Hello |
| You |
| |
| blob |
| mark :5 |
| data 8 |
| Bye |
| You |
| |
| commit refs/heads/origin |
| mark :6 |
| author Alice <[email protected]> 1253744424 -0700 |
| committer Alice <[email protected]> 1253744424 -0700 |
| data 13 |
| Personalized |
| from :3 |
| M 100644 :4 a |
| M 100644 :5 b |
| |
| reset refs/heads/master |
| from :3 |
| """ |
| def Options(self, *args, **kwargs): |
| return self.OptionsObject(self, *args, **kwargs) |
| |
| def CreateGitRepo(self, git_import, path): |
| """Do it for real.""" |
| try: |
| Popen(['git', 'init'], stdout=PIPE, stderr=STDOUT, |
| cwd=path).communicate() |
| except OSError: |
| # git is not available, skip this test. |
| return False |
| Popen(['git', 'fast-import'], stdin=PIPE, stdout=PIPE, stderr=STDOUT, |
| cwd=path).communicate(input=git_import) |
| Popen(['git', 'checkout'], stdout=PIPE, stderr=STDOUT, |
| cwd=path).communicate() |
| return True |
| |
| def setUp(self): |
| self.args = self.Args() |
| self.url = 'git://foo' |
| self.root_dir = tempfile.mkdtemp() |
| self.relpath = '.' |
| self.base_path = gclient_scm.os.path.join(self.root_dir, self.relpath) |
| self.enabled = self.CreateGitRepo(self.sample_git_import, self.base_path) |
| SuperMoxBaseTestBase.setUp(self) |
| |
| def tearDown(self): |
| SuperMoxBaseTestBase.tearDown(self) |
| shutil.rmtree(self.root_dir) |
| |
| def testDir(self): |
| members = [ |
| 'COMMAND', 'Capture', 'CaptureStatus', 'GetEmail', |
| 'RunCommand', 'cleanup', 'diff', 'export', 'relpath', 'revert', |
| 'revinfo', 'runhooks', 'scm_name', 'status', 'update', 'url', |
| ] |
| |
| # If you add a member, be sure to add the relevant test! |
| self.compareMembers(gclient_scm.CreateSCM(url=self.url), members) |
| |
| def testRevertMissing(self): |
| if not self.enabled: |
| return |
| options = self.Options() |
| file_path = gclient_scm.os.path.join(self.base_path, 'a') |
| gclient_scm.os.remove(file_path) |
| scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir, |
| relpath=self.relpath) |
| file_list = [] |
| scm.revert(options, self.args, file_list) |
| self.assertEquals(file_list, [file_path]) |
| file_list = [] |
| scm.diff(options, self.args, file_list) |
| self.assertEquals(file_list, []) |
| |
| def testRevertNone(self): |
| if not self.enabled: |
| return |
| options = self.Options() |
| scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir, |
| relpath=self.relpath) |
| file_list = [] |
| scm.revert(options, self.args, file_list) |
| self.assertEquals(file_list, []) |
| self.assertEquals(scm.revinfo(options, self.args, None), |
| '069c602044c5388d2d15c3f875b057c852003458') |
| |
| |
| def testRevertModified(self): |
| if not self.enabled: |
| return |
| options = self.Options() |
| file_path = gclient_scm.os.path.join(self.base_path, 'a') |
| open(file_path, 'a').writelines('touched\n') |
| scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir, |
| relpath=self.relpath) |
| file_list = [] |
| scm.revert(options, self.args, file_list) |
| self.assertEquals(file_list, [file_path]) |
| file_list = [] |
| scm.diff(options, self.args, file_list) |
| self.assertEquals(file_list, []) |
| self.assertEquals(scm.revinfo(options, self.args, None), |
| '069c602044c5388d2d15c3f875b057c852003458') |
| |
| def testRevertNew(self): |
| if not self.enabled: |
| return |
| options = self.Options() |
| file_path = gclient_scm.os.path.join(self.base_path, 'c') |
| f = open(file_path, 'w') |
| f.writelines('new\n') |
| f.close() |
| Popen(['git', 'add', 'c'], stdout=PIPE, |
| stderr=STDOUT, cwd=self.base_path).communicate() |
| scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir, |
| relpath=self.relpath) |
| file_list = [] |
| scm.revert(options, self.args, file_list) |
| self.assertEquals(file_list, [file_path]) |
| file_list = [] |
| scm.diff(options, self.args, file_list) |
| self.assertEquals(file_list, []) |
| self.assertEquals(scm.revinfo(options, self.args, None), |
| '069c602044c5388d2d15c3f875b057c852003458') |
| |
| def testStatusNew(self): |
| if not self.enabled: |
| return |
| options = self.Options() |
| file_path = gclient_scm.os.path.join(self.base_path, 'a') |
| open(file_path, 'a').writelines('touched\n') |
| scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir, |
| relpath=self.relpath) |
| file_list = [] |
| scm.status(options, self.args, file_list) |
| self.assertEquals(file_list, [file_path]) |
| |
| def testStatus2New(self): |
| if not self.enabled: |
| return |
| options = self.Options() |
| expected_file_list = [] |
| for f in ['a', 'b']: |
| file_path = gclient_scm.os.path.join(self.base_path, f) |
| open(file_path, 'a').writelines('touched\n') |
| expected_file_list.extend([file_path]) |
| scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir, |
| relpath=self.relpath) |
| file_list = [] |
| scm.status(options, self.args, file_list) |
| expected_file_list = [gclient_scm.os.path.join(self.base_path, x) |
| for x in ['a', 'b']] |
| self.assertEquals(sorted(file_list), expected_file_list) |
| |
| def testUpdateCheckout(self): |
| if not self.enabled: |
| return |
| options = self.Options(verbose=True) |
| root_dir = tempfile.mkdtemp() |
| relpath = 'foo' |
| base_path = gclient_scm.os.path.join(root_dir, relpath) |
| url = gclient_scm.os.path.join(self.root_dir, self.relpath, '.git') |
| try: |
| scm = gclient_scm.CreateSCM(url=url, root_dir=root_dir, |
| relpath=relpath) |
| file_list = [] |
| scm.update(options, (), file_list) |
| self.assertEquals(len(file_list), 2) |
| self.assert_(gclient_scm.os.path.isfile( |
| gclient_scm.os.path.join(base_path, 'a'))) |
| self.assertEquals(scm.revinfo(options, (), None), |
| '069c602044c5388d2d15c3f875b057c852003458') |
| finally: |
| shutil.rmtree(root_dir) |
| |
| def testUpdateUpdate(self): |
| if not self.enabled: |
| return |
| options = self.Options() |
| expected_file_list = [gclient_scm.os.path.join(self.base_path, x) |
| for x in ['a', 'b']] |
| scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir, |
| relpath=self.relpath) |
| file_list = [] |
| scm.update(options, (), file_list) |
| self.assertEquals(file_list, expected_file_list) |
| self.assertEquals(scm.revinfo(options, (), None), |
| 'a7142dc9f0009350b96a11f372b6ea658592aa95') |
| |
| def testUpdateConflict(self): |
| if not self.enabled: |
| return |
| options = self.Options() |
| scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir, |
| relpath=self.relpath) |
| file_path = gclient_scm.os.path.join(self.base_path, 'b') |
| f = open(file_path, 'w').writelines('conflict\n') |
| scm._Run(['commit', '-am', 'test']) |
| exception = \ |
| '\n____ .\n' \ |
| '\nConflict while rebasing this branch.\n' \ |
| 'Fix the conflict and run gclient again.\n' \ |
| 'See man git-rebase for details.\n' |
| self.assertRaisesError(exception, scm.update, options, (), []) |
| exception = \ |
| '\n____ .\n' \ |
| '\tAlready in a conflict, i.e. (no branch).\n' \ |
| '\tFix the conflict and run gclient again.\n' \ |
| '\tOr to abort run:\n\t\tgit-rebase --abort\n' \ |
| '\tSee man git-rebase for details.\n' |
| self.assertRaisesError(exception, scm.update, options, (), []) |
| |
| def testRevinfo(self): |
| if not self.enabled: |
| return |
| options = self.Options() |
| scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir, |
| relpath=self.relpath) |
| rev_info = scm.revinfo(options, (), None) |
| self.assertEquals(rev_info, '069c602044c5388d2d15c3f875b057c852003458') |
| |
| |
| if __name__ == '__main__': |
| import unittest |
| unittest.main() |
| |
| # vim: ts=2:sw=2:tw=80:et: |
