blob: a075d86c750f5eb49d69eda3fe00a98980665346 [file] [log] [blame]
msb@chromium.orge28e4982009-09-25 20:51:45 +00001#!/usr/bin/python
maruel@chromium.orgba551772010-02-03 18:21:42 +00002# Copyright (c) 2010 The Chromium Authors. All rights reserved.
3# Use of this source code is governed by a BSD-style license that can be
4# found in the LICENSE file.
msb@chromium.orge28e4982009-09-25 20:51:45 +00005
6"""Unit tests for gclient_scm.py."""
7
maruel@chromium.orgba551772010-02-03 18:21:42 +00008# Import before super_mox to keep valid references.
9from os import rename
10from shutil import rmtree
maruel@chromium.org54019f32010-09-09 13:50:11 +000011import StringIO
maruel@chromium.org8ef5f542009-11-12 02:05:02 +000012from subprocess import Popen, PIPE, STDOUT
msb@chromium.orge28e4982009-09-25 20:51:45 +000013import tempfile
maruel@chromium.org389d6de2010-09-09 14:14:37 +000014import unittest
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +000015import __builtin__
msb@chromium.orge28e4982009-09-25 20:51:45 +000016
maruel@chromium.orgba551772010-02-03 18:21:42 +000017# Fixes include path.
maruel@chromium.orgdd057f32010-09-20 19:33:31 +000018from super_mox import mox, StdoutCheck, TestCaseUtils, SuperMoxTestBase
maruel@chromium.orgba551772010-02-03 18:21:42 +000019
msb@chromium.orge28e4982009-09-25 20:51:45 +000020import gclient_scm
maruel@chromium.org96913eb2010-06-01 16:22:47 +000021
maruel@chromium.org795a8c12010-10-05 19:54:29 +000022# Shortcut since this function is used often
23join = gclient_scm.os.path.join
24
maruel@chromium.org96913eb2010-06-01 16:22:47 +000025
maruel@chromium.org389d6de2010-09-09 14:14:37 +000026class GCBaseTestCase(object):
maruel@chromium.org96913eb2010-06-01 16:22:47 +000027 def assertRaisesError(self, msg, fn, *args, **kwargs):
maruel@chromium.org389d6de2010-09-09 14:14:37 +000028 """Like unittest's assertRaises() but checks for Gclient.Error."""
maruel@chromium.org96913eb2010-06-01 16:22:47 +000029 try:
30 fn(*args, **kwargs)
31 except gclient_scm.gclient_utils.Error, e:
32 self.assertEquals(e.args[0], msg)
33 else:
34 self.fail('%s not raised' % msg)
msb@chromium.orge28e4982009-09-25 20:51:45 +000035
maruel@chromium.org8ef5f542009-11-12 02:05:02 +000036
maruel@chromium.org389d6de2010-09-09 14:14:37 +000037class BaseTestCase(GCBaseTestCase, SuperMoxTestBase):
38 def setUp(self):
maruel@chromium.org389d6de2010-09-09 14:14:37 +000039 SuperMoxTestBase.setUp(self)
40 self.mox.StubOutWithMock(gclient_scm.gclient_utils, 'CheckCall')
41 self.mox.StubOutWithMock(gclient_scm.gclient_utils, 'CheckCallAndFilter')
42 self.mox.StubOutWithMock(gclient_scm.gclient_utils,
43 'CheckCallAndFilterAndHeader')
44 self.mox.StubOutWithMock(gclient_scm.gclient_utils, 'FileRead')
45 self.mox.StubOutWithMock(gclient_scm.gclient_utils, 'FileWrite')
46 self.mox.StubOutWithMock(gclient_scm.gclient_utils, 'Popen')
47 self.mox.StubOutWithMock(gclient_scm.gclient_utils, 'RemoveDirectory')
48 self._CaptureSVNInfo = gclient_scm.scm.SVN.CaptureInfo
49 self.mox.StubOutWithMock(gclient_scm.scm.SVN, 'Capture')
50 self.mox.StubOutWithMock(gclient_scm.scm.SVN, 'CaptureInfo')
51 self.mox.StubOutWithMock(gclient_scm.scm.SVN, 'CaptureStatus')
52 self.mox.StubOutWithMock(gclient_scm.scm.SVN, 'RunAndGetFileList')
53 self._scm_wrapper = gclient_scm.CreateSCM
maruel@chromium.org389d6de2010-09-09 14:14:37 +000054 gclient_scm.scm.SVN.current_version = None
maruel@chromium.org795a8c12010-10-05 19:54:29 +000055 # Absolute path of the fake checkout directory.
56 self.base_path = join(self.root_dir, self.relpath)
maruel@chromium.org389d6de2010-09-09 14:14:37 +000057
58 def tearDown(self):
maruel@chromium.org389d6de2010-09-09 14:14:37 +000059 SuperMoxTestBase.tearDown(self)
60
61
maruel@chromium.org8ef5f542009-11-12 02:05:02 +000062class SVNWrapperTestCase(BaseTestCase):
msb@chromium.orge28e4982009-09-25 20:51:45 +000063 class OptionsObject(object):
maruel@chromium.org8071c282010-09-20 19:44:19 +000064 def __init__(self, verbose=False, revision=None):
msb@chromium.orge28e4982009-09-25 20:51:45 +000065 self.verbose = verbose
66 self.revision = revision
67 self.manually_grab_svn_rev = True
68 self.deps_os = None
69 self.force = False
davemoore@chromium.org8bf27312010-02-19 17:29:44 +000070 self.reset = False
msb@chromium.orge28e4982009-09-25 20:51:45 +000071 self.nohooks = False
72
maruel@chromium.org8ef5f542009-11-12 02:05:02 +000073 def Options(self, *args, **kwargs):
maruel@chromium.org8071c282010-09-20 19:44:19 +000074 return self.OptionsObject(*args, **kwargs)
maruel@chromium.org8ef5f542009-11-12 02:05:02 +000075
msb@chromium.orge28e4982009-09-25 20:51:45 +000076 def setUp(self):
maruel@chromium.org8ef5f542009-11-12 02:05:02 +000077 BaseTestCase.setUp(self)
maruel@chromium.org389d6de2010-09-09 14:14:37 +000078 self.url = self.SvnUrl()
msb@chromium.orge28e4982009-09-25 20:51:45 +000079
80 def testDir(self):
81 members = [
maruel@chromium.org2b9aa8e2010-08-25 20:01:42 +000082 'FullUrlForRelativeUrl', 'RunCommand',
maruel@chromium.org55e724e2010-03-11 19:36:49 +000083 'cleanup', 'diff', 'export', 'pack', 'relpath', 'revert',
maruel@chromium.org9eda4112010-06-11 18:56:10 +000084 'revinfo', 'runhooks', 'status', 'update',
tony@chromium.org4b5b1772010-04-08 01:52:56 +000085 'updatesingle', 'url',
msb@chromium.orge28e4982009-09-25 20:51:45 +000086 ]
87
88 # If you add a member, be sure to add the relevant test!
maruel@chromium.org9eda4112010-06-11 18:56:10 +000089 self.compareMembers(self._scm_wrapper('svn://a'), members)
msb@chromium.orge28e4982009-09-25 20:51:45 +000090
91 def testUnsupportedSCM(self):
maruel@chromium.org9eda4112010-06-11 18:56:10 +000092 args = ['gopher://foo', self.root_dir, self.relpath]
93 exception_msg = 'No SCM found for url gopher://foo'
94 self.assertRaisesError(exception_msg, self._scm_wrapper, *args)
msb@chromium.orge28e4982009-09-25 20:51:45 +000095
msb@chromium.orge6f78352010-01-13 17:05:33 +000096 def testSVNFullUrlForRelativeUrl(self):
97 self.url = 'svn://a/b/c/d'
98
99 self.mox.ReplayAll()
100 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
101 relpath=self.relpath)
102 self.assertEqual(scm.FullUrlForRelativeUrl('/crap'), 'svn://a/b/crap')
103
104 def testGITFullUrlForRelativeUrl(self):
105 self.url = 'git://a/b/c/d'
106
107 self.mox.ReplayAll()
108 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
109 relpath=self.relpath)
110 self.assertEqual(scm.FullUrlForRelativeUrl('/crap'), 'git://a/b/c/crap')
111
msb@chromium.orge28e4982009-09-25 20:51:45 +0000112 def testRunCommandException(self):
113 options = self.Options(verbose=False)
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000114 gclient_scm.os.path.exists(join(self.base_path, '.git')).AndReturn(False)
115 gclient_scm.os.path.exists(join(self.base_path, '.hg')).AndReturn(False)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000116
117 self.mox.ReplayAll()
118 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
119 relpath=self.relpath)
120 exception = "Unsupported argument(s): %s" % ','.join(self.args)
121 self.assertRaisesError(exception, scm.RunCommand,
122 'update', options, self.args)
123
124 def testRunCommandUnknown(self):
125 # TODO(maruel): if ever used.
maruel@chromium.orgdd057f32010-09-20 19:33:31 +0000126 pass
msb@chromium.orge28e4982009-09-25 20:51:45 +0000127
128 def testRevertMissing(self):
129 options = self.Options(verbose=True)
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000130 gclient_scm.os.path.isdir(self.base_path).AndReturn(False)
maruel@chromium.org2b9aa8e2010-08-25 20:01:42 +0000131 gclient_scm.scm.SVN.Capture(['--version']
132 ).AndReturn('svn, version 1.5.1 (r32289)')
msb@chromium.orge28e4982009-09-25 20:51:45 +0000133 # It'll to a checkout instead.
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000134 gclient_scm.os.path.exists(join(self.base_path, '.git')).AndReturn(False)
135 gclient_scm.os.path.exists(join(self.base_path, '.hg')).AndReturn(False)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000136 # Checkout.
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000137 gclient_scm.os.path.exists(self.base_path).AndReturn(False)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000138 files_list = self.mox.CreateMockAnything()
maruel@chromium.org03807072010-08-16 17:18:44 +0000139 gclient_scm.scm.SVN.RunAndGetFileList(options.verbose,
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000140 ['checkout', self.url, self.base_path,
maruel@chromium.org2b9aa8e2010-08-25 20:01:42 +0000141 '--force'],
maruel@chromium.org669600d2010-09-01 19:06:31 +0000142 cwd=self.root_dir,
maruel@chromium.org77e4eca2010-09-21 13:23:07 +0000143 file_list=files_list)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000144
145 self.mox.ReplayAll()
146 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
147 relpath=self.relpath)
148 scm.revert(options, self.args, files_list)
maruel@chromium.org54019f32010-09-09 13:50:11 +0000149 self.checkstdout(
150 ('\n_____ %s is missing, synching instead\n' % self.relpath))
msb@chromium.orge28e4982009-09-25 20:51:45 +0000151
152 def testRevertNone(self):
153 options = self.Options(verbose=True)
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000154 gclient_scm.os.path.isdir(self.base_path).AndReturn(True)
155 gclient_scm.scm.SVN.CaptureStatus(self.base_path).AndReturn([])
maruel@chromium.org03807072010-08-16 17:18:44 +0000156 gclient_scm.scm.SVN.RunAndGetFileList(options.verbose,
maruel@chromium.org5aeb7dd2009-11-17 18:09:01 +0000157 ['update', '--revision', 'BASE'],
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000158 cwd=self.base_path,
maruel@chromium.org77e4eca2010-09-21 13:23:07 +0000159 file_list=mox.IgnoreArg())
msb@chromium.orge28e4982009-09-25 20:51:45 +0000160
161 self.mox.ReplayAll()
162 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
163 relpath=self.relpath)
164 file_list = []
165 scm.revert(options, self.args, file_list)
166
167 def testRevert2Files(self):
168 options = self.Options(verbose=True)
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000169 gclient_scm.os.path.isdir(self.base_path).AndReturn(True)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000170 items = [
171 ('M ', 'a'),
172 ('A ', 'b'),
173 ]
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000174 file_path1 = join(self.base_path, 'a')
175 file_path2 = join(self.base_path, 'b')
176 gclient_scm.scm.SVN.CaptureStatus(self.base_path).AndReturn(items)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000177 gclient_scm.os.path.exists(file_path1).AndReturn(True)
178 gclient_scm.os.path.isfile(file_path1).AndReturn(True)
179 gclient_scm.os.remove(file_path1)
180 gclient_scm.os.path.exists(file_path2).AndReturn(True)
181 gclient_scm.os.path.isfile(file_path2).AndReturn(True)
182 gclient_scm.os.remove(file_path2)
maruel@chromium.org03807072010-08-16 17:18:44 +0000183 gclient_scm.scm.SVN.RunAndGetFileList(options.verbose,
maruel@chromium.org5aeb7dd2009-11-17 18:09:01 +0000184 ['update', '--revision', 'BASE'],
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000185 cwd=self.base_path,
maruel@chromium.org77e4eca2010-09-21 13:23:07 +0000186 file_list=mox.IgnoreArg())
msb@chromium.orge28e4982009-09-25 20:51:45 +0000187
188 self.mox.ReplayAll()
189 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
190 relpath=self.relpath)
191 file_list = []
192 scm.revert(options, self.args, file_list)
maruel@chromium.org54019f32010-09-09 13:50:11 +0000193 self.checkstdout(
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000194 ('%s\n%s\n' % (join(self.base_path, 'a'),
195 join(self.base_path, 'b'))))
msb@chromium.orge28e4982009-09-25 20:51:45 +0000196
197 def testRevertDirectory(self):
198 options = self.Options(verbose=True)
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000199 gclient_scm.os.path.isdir(self.base_path).AndReturn(True)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000200 items = [
201 ('~ ', 'a'),
202 ]
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000203 gclient_scm.scm.SVN.CaptureStatus(self.base_path).AndReturn(items)
204 file_path = join(self.base_path, 'a')
msb@chromium.orge28e4982009-09-25 20:51:45 +0000205 gclient_scm.os.path.exists(file_path).AndReturn(True)
206 gclient_scm.os.path.isfile(file_path).AndReturn(False)
msb@chromium.org32906d12010-01-12 18:22:09 +0000207 gclient_scm.os.path.islink(file_path).AndReturn(False)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000208 gclient_scm.os.path.isdir(file_path).AndReturn(True)
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000209 gclient_scm.gclient_utils.RemoveDirectory(file_path)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000210 file_list1 = []
maruel@chromium.org03807072010-08-16 17:18:44 +0000211 gclient_scm.scm.SVN.RunAndGetFileList(options.verbose,
maruel@chromium.org5aeb7dd2009-11-17 18:09:01 +0000212 ['update', '--revision', 'BASE'],
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000213 cwd=self.base_path,
maruel@chromium.org77e4eca2010-09-21 13:23:07 +0000214 file_list=mox.IgnoreArg())
msb@chromium.orge28e4982009-09-25 20:51:45 +0000215
216 self.mox.ReplayAll()
217 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
218 relpath=self.relpath)
219 file_list2 = []
220 scm.revert(options, self.args, file_list2)
maruel@chromium.org54019f32010-09-09 13:50:11 +0000221 self.checkstdout(('%s\n' % file_path))
msb@chromium.orge28e4982009-09-25 20:51:45 +0000222
223 def testStatus(self):
224 options = self.Options(verbose=True)
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000225 gclient_scm.os.path.isdir(self.base_path).AndReturn(True)
maruel@chromium.org77e4eca2010-09-21 13:23:07 +0000226 gclient_scm.scm.SVN.RunAndGetFileList(
227 options.verbose, ['status'] + self.args,
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000228 cwd=self.base_path, file_list=[]).AndReturn(None)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000229
230 self.mox.ReplayAll()
231 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
232 relpath=self.relpath)
233 file_list = []
234 self.assertEqual(scm.status(options, self.args, file_list), None)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000235
236 # TODO(maruel): TEST REVISIONS!!!
237 # TODO(maruel): TEST RELOCATE!!!
238 def testUpdateCheckout(self):
239 options = self.Options(verbose=True)
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000240 file_info = gclient_scm.gclient_utils.PrintableObject()
msb@chromium.orge28e4982009-09-25 20:51:45 +0000241 file_info.root = 'blah'
242 file_info.url = self.url
243 file_info.uuid = 'ABC'
244 file_info.revision = 42
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000245 gclient_scm.os.path.exists(join(self.base_path, '.git')).AndReturn(False)
246 gclient_scm.os.path.exists(join(self.base_path, '.hg')).AndReturn(False)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000247 # Checkout.
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000248 gclient_scm.os.path.exists(self.base_path).AndReturn(False)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000249 files_list = self.mox.CreateMockAnything()
maruel@chromium.org669600d2010-09-01 19:06:31 +0000250 gclient_scm.scm.SVN.Capture(['--version']
251 ).AndReturn('svn, version 1.5.1 (r32289)')
maruel@chromium.org03807072010-08-16 17:18:44 +0000252 gclient_scm.scm.SVN.RunAndGetFileList(options.verbose,
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000253 ['checkout', self.url, self.base_path,
maruel@chromium.org2b9aa8e2010-08-25 20:01:42 +0000254 '--force'],
maruel@chromium.org669600d2010-09-01 19:06:31 +0000255 cwd=self.root_dir,
maruel@chromium.org77e4eca2010-09-21 13:23:07 +0000256 file_list=files_list)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000257 self.mox.ReplayAll()
258 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
259 relpath=self.relpath)
260 scm.update(options, (), files_list)
261
262 def testUpdateUpdate(self):
263 options = self.Options(verbose=True)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000264 options.force = True
265 options.nohooks = False
266 file_info = {
267 'Repository Root': 'blah',
268 'URL': self.url,
269 'UUID': 'ABC',
270 'Revision': 42,
271 }
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000272 gclient_scm.os.path.exists(join(self.base_path, '.git')).AndReturn(False)
273 gclient_scm.os.path.exists(join(self.base_path, '.hg')).AndReturn(False)
maruel@chromium.orge407c9a2010-08-09 19:11:37 +0000274
275 # Verify no locked files.
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000276 dotted_path = join(self.base_path, '.')
277 gclient_scm.scm.SVN.CaptureStatus(dotted_path).AndReturn([])
maruel@chromium.orge407c9a2010-08-09 19:11:37 +0000278
msb@chromium.orge28e4982009-09-25 20:51:45 +0000279 # Checkout or update.
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000280 gclient_scm.os.path.exists(self.base_path).AndReturn(True)
281 gclient_scm.scm.SVN.CaptureInfo(dotted_path).AndReturn(file_info)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000282 # Cheat a bit here.
maruel@chromium.org54019f32010-09-09 13:50:11 +0000283 gclient_scm.scm.SVN.CaptureInfo(file_info['URL']).AndReturn(file_info)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000284 additional_args = []
285 if options.manually_grab_svn_rev:
286 additional_args = ['--revision', str(file_info['Revision'])]
maruel@chromium.org669600d2010-09-01 19:06:31 +0000287 gclient_scm.scm.SVN.Capture(['--version']
288 ).AndReturn('svn, version 1.5.1 (r32289)')
289 additional_args.append('--force')
msb@chromium.orge28e4982009-09-25 20:51:45 +0000290 files_list = []
maruel@chromium.org5aeb7dd2009-11-17 18:09:01 +0000291 gclient_scm.scm.SVN.RunAndGetFileList(
maruel@chromium.org03807072010-08-16 17:18:44 +0000292 options.verbose,
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000293 ['update', self.base_path] + additional_args,
maruel@chromium.org77e4eca2010-09-21 13:23:07 +0000294 cwd=self.root_dir, file_list=files_list)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000295
296 self.mox.ReplayAll()
297 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
298 relpath=self.relpath)
299 scm.update(options, (), files_list)
300
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000301 def testUpdateSingleCheckout(self):
302 options = self.Options(verbose=True)
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000303 file_info = {
304 'URL': self.url,
305 'Revision': 42,
306 }
tony@chromium.org57564662010-04-14 02:35:12 +0000307
308 # Checks to make sure that we support svn co --depth.
309 gclient_scm.scm.SVN.current_version = None
310 gclient_scm.scm.SVN.Capture(['--version']
311 ).AndReturn('svn, version 1.5.1 (r32289)')
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000312 gclient_scm.os.path.exists(join(self.base_path, '.svn')).AndReturn(False)
313 gclient_scm.os.path.exists(join(self.base_path, 'DEPS')).AndReturn(False)
tony@chromium.org57564662010-04-14 02:35:12 +0000314
maruel@chromium.orge407c9a2010-08-09 19:11:37 +0000315 # Verify no locked files.
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000316 dotted_path = join(self.base_path, '.')
317 gclient_scm.scm.SVN.CaptureStatus(dotted_path).AndReturn([])
maruel@chromium.orge407c9a2010-08-09 19:11:37 +0000318
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000319 # When checking out a single file, we issue an svn checkout and svn update.
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000320 files_list = self.mox.CreateMockAnything()
maruel@chromium.org669600d2010-09-01 19:06:31 +0000321 gclient_scm.gclient_utils.CheckCallAndFilterAndHeader(
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000322 ['svn', 'checkout', '--depth', 'empty', self.url, self.base_path],
maruel@chromium.org77e4eca2010-09-21 13:23:07 +0000323 always=True, cwd=self.root_dir)
maruel@chromium.org03807072010-08-16 17:18:44 +0000324 gclient_scm.scm.SVN.RunAndGetFileList(options.verbose, ['update', 'DEPS'],
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000325 cwd=self.base_path, file_list=files_list)
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000326
327 # Now we fall back on scm.update().
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000328 gclient_scm.os.path.exists(join(self.base_path, '.git')).AndReturn(False)
329 gclient_scm.os.path.exists(join(self.base_path, '.hg')).AndReturn(False)
330 gclient_scm.os.path.exists(self.base_path).AndReturn(True)
331 gclient_scm.scm.SVN.CaptureInfo(dotted_path).AndReturn(file_info)
maruel@chromium.org54019f32010-09-09 13:50:11 +0000332 gclient_scm.scm.SVN.CaptureInfo(file_info['URL']).AndReturn(file_info)
tony@chromium.org57564662010-04-14 02:35:12 +0000333
334 self.mox.ReplayAll()
335 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
336 relpath=self.relpath)
337 scm.updatesingle(options, ['DEPS'], files_list)
maruel@chromium.orgdd057f32010-09-20 19:33:31 +0000338 self.checkstdout('\n_____ %s at 42\n' % self.relpath)
tony@chromium.org57564662010-04-14 02:35:12 +0000339
340 def testUpdateSingleCheckoutSVN14(self):
341 options = self.Options(verbose=True)
tony@chromium.org57564662010-04-14 02:35:12 +0000342 file_info = {
343 'URL': self.url,
344 'Revision': 42,
345 }
346
347 # Checks to make sure that we support svn co --depth.
348 gclient_scm.scm.SVN.current_version = None
349 gclient_scm.scm.SVN.Capture(['--version']
350 ).AndReturn('svn, version 1.4.4 (r25188)')
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000351 gclient_scm.os.path.exists(self.base_path).AndReturn(True)
tony@chromium.org57564662010-04-14 02:35:12 +0000352
353 # When checking out a single file with svn 1.4, we use svn export
354 files_list = self.mox.CreateMockAnything()
maruel@chromium.org669600d2010-09-01 19:06:31 +0000355 gclient_scm.gclient_utils.CheckCallAndFilterAndHeader(
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000356 ['svn', 'export', join(self.url, 'DEPS'), join(self.base_path, 'DEPS')],
maruel@chromium.org77e4eca2010-09-21 13:23:07 +0000357 always=True, cwd=self.root_dir)
tony@chromium.org57564662010-04-14 02:35:12 +0000358
359 self.mox.ReplayAll()
360 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
361 relpath=self.relpath)
362 scm.updatesingle(options, ['DEPS'], files_list)
363
364 def testUpdateSingleCheckoutSVNUpgrade(self):
365 options = self.Options(verbose=True)
tony@chromium.org57564662010-04-14 02:35:12 +0000366 file_info = {
367 'URL': self.url,
368 'Revision': 42,
369 }
370
371 # Checks to make sure that we support svn co --depth.
372 gclient_scm.scm.SVN.current_version = None
373 gclient_scm.scm.SVN.Capture(['--version']
374 ).AndReturn('svn, version 1.5.1 (r32289)')
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000375 gclient_scm.os.path.exists(join(self.base_path, '.svn')).AndReturn(False)
tony@chromium.org57564662010-04-14 02:35:12 +0000376 # If DEPS already exists, assume we're upgrading from svn1.4, so delete
377 # the old DEPS file.
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000378 gclient_scm.os.path.exists(join(self.base_path, 'DEPS')).AndReturn(True)
379 gclient_scm.os.remove(join(self.base_path, 'DEPS'))
tony@chromium.org57564662010-04-14 02:35:12 +0000380
maruel@chromium.orge407c9a2010-08-09 19:11:37 +0000381 # Verify no locked files.
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000382 gclient_scm.scm.SVN.CaptureStatus(join(self.base_path, '.')).AndReturn([])
maruel@chromium.orge407c9a2010-08-09 19:11:37 +0000383
tony@chromium.org57564662010-04-14 02:35:12 +0000384 # When checking out a single file, we issue an svn checkout and svn update.
385 files_list = self.mox.CreateMockAnything()
maruel@chromium.org669600d2010-09-01 19:06:31 +0000386 gclient_scm.gclient_utils.CheckCallAndFilterAndHeader(
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000387 ['svn', 'checkout', '--depth', 'empty', self.url, self.base_path],
maruel@chromium.org77e4eca2010-09-21 13:23:07 +0000388 always=True, cwd=self.root_dir)
maruel@chromium.org03807072010-08-16 17:18:44 +0000389 gclient_scm.scm.SVN.RunAndGetFileList(options.verbose, ['update', 'DEPS'],
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000390 cwd=self.base_path, file_list=files_list)
tony@chromium.org57564662010-04-14 02:35:12 +0000391
392 # Now we fall back on scm.update().
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000393 gclient_scm.os.path.exists(join(self.base_path, '.git')).AndReturn(False)
394 gclient_scm.os.path.exists(join(self.base_path, '.hg')).AndReturn(False)
395 gclient_scm.os.path.exists(self.base_path).AndReturn(True)
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000396 gclient_scm.scm.SVN.CaptureInfo(
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000397 join(self.base_path, ".")).AndReturn(file_info)
maruel@chromium.org54019f32010-09-09 13:50:11 +0000398 gclient_scm.scm.SVN.CaptureInfo(file_info['URL']).AndReturn(file_info)
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000399
400 self.mox.ReplayAll()
401 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
402 relpath=self.relpath)
403 scm.updatesingle(options, ['DEPS'], files_list)
maruel@chromium.org54019f32010-09-09 13:50:11 +0000404 self.checkstdout(
405 ('\n_____ %s at 42\n' % self.relpath))
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000406
407 def testUpdateSingleUpdate(self):
408 options = self.Options(verbose=True)
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000409 file_info = {
410 'URL': self.url,
411 'Revision': 42,
412 }
tony@chromium.org57564662010-04-14 02:35:12 +0000413 # Checks to make sure that we support svn co --depth.
414 gclient_scm.scm.SVN.current_version = None
415 gclient_scm.scm.SVN.Capture(['--version']
416 ).AndReturn('svn, version 1.5.1 (r32289)')
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000417 gclient_scm.os.path.exists(join(self.base_path, '.svn')).AndReturn(True)
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000418
maruel@chromium.orge407c9a2010-08-09 19:11:37 +0000419 # Verify no locked files.
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000420 gclient_scm.scm.SVN.CaptureStatus(join(self.base_path, '.')).AndReturn([])
maruel@chromium.orge407c9a2010-08-09 19:11:37 +0000421
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000422 # Now we fall back on scm.update().
423 files_list = self.mox.CreateMockAnything()
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000424 gclient_scm.os.path.exists(join(self.base_path, '.git')).AndReturn(False)
425 gclient_scm.os.path.exists(join(self.base_path, '.hg')).AndReturn(False)
426 gclient_scm.os.path.exists(self.base_path).AndReturn(True)
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000427 gclient_scm.scm.SVN.CaptureInfo(
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000428 join(self.base_path, '.')).AndReturn(file_info)
maruel@chromium.org54019f32010-09-09 13:50:11 +0000429 gclient_scm.scm.SVN.CaptureInfo(file_info['URL']).AndReturn(file_info)
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000430
431 self.mox.ReplayAll()
432 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
433 relpath=self.relpath)
434 scm.updatesingle(options, ['DEPS'], files_list)
maruel@chromium.org54019f32010-09-09 13:50:11 +0000435 self.checkstdout('\n_____ %s at 42\n' % self.relpath)
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000436
msb@chromium.orge28e4982009-09-25 20:51:45 +0000437 def testUpdateGit(self):
438 options = self.Options(verbose=True)
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000439 file_path = gclient_scm.os.path.join(self.root_dir, self.relpath, '.git')
440 gclient_scm.os.path.exists(file_path).AndReturn(True)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000441
442 self.mox.ReplayAll()
443 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
444 relpath=self.relpath)
445 file_list = []
446 scm.update(options, self.args, file_list)
maruel@chromium.org54019f32010-09-09 13:50:11 +0000447 self.checkstdout(
448 ('________ found .git directory; skipping %s\n' % self.relpath))
msb@chromium.orge28e4982009-09-25 20:51:45 +0000449
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000450 def testUpdateHg(self):
451 options = self.Options(verbose=True)
452 gclient_scm.os.path.exists(join(self.base_path, '.git')).AndReturn(False)
453 gclient_scm.os.path.exists(join(self.base_path, '.hg')).AndReturn(True)
454
455 self.mox.ReplayAll()
456 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
457 relpath=self.relpath)
458 file_list = []
459 scm.update(options, self.args, file_list)
460 self.checkstdout(
461 ('________ found .hg directory; skipping %s\n' % self.relpath))
462
msb@chromium.orge28e4982009-09-25 20:51:45 +0000463
maruel@chromium.orgdd057f32010-09-20 19:33:31 +0000464class GitWrapperTestCase(GCBaseTestCase, StdoutCheck, TestCaseUtils,
465 unittest.TestCase):
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000466 """This class doesn't use pymox."""
msb@chromium.orge28e4982009-09-25 20:51:45 +0000467 class OptionsObject(object):
maruel@chromium.org8071c282010-09-20 19:44:19 +0000468 def __init__(self, verbose=False, revision=None):
msb@chromium.orge28e4982009-09-25 20:51:45 +0000469 self.verbose = verbose
470 self.revision = revision
471 self.manually_grab_svn_rev = True
472 self.deps_os = None
473 self.force = False
davemoore@chromium.org8bf27312010-02-19 17:29:44 +0000474 self.reset = False
msb@chromium.orge28e4982009-09-25 20:51:45 +0000475 self.nohooks = False
476
477 sample_git_import = """blob
478mark :1
479data 6
480Hello
481
482blob
483mark :2
484data 4
485Bye
486
487reset refs/heads/master
488commit refs/heads/master
489mark :3
490author Bob <bob@example.com> 1253744361 -0700
491committer Bob <bob@example.com> 1253744361 -0700
492data 8
493A and B
494M 100644 :1 a
495M 100644 :2 b
496
497blob
498mark :4
499data 10
500Hello
501You
502
503blob
504mark :5
505data 8
506Bye
507You
508
509commit refs/heads/origin
510mark :6
511author Alice <alice@example.com> 1253744424 -0700
512committer Alice <alice@example.com> 1253744424 -0700
513data 13
514Personalized
515from :3
516M 100644 :4 a
517M 100644 :5 b
518
519reset refs/heads/master
520from :3
521"""
msb@chromium.orge28e4982009-09-25 20:51:45 +0000522 def Options(self, *args, **kwargs):
maruel@chromium.org8071c282010-09-20 19:44:19 +0000523 return self.OptionsObject(*args, **kwargs)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000524
525 def CreateGitRepo(self, git_import, path):
maruel@chromium.orgd5800f12009-11-12 20:03:43 +0000526 """Do it for real."""
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000527 try:
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000528 Popen(['git', 'init', '-q'], stdout=PIPE, stderr=STDOUT,
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000529 cwd=path).communicate()
530 except OSError:
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000531 # git is not available, skip this test.
532 return False
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000533 Popen(['git', 'fast-import', '--quiet'], stdin=PIPE, stdout=PIPE,
534 stderr=STDOUT, cwd=path).communicate(input=git_import)
535 Popen(['git', 'checkout', '-q'], stdout=PIPE, stderr=STDOUT,
536 cwd=path).communicate()
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000537 Popen(['git', 'remote', 'add', '-f', 'origin', '.'], stdout=PIPE,
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000538 stderr=STDOUT, cwd=path).communicate()
539 Popen(['git', 'checkout', '-b', 'new', 'origin/master', '-q'], stdout=PIPE,
540 stderr=STDOUT, cwd=path).communicate()
541 Popen(['git', 'push', 'origin', 'origin/origin:origin/master', '-q'],
542 stdout=PIPE, stderr=STDOUT, cwd=path).communicate()
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000543 Popen(['git', 'config', '--unset', 'remote.origin.fetch'], stdout=PIPE,
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000544 stderr=STDOUT, cwd=path).communicate()
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000545 return True
msb@chromium.orge28e4982009-09-25 20:51:45 +0000546
msb@chromium.orge28e4982009-09-25 20:51:45 +0000547 def setUp(self):
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000548 TestCaseUtils.setUp(self)
549 unittest.TestCase.setUp(self)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000550 self.url = 'git://foo'
551 self.root_dir = tempfile.mkdtemp()
552 self.relpath = '.'
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000553 self.base_path = join(self.root_dir, self.relpath)
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000554 self.enabled = self.CreateGitRepo(self.sample_git_import, self.base_path)
maruel@chromium.orgdd057f32010-09-20 19:33:31 +0000555 StdoutCheck.setUp(self)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000556
557 def tearDown(self):
maruel@chromium.orgdd057f32010-09-20 19:33:31 +0000558 StdoutCheck.tearDown(self)
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000559 TestCaseUtils.tearDown(self)
560 unittest.TestCase.tearDown(self)
maruel@chromium.orgba551772010-02-03 18:21:42 +0000561 rmtree(self.root_dir)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000562
563 def testDir(self):
564 members = [
maruel@chromium.org55e724e2010-03-11 19:36:49 +0000565 'FullUrlForRelativeUrl', 'RunCommand',
msb@chromium.orge6f78352010-01-13 17:05:33 +0000566 'cleanup', 'diff', 'export', 'pack', 'relpath', 'revert',
maruel@chromium.org9eda4112010-06-11 18:56:10 +0000567 'revinfo', 'runhooks', 'status', 'update', 'url',
msb@chromium.orge28e4982009-09-25 20:51:45 +0000568 ]
569
570 # If you add a member, be sure to add the relevant test!
571 self.compareMembers(gclient_scm.CreateSCM(url=self.url), members)
572
573 def testRevertMissing(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000574 if not self.enabled:
575 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000576 options = self.Options()
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000577 file_path = join(self.base_path, 'a')
msb@chromium.orge28e4982009-09-25 20:51:45 +0000578 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
579 relpath=self.relpath)
580 file_list = []
nasser@codeaurora.orgb2b46312010-04-30 20:58:03 +0000581 scm.update(options, None, file_list)
582 gclient_scm.os.remove(file_path)
583 file_list = []
msb@chromium.orge28e4982009-09-25 20:51:45 +0000584 scm.revert(options, self.args, file_list)
585 self.assertEquals(file_list, [file_path])
586 file_list = []
587 scm.diff(options, self.args, file_list)
588 self.assertEquals(file_list, [])
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000589 self.checkstdout(
maruel@chromium.orgad80e3b2010-09-09 14:18:28 +0000590 ('\n_____ . at refs/heads/master\nUpdating 069c602..a7142dc\n'
591 'Fast-forward\n a | 1 +\n b | 1 +\n'
592 ' 2 files changed, 2 insertions(+), 0 deletions(-)\n\n\n'
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000593 '________ running \'git reset --hard origin/master\' in \'%s\'\n'
594 'HEAD is now at a7142dc Personalized\n') %
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000595 join(self.root_dir, '.'))
msb@chromium.orge28e4982009-09-25 20:51:45 +0000596
597 def testRevertNone(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000598 if not self.enabled:
599 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000600 options = self.Options()
601 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
602 relpath=self.relpath)
603 file_list = []
nasser@codeaurora.orgb2b46312010-04-30 20:58:03 +0000604 scm.update(options, None, file_list)
605 file_list = []
msb@chromium.orge28e4982009-09-25 20:51:45 +0000606 scm.revert(options, self.args, file_list)
607 self.assertEquals(file_list, [])
msb@chromium.org0f282062009-11-06 20:14:02 +0000608 self.assertEquals(scm.revinfo(options, self.args, None),
nasser@codeaurora.orgb2b46312010-04-30 20:58:03 +0000609 'a7142dc9f0009350b96a11f372b6ea658592aa95')
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000610 self.checkstdout(
maruel@chromium.orgad80e3b2010-09-09 14:18:28 +0000611 ('\n_____ . at refs/heads/master\nUpdating 069c602..a7142dc\n'
612 'Fast-forward\n a | 1 +\n b | 1 +\n'
613 ' 2 files changed, 2 insertions(+), 0 deletions(-)\n\n\n'
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000614 '________ running \'git reset --hard origin/master\' in \'%s\'\n'
615 'HEAD is now at a7142dc Personalized\n') %
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000616 join(self.root_dir, '.'))
msb@chromium.orge28e4982009-09-25 20:51:45 +0000617
618 def testRevertModified(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000619 if not self.enabled:
620 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000621 options = self.Options()
msb@chromium.orge28e4982009-09-25 20:51:45 +0000622 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
623 relpath=self.relpath)
624 file_list = []
nasser@codeaurora.orgb2b46312010-04-30 20:58:03 +0000625 scm.update(options, None, file_list)
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000626 file_path = join(self.base_path, 'a')
nasser@codeaurora.orgb2b46312010-04-30 20:58:03 +0000627 open(file_path, 'a').writelines('touched\n')
628 file_list = []
msb@chromium.orge28e4982009-09-25 20:51:45 +0000629 scm.revert(options, self.args, file_list)
630 self.assertEquals(file_list, [file_path])
631 file_list = []
632 scm.diff(options, self.args, file_list)
633 self.assertEquals(file_list, [])
msb@chromium.org0f282062009-11-06 20:14:02 +0000634 self.assertEquals(scm.revinfo(options, self.args, None),
nasser@codeaurora.orgb2b46312010-04-30 20:58:03 +0000635 'a7142dc9f0009350b96a11f372b6ea658592aa95')
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000636 self.checkstdout(
maruel@chromium.orgad80e3b2010-09-09 14:18:28 +0000637 ('\n_____ . at refs/heads/master\nUpdating 069c602..a7142dc\n'
638 'Fast-forward\n a | 1 +\n b | 1 +\n'
639 ' 2 files changed, 2 insertions(+), 0 deletions(-)\n\n\n'
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000640 '________ running \'git reset --hard origin/master\' in \'%s\'\n'
641 'HEAD is now at a7142dc Personalized\n') %
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000642 join(self.root_dir, '.'))
msb@chromium.orge28e4982009-09-25 20:51:45 +0000643
644 def testRevertNew(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000645 if not self.enabled:
646 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000647 options = self.Options()
nasser@codeaurora.orgb2b46312010-04-30 20:58:03 +0000648 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
649 relpath=self.relpath)
650 file_list = []
651 scm.update(options, None, file_list)
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000652 file_path = join(self.base_path, 'c')
msb@chromium.orge28e4982009-09-25 20:51:45 +0000653 f = open(file_path, 'w')
654 f.writelines('new\n')
655 f.close()
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000656 Popen(['git', 'add', 'c'], stdout=PIPE,
657 stderr=STDOUT, cwd=self.base_path).communicate()
msb@chromium.orge28e4982009-09-25 20:51:45 +0000658 file_list = []
659 scm.revert(options, self.args, file_list)
660 self.assertEquals(file_list, [file_path])
661 file_list = []
662 scm.diff(options, self.args, file_list)
663 self.assertEquals(file_list, [])
msb@chromium.org0f282062009-11-06 20:14:02 +0000664 self.assertEquals(scm.revinfo(options, self.args, None),
nasser@codeaurora.orgb2b46312010-04-30 20:58:03 +0000665 'a7142dc9f0009350b96a11f372b6ea658592aa95')
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000666 self.checkstdout(
maruel@chromium.orgad80e3b2010-09-09 14:18:28 +0000667 ('\n_____ . at refs/heads/master\nUpdating 069c602..a7142dc\n'
668 'Fast-forward\n a | 1 +\n b | 1 +\n'
669 ' 2 files changed, 2 insertions(+), 0 deletions(-)\n\n\n'
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000670 '________ running \'git reset --hard origin/master\' in \'%s\'\n'
671 'HEAD is now at a7142dc Personalized\n') %
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000672 join(self.root_dir, '.'))
msb@chromium.orge28e4982009-09-25 20:51:45 +0000673
674 def testStatusNew(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000675 if not self.enabled:
676 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000677 options = self.Options()
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000678 file_path = join(self.base_path, 'a')
msb@chromium.orge28e4982009-09-25 20:51:45 +0000679 open(file_path, 'a').writelines('touched\n')
680 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
681 relpath=self.relpath)
682 file_list = []
683 scm.status(options, self.args, file_list)
684 self.assertEquals(file_list, [file_path])
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000685 self.checkstdout(
686 ('\n________ running \'git diff --name-status '
687 '069c602044c5388d2d15c3f875b057c852003458\' in \'%s\'\nM\ta\n') %
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000688 join(self.root_dir, '.'))
msb@chromium.orge28e4982009-09-25 20:51:45 +0000689
690 def testStatus2New(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000691 if not self.enabled:
692 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000693 options = self.Options()
694 expected_file_list = []
695 for f in ['a', 'b']:
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000696 file_path = join(self.base_path, f)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000697 open(file_path, 'a').writelines('touched\n')
698 expected_file_list.extend([file_path])
699 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
700 relpath=self.relpath)
701 file_list = []
702 scm.status(options, self.args, file_list)
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000703 expected_file_list = [join(self.base_path, x) for x in ['a', 'b']]
msb@chromium.orge28e4982009-09-25 20:51:45 +0000704 self.assertEquals(sorted(file_list), expected_file_list)
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000705 self.checkstdout(
706 ('\n________ running \'git diff --name-status '
707 '069c602044c5388d2d15c3f875b057c852003458\' in \'%s\'\nM\ta\nM\tb\n') %
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000708 join(self.root_dir, '.'))
msb@chromium.orge28e4982009-09-25 20:51:45 +0000709
710 def testUpdateCheckout(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000711 if not self.enabled:
712 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000713 options = self.Options(verbose=True)
714 root_dir = tempfile.mkdtemp()
715 relpath = 'foo'
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000716 base_path = join(root_dir, relpath)
717 url = join(self.base_path, '.git')
msb@chromium.orge28e4982009-09-25 20:51:45 +0000718 try:
719 scm = gclient_scm.CreateSCM(url=url, root_dir=root_dir,
720 relpath=relpath)
721 file_list = []
722 scm.update(options, (), file_list)
723 self.assertEquals(len(file_list), 2)
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000724 self.assert_(gclient_scm.os.path.isfile(join(base_path, 'a')))
msb@chromium.org0f282062009-11-06 20:14:02 +0000725 self.assertEquals(scm.revinfo(options, (), None),
msb@chromium.orge28e4982009-09-25 20:51:45 +0000726 '069c602044c5388d2d15c3f875b057c852003458')
727 finally:
maruel@chromium.orgba551772010-02-03 18:21:42 +0000728 rmtree(root_dir)
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000729 self.checkstdout(
730 ('\n_____ foo at refs/heads/master\n\n'
731 '________ running \'git clone -b master --verbose %s %s\' in \'%s\'\n'
732 'Initialized empty Git repository in %s\n') %
733 (join(self.root_dir, '.', '.git'), join(root_dir, 'foo'), root_dir,
bauerb@chromium.org4acf7422010-10-07 14:20:08 +0000734 join(gclient_scm.os.path.realpath(root_dir), 'foo', '.git') + '/'))
msb@chromium.orge28e4982009-09-25 20:51:45 +0000735
736 def testUpdateUpdate(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000737 if not self.enabled:
738 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000739 options = self.Options()
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000740 expected_file_list = [join(self.base_path, x) for x in ['a', 'b']]
msb@chromium.orge28e4982009-09-25 20:51:45 +0000741 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
742 relpath=self.relpath)
743 file_list = []
744 scm.update(options, (), file_list)
msb@chromium.org0f282062009-11-06 20:14:02 +0000745 self.assertEquals(file_list, expected_file_list)
746 self.assertEquals(scm.revinfo(options, (), None),
msb@chromium.orge28e4982009-09-25 20:51:45 +0000747 'a7142dc9f0009350b96a11f372b6ea658592aa95')
maruel@chromium.orgad80e3b2010-09-09 14:18:28 +0000748 self.checkstdout(
749 '\n_____ . at refs/heads/master\n'
750 'Updating 069c602..a7142dc\nFast-forward\n a | 1 +\n b | 1 +\n'
751 ' 2 files changed, 2 insertions(+), 0 deletions(-)\n\n')
msb@chromium.orge28e4982009-09-25 20:51:45 +0000752
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000753 def testUpdateUnstagedConflict(self):
754 if not self.enabled:
755 return
756 options = self.Options()
757 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
758 relpath=self.relpath)
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000759 file_path = join(self.base_path, 'b')
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000760 f = open(file_path, 'w').writelines('conflict\n')
761 exception = (
762 "error: Your local changes to 'b' would be overwritten by merge. "
763 "Aborting.\n"
764 "Please, commit your changes or stash them before you can merge.\n")
765 self.assertRaisesError(exception, scm.update, options, (), [])
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000766 self.checkstdout('\n_____ . at refs/heads/master\n')
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000767
msb@chromium.org5bde4852009-12-14 16:47:12 +0000768 def testUpdateConflict(self):
769 if not self.enabled:
770 return
771 options = self.Options()
772 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
773 relpath=self.relpath)
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000774 file_path = join(self.base_path, 'b')
msb@chromium.org5bde4852009-12-14 16:47:12 +0000775 f = open(file_path, 'w').writelines('conflict\n')
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000776 scm._Run(['commit', '-am', 'test'], options)
777 __builtin__.raw_input = lambda x: 'y'
778 exception = ('Conflict while rebasing this branch.\n'
779 'Fix the conflict and run gclient again.\n'
780 'See \'man git-rebase\' for details.\n')
msb@chromium.org5bde4852009-12-14 16:47:12 +0000781 self.assertRaisesError(exception, scm.update, options, (), [])
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000782 exception = ('\n____ . at refs/heads/master\n'
783 '\tYou have unstaged changes.\n'
784 '\tPlease commit, stash, or reset.\n')
msb@chromium.org5bde4852009-12-14 16:47:12 +0000785 self.assertRaisesError(exception, scm.update, options, (), [])
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000786 # The hash always changes. Use a cheap trick.
787 start = ('\n________ running \'git commit -am test\' in \'%s\'\n'
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000788 '[new ') % join(self.root_dir, '.')
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000789 end = ('] test\n 1 files changed, 1 insertions(+), '
790 '1 deletions(-)\n\n_____ . at refs/heads/master\n'
791 'Attempting rebase onto refs/remotes/origin/master...\n')
maruel@chromium.orgdd057f32010-09-20 19:33:31 +0000792 self.assertTrue(gclient_scm.sys.stdout.getvalue().startswith(start))
793 self.assertTrue(gclient_scm.sys.stdout.getvalue().endswith(end))
794 self.assertEquals(len(gclient_scm.sys.stdout.getvalue()),
795 len(start) + len(end) + 7)
796 gclient_scm.sys.stdout.close()
msb@chromium.org5bde4852009-12-14 16:47:12 +0000797
msb@chromium.orge4af1ab2010-01-13 21:26:09 +0000798 def testUpdateNotGit(self):
799 if not self.enabled:
800 return
801 options = self.Options()
802 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
803 relpath=self.relpath)
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000804 git_path = join(self.base_path, '.git')
maruel@chromium.orgba551772010-02-03 18:21:42 +0000805 rename(git_path, git_path + 'foo')
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000806 exception = ('\n____ . at refs/heads/master\n'
807 '\tPath is not a git repo. No .git dir.\n'
808 '\tTo resolve:\n'
809 '\t\trm -rf .\n'
810 '\tAnd run gclient sync again\n')
msb@chromium.orge4af1ab2010-01-13 21:26:09 +0000811 self.assertRaisesError(exception, scm.update, options, (), [])
812
msb@chromium.org0f282062009-11-06 20:14:02 +0000813 def testRevinfo(self):
814 if not self.enabled:
815 return
816 options = self.Options()
817 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
818 relpath=self.relpath)
819 rev_info = scm.revinfo(options, (), None)
820 self.assertEquals(rev_info, '069c602044c5388d2d15c3f875b057c852003458')
821
msb@chromium.orge28e4982009-09-25 20:51:45 +0000822
msb@chromium.orge28e4982009-09-25 20:51:45 +0000823if __name__ == '__main__':
msb@chromium.orge28e4982009-09-25 20:51:45 +0000824 unittest.main()
825
826# vim: ts=2:sw=2:tw=80:et: