blob: 46e553605eb23b3d8c8c6a88f253fa1a61929536 [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.org8ef5f542009-11-12 02:05:02 +000011from subprocess import Popen, PIPE, STDOUT
msb@chromium.orge28e4982009-09-25 20:51:45 +000012import tempfile
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +000013import __builtin__
msb@chromium.orge28e4982009-09-25 20:51:45 +000014
maruel@chromium.orgba551772010-02-03 18:21:42 +000015# Fixes include path.
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +000016from super_mox import mox, SuperMoxBaseTestBase, SuperMoxTestBase
maruel@chromium.orgba551772010-02-03 18:21:42 +000017
msb@chromium.orge28e4982009-09-25 20:51:45 +000018import gclient_scm
maruel@chromium.org8ef5f542009-11-12 02:05:02 +000019from gclient_test import BaseTestCase as GCBaseTestCase
msb@chromium.orge28e4982009-09-25 20:51:45 +000020
21
maruel@chromium.org8ef5f542009-11-12 02:05:02 +000022class BaseTestCase(GCBaseTestCase):
23 def setUp(self):
24 GCBaseTestCase.setUp(self)
25 self.mox.StubOutWithMock(gclient_scm.gclient_utils, 'FileRead')
26 self.mox.StubOutWithMock(gclient_scm.gclient_utils, 'FileWrite')
27 self.mox.StubOutWithMock(gclient_scm.gclient_utils, 'SubprocessCall')
28 self.mox.StubOutWithMock(gclient_scm.gclient_utils, 'RemoveDirectory')
maruel@chromium.org5aeb7dd2009-11-17 18:09:01 +000029 self._CaptureSVNInfo = gclient_scm.scm.SVN.CaptureInfo
30 self.mox.StubOutWithMock(gclient_scm.scm.SVN, 'Capture')
31 self.mox.StubOutWithMock(gclient_scm.scm.SVN, 'CaptureInfo')
32 self.mox.StubOutWithMock(gclient_scm.scm.SVN, 'CaptureStatus')
33 self.mox.StubOutWithMock(gclient_scm.scm.SVN, 'Run')
34 self.mox.StubOutWithMock(gclient_scm.scm.SVN, 'RunAndGetFileList')
maruel@chromium.org8ef5f542009-11-12 02:05:02 +000035 self._scm_wrapper = gclient_scm.CreateSCM
36
37
38class SVNWrapperTestCase(BaseTestCase):
msb@chromium.orge28e4982009-09-25 20:51:45 +000039 class OptionsObject(object):
40 def __init__(self, test_case, verbose=False, revision=None):
41 self.verbose = verbose
42 self.revision = revision
43 self.manually_grab_svn_rev = True
44 self.deps_os = None
45 self.force = False
davemoore@chromium.org8bf27312010-02-19 17:29:44 +000046 self.reset = False
msb@chromium.orge28e4982009-09-25 20:51:45 +000047 self.nohooks = False
48
maruel@chromium.org8ef5f542009-11-12 02:05:02 +000049 def Options(self, *args, **kwargs):
50 return self.OptionsObject(self, *args, **kwargs)
51
msb@chromium.orge28e4982009-09-25 20:51:45 +000052 def setUp(self):
maruel@chromium.org8ef5f542009-11-12 02:05:02 +000053 BaseTestCase.setUp(self)
msb@chromium.orge28e4982009-09-25 20:51:45 +000054 self.root_dir = self.Dir()
55 self.args = self.Args()
56 self.url = self.Url()
57 self.relpath = 'asf'
58
59 def testDir(self):
60 members = [
maruel@chromium.org55e724e2010-03-11 19:36:49 +000061 'FullUrlForRelativeUrl', 'RunCommand',
62 'cleanup', 'diff', 'export', 'pack', 'relpath', 'revert',
tony@chromium.org4b5b1772010-04-08 01:52:56 +000063 'revinfo', 'runhooks', 'scm_name', 'status', 'update',
64 'updatesingle', 'url',
msb@chromium.orge28e4982009-09-25 20:51:45 +000065 ]
66
67 # If you add a member, be sure to add the relevant test!
68 self.compareMembers(self._scm_wrapper(), members)
69
70 def testUnsupportedSCM(self):
71 args = [self.url, self.root_dir, self.relpath]
72 kwargs = {'scm_name' : 'foo'}
73 exception_msg = 'Unsupported scm %(scm_name)s' % kwargs
74 self.assertRaisesError(exception_msg, self._scm_wrapper, *args, **kwargs)
75
msb@chromium.orge6f78352010-01-13 17:05:33 +000076 def testSVNFullUrlForRelativeUrl(self):
77 self.url = 'svn://a/b/c/d'
78
79 self.mox.ReplayAll()
80 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
81 relpath=self.relpath)
82 self.assertEqual(scm.FullUrlForRelativeUrl('/crap'), 'svn://a/b/crap')
83
84 def testGITFullUrlForRelativeUrl(self):
85 self.url = 'git://a/b/c/d'
86
87 self.mox.ReplayAll()
88 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
89 relpath=self.relpath)
90 self.assertEqual(scm.FullUrlForRelativeUrl('/crap'), 'git://a/b/c/crap')
91
msb@chromium.orge28e4982009-09-25 20:51:45 +000092 def testRunCommandException(self):
93 options = self.Options(verbose=False)
maruel@chromium.org8ef5f542009-11-12 02:05:02 +000094 file_path = gclient_scm.os.path.join(self.root_dir, self.relpath, '.git')
95 gclient_scm.os.path.exists(file_path).AndReturn(False)
msb@chromium.orge28e4982009-09-25 20:51:45 +000096
97 self.mox.ReplayAll()
98 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
99 relpath=self.relpath)
100 exception = "Unsupported argument(s): %s" % ','.join(self.args)
101 self.assertRaisesError(exception, scm.RunCommand,
102 'update', options, self.args)
103
104 def testRunCommandUnknown(self):
105 # TODO(maruel): if ever used.
106 pass
107
108 def testRevertMissing(self):
109 options = self.Options(verbose=True)
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000110 base_path = gclient_scm.os.path.join(self.root_dir, self.relpath)
111 gclient_scm.os.path.isdir(base_path).AndReturn(False)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000112 # It'll to a checkout instead.
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000113 gclient_scm.os.path.exists(gclient_scm.os.path.join(base_path, '.git')
114 ).AndReturn(False)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000115 print("\n_____ %s is missing, synching instead" % self.relpath)
116 # Checkout.
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000117 gclient_scm.os.path.exists(base_path).AndReturn(False)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000118 files_list = self.mox.CreateMockAnything()
maruel@chromium.org5aeb7dd2009-11-17 18:09:01 +0000119 gclient_scm.scm.SVN.RunAndGetFileList(options,
120 ['checkout', self.url, base_path],
121 self.root_dir, files_list)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000122
123 self.mox.ReplayAll()
124 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
125 relpath=self.relpath)
126 scm.revert(options, self.args, files_list)
127
128 def testRevertNone(self):
129 options = self.Options(verbose=True)
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000130 base_path = gclient_scm.os.path.join(self.root_dir, self.relpath)
131 gclient_scm.os.path.isdir(base_path).AndReturn(True)
maruel@chromium.org5aeb7dd2009-11-17 18:09:01 +0000132 gclient_scm.scm.SVN.CaptureStatus(base_path).AndReturn([])
133 gclient_scm.scm.SVN.RunAndGetFileList(options,
134 ['update', '--revision', 'BASE'],
135 base_path, mox.IgnoreArg())
msb@chromium.orge28e4982009-09-25 20:51:45 +0000136
137 self.mox.ReplayAll()
138 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
139 relpath=self.relpath)
140 file_list = []
141 scm.revert(options, self.args, file_list)
142
143 def testRevert2Files(self):
144 options = self.Options(verbose=True)
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000145 base_path = gclient_scm.os.path.join(self.root_dir, self.relpath)
146 gclient_scm.os.path.isdir(base_path).AndReturn(True)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000147 items = [
148 ('M ', 'a'),
149 ('A ', 'b'),
150 ]
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000151 file_path1 = gclient_scm.os.path.join(base_path, 'a')
152 file_path2 = gclient_scm.os.path.join(base_path, 'b')
maruel@chromium.org5aeb7dd2009-11-17 18:09:01 +0000153 gclient_scm.scm.SVN.CaptureStatus(base_path).AndReturn(items)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000154 gclient_scm.os.path.exists(file_path1).AndReturn(True)
155 gclient_scm.os.path.isfile(file_path1).AndReturn(True)
156 gclient_scm.os.remove(file_path1)
157 gclient_scm.os.path.exists(file_path2).AndReturn(True)
158 gclient_scm.os.path.isfile(file_path2).AndReturn(True)
159 gclient_scm.os.remove(file_path2)
maruel@chromium.org5aeb7dd2009-11-17 18:09:01 +0000160 gclient_scm.scm.SVN.RunAndGetFileList(options,
161 ['update', '--revision', 'BASE'],
162 base_path, mox.IgnoreArg())
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000163 print(gclient_scm.os.path.join(base_path, 'a'))
164 print(gclient_scm.os.path.join(base_path, 'b'))
msb@chromium.orge28e4982009-09-25 20:51:45 +0000165
166 self.mox.ReplayAll()
167 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
168 relpath=self.relpath)
169 file_list = []
170 scm.revert(options, self.args, file_list)
171
172 def testRevertDirectory(self):
173 options = self.Options(verbose=True)
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000174 base_path = gclient_scm.os.path.join(self.root_dir, self.relpath)
175 gclient_scm.os.path.isdir(base_path).AndReturn(True)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000176 items = [
177 ('~ ', 'a'),
178 ]
maruel@chromium.org5aeb7dd2009-11-17 18:09:01 +0000179 gclient_scm.scm.SVN.CaptureStatus(base_path).AndReturn(items)
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000180 file_path = gclient_scm.os.path.join(base_path, 'a')
msb@chromium.orge28e4982009-09-25 20:51:45 +0000181 print(file_path)
182 gclient_scm.os.path.exists(file_path).AndReturn(True)
183 gclient_scm.os.path.isfile(file_path).AndReturn(False)
msb@chromium.org32906d12010-01-12 18:22:09 +0000184 gclient_scm.os.path.islink(file_path).AndReturn(False)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000185 gclient_scm.os.path.isdir(file_path).AndReturn(True)
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000186 gclient_scm.gclient_utils.RemoveDirectory(file_path)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000187 file_list1 = []
maruel@chromium.org5aeb7dd2009-11-17 18:09:01 +0000188 gclient_scm.scm.SVN.RunAndGetFileList(options,
189 ['update', '--revision', 'BASE'],
190 base_path, mox.IgnoreArg())
msb@chromium.orge28e4982009-09-25 20:51:45 +0000191
192 self.mox.ReplayAll()
193 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
194 relpath=self.relpath)
195 file_list2 = []
196 scm.revert(options, self.args, file_list2)
197
198 def testStatus(self):
199 options = self.Options(verbose=True)
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000200 base_path = gclient_scm.os.path.join(self.root_dir, self.relpath)
201 gclient_scm.os.path.isdir(base_path).AndReturn(True)
maruel@chromium.org5aeb7dd2009-11-17 18:09:01 +0000202 gclient_scm.scm.SVN.RunAndGetFileList(options,
203 ['status'] + self.args,
204 base_path, []).AndReturn(None)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000205
206 self.mox.ReplayAll()
207 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
208 relpath=self.relpath)
209 file_list = []
210 self.assertEqual(scm.status(options, self.args, file_list), None)
211
212
213 # TODO(maruel): TEST REVISIONS!!!
214 # TODO(maruel): TEST RELOCATE!!!
215 def testUpdateCheckout(self):
216 options = self.Options(verbose=True)
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000217 base_path = gclient_scm.os.path.join(self.root_dir, self.relpath)
218 file_info = gclient_scm.gclient_utils.PrintableObject()
msb@chromium.orge28e4982009-09-25 20:51:45 +0000219 file_info.root = 'blah'
220 file_info.url = self.url
221 file_info.uuid = 'ABC'
222 file_info.revision = 42
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000223 gclient_scm.os.path.exists(gclient_scm.os.path.join(base_path, '.git')
224 ).AndReturn(False)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000225 # Checkout.
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000226 gclient_scm.os.path.exists(base_path).AndReturn(False)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000227 files_list = self.mox.CreateMockAnything()
maruel@chromium.org5aeb7dd2009-11-17 18:09:01 +0000228 gclient_scm.scm.SVN.RunAndGetFileList(options,
229 ['checkout', self.url, base_path],
230 self.root_dir, files_list)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000231 self.mox.ReplayAll()
232 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
233 relpath=self.relpath)
234 scm.update(options, (), files_list)
235
236 def testUpdateUpdate(self):
237 options = self.Options(verbose=True)
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000238 base_path = gclient_scm.os.path.join(self.root_dir, self.relpath)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000239 options.force = True
240 options.nohooks = False
241 file_info = {
242 'Repository Root': 'blah',
243 'URL': self.url,
244 'UUID': 'ABC',
245 'Revision': 42,
246 }
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000247 gclient_scm.os.path.exists(gclient_scm.os.path.join(base_path, '.git')
248 ).AndReturn(False)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000249 # Checkout or update.
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000250 gclient_scm.os.path.exists(base_path).AndReturn(True)
maruel@chromium.org5aeb7dd2009-11-17 18:09:01 +0000251 gclient_scm.scm.SVN.CaptureInfo(
252 gclient_scm.os.path.join(base_path, "."), '.'
253 ).AndReturn(file_info)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000254 # Cheat a bit here.
maruel@chromium.org5aeb7dd2009-11-17 18:09:01 +0000255 gclient_scm.scm.SVN.CaptureInfo(file_info['URL'], '.').AndReturn(file_info)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000256 additional_args = []
257 if options.manually_grab_svn_rev:
258 additional_args = ['--revision', str(file_info['Revision'])]
259 files_list = []
maruel@chromium.org5aeb7dd2009-11-17 18:09:01 +0000260 gclient_scm.scm.SVN.RunAndGetFileList(
261 options,
262 ['update', base_path] + additional_args,
263 self.root_dir, files_list)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000264
265 self.mox.ReplayAll()
266 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
267 relpath=self.relpath)
268 scm.update(options, (), files_list)
269
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000270 def testUpdateSingleCheckout(self):
271 options = self.Options(verbose=True)
272 base_path = gclient_scm.os.path.join(self.root_dir, self.relpath)
273 file_info = {
274 'URL': self.url,
275 'Revision': 42,
276 }
277 # When checking out a single file, we issue an svn checkout and svn update.
278 gclient_scm.os.path.exists(base_path).AndReturn(False)
279 files_list = self.mox.CreateMockAnything()
280 gclient_scm.scm.SVN.Run(
281 ['checkout', '--depth', 'empty', self.url, base_path], self.root_dir)
282 gclient_scm.scm.SVN.RunAndGetFileList(options, ['update', 'DEPS'],
283 gclient_scm.os.path.join(self.root_dir, self.relpath), files_list)
284
285 # Now we fall back on scm.update().
286 gclient_scm.os.path.exists(gclient_scm.os.path.join(base_path, '.git')
287 ).AndReturn(False)
288 gclient_scm.os.path.exists(base_path).AndReturn(True)
289 gclient_scm.scm.SVN.CaptureInfo(
290 gclient_scm.os.path.join(base_path, "."), '.'
291 ).AndReturn(file_info)
292 gclient_scm.scm.SVN.CaptureInfo(file_info['URL'], '.').AndReturn(file_info)
293 print("\n_____ %s at 42" % self.relpath)
294
295 self.mox.ReplayAll()
296 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
297 relpath=self.relpath)
298 scm.updatesingle(options, ['DEPS'], files_list)
299
300 def testUpdateSingleUpdate(self):
301 options = self.Options(verbose=True)
302 base_path = gclient_scm.os.path.join(self.root_dir, self.relpath)
303 file_info = {
304 'URL': self.url,
305 'Revision': 42,
306 }
307 gclient_scm.os.path.exists(base_path).AndReturn(True)
308
309 # Now we fall back on scm.update().
310 files_list = self.mox.CreateMockAnything()
311 gclient_scm.os.path.exists(gclient_scm.os.path.join(base_path, '.git')
312 ).AndReturn(False)
313 gclient_scm.os.path.exists(base_path).AndReturn(True)
314 gclient_scm.scm.SVN.CaptureInfo(
315 gclient_scm.os.path.join(base_path, "."), '.'
316 ).AndReturn(file_info)
317 gclient_scm.scm.SVN.CaptureInfo(file_info['URL'], '.').AndReturn(file_info)
318 print("\n_____ %s at 42" % self.relpath)
319
320 self.mox.ReplayAll()
321 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
322 relpath=self.relpath)
323 scm.updatesingle(options, ['DEPS'], files_list)
324
msb@chromium.orge28e4982009-09-25 20:51:45 +0000325 def testUpdateGit(self):
326 options = self.Options(verbose=True)
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000327 file_path = gclient_scm.os.path.join(self.root_dir, self.relpath, '.git')
328 gclient_scm.os.path.exists(file_path).AndReturn(True)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000329 print("________ found .git directory; skipping %s" % self.relpath)
330
331 self.mox.ReplayAll()
332 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
333 relpath=self.relpath)
334 file_list = []
335 scm.update(options, self.args, file_list)
336
msb@chromium.orge28e4982009-09-25 20:51:45 +0000337
msb@chromium.org5bde4852009-12-14 16:47:12 +0000338class GitWrapperTestCase(BaseTestCase):
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000339 """This class doesn't use pymox."""
msb@chromium.orge28e4982009-09-25 20:51:45 +0000340 class OptionsObject(object):
341 def __init__(self, test_case, verbose=False, revision=None):
342 self.verbose = verbose
343 self.revision = revision
344 self.manually_grab_svn_rev = True
345 self.deps_os = None
346 self.force = False
davemoore@chromium.org8bf27312010-02-19 17:29:44 +0000347 self.reset = False
msb@chromium.orge28e4982009-09-25 20:51:45 +0000348 self.nohooks = False
349
350 sample_git_import = """blob
351mark :1
352data 6
353Hello
354
355blob
356mark :2
357data 4
358Bye
359
360reset refs/heads/master
361commit refs/heads/master
362mark :3
363author Bob <bob@example.com> 1253744361 -0700
364committer Bob <bob@example.com> 1253744361 -0700
365data 8
366A and B
367M 100644 :1 a
368M 100644 :2 b
369
370blob
371mark :4
372data 10
373Hello
374You
375
376blob
377mark :5
378data 8
379Bye
380You
381
382commit refs/heads/origin
383mark :6
384author Alice <alice@example.com> 1253744424 -0700
385committer Alice <alice@example.com> 1253744424 -0700
386data 13
387Personalized
388from :3
389M 100644 :4 a
390M 100644 :5 b
391
392reset refs/heads/master
393from :3
394"""
msb@chromium.orge28e4982009-09-25 20:51:45 +0000395 def Options(self, *args, **kwargs):
396 return self.OptionsObject(self, *args, **kwargs)
397
398 def CreateGitRepo(self, git_import, path):
maruel@chromium.orgd5800f12009-11-12 20:03:43 +0000399 """Do it for real."""
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000400 try:
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000401 Popen(['git', 'init'], stdout=PIPE, stderr=STDOUT,
402 cwd=path).communicate()
403 except OSError:
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000404 # git is not available, skip this test.
405 return False
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000406 Popen(['git', 'fast-import'], stdin=PIPE, stdout=PIPE, stderr=STDOUT,
407 cwd=path).communicate(input=git_import)
408 Popen(['git', 'checkout'], stdout=PIPE, stderr=STDOUT,
409 cwd=path).communicate()
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000410 Popen(['git', 'remote', 'add', '-f', 'origin', '.'], stdout=PIPE,
411 stderr=STDOUT, cwd=path).communicate()
412 Popen(['git', 'checkout', '-b', 'new', 'origin/master'], stdout=PIPE,
413 stderr=STDOUT, cwd=path).communicate()
414 Popen(['git', 'push', 'origin', 'origin/origin:origin/master'], stdout=PIPE,
415 stderr=STDOUT, cwd=path).communicate()
416 Popen(['git', 'config', '--unset', 'remote.origin.fetch'], stdout=PIPE,
417 stderr=STDOUT, cwd=path).communicate()
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000418 return True
msb@chromium.orge28e4982009-09-25 20:51:45 +0000419
msb@chromium.orge28e4982009-09-25 20:51:45 +0000420 def setUp(self):
msb@chromium.orge28e4982009-09-25 20:51:45 +0000421 self.args = self.Args()
422 self.url = 'git://foo'
423 self.root_dir = tempfile.mkdtemp()
424 self.relpath = '.'
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000425 self.base_path = gclient_scm.os.path.join(self.root_dir, self.relpath)
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000426 self.enabled = self.CreateGitRepo(self.sample_git_import, self.base_path)
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000427 SuperMoxBaseTestBase.setUp(self)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000428
429 def tearDown(self):
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000430 SuperMoxBaseTestBase.tearDown(self)
maruel@chromium.orgba551772010-02-03 18:21:42 +0000431 rmtree(self.root_dir)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000432
433 def testDir(self):
434 members = [
maruel@chromium.org55e724e2010-03-11 19:36:49 +0000435 'FullUrlForRelativeUrl', 'RunCommand',
msb@chromium.orge6f78352010-01-13 17:05:33 +0000436 'cleanup', 'diff', 'export', 'pack', 'relpath', 'revert',
maruel@chromium.org5aeb7dd2009-11-17 18:09:01 +0000437 'revinfo', 'runhooks', 'scm_name', 'status', 'update', 'url',
msb@chromium.orge28e4982009-09-25 20:51:45 +0000438 ]
439
440 # If you add a member, be sure to add the relevant test!
441 self.compareMembers(gclient_scm.CreateSCM(url=self.url), members)
442
443 def testRevertMissing(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000444 if not self.enabled:
445 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000446 options = self.Options()
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000447 file_path = gclient_scm.os.path.join(self.base_path, 'a')
448 gclient_scm.os.remove(file_path)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000449 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
450 relpath=self.relpath)
451 file_list = []
452 scm.revert(options, self.args, file_list)
453 self.assertEquals(file_list, [file_path])
454 file_list = []
455 scm.diff(options, self.args, file_list)
456 self.assertEquals(file_list, [])
457
458 def testRevertNone(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000459 if not self.enabled:
460 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000461 options = self.Options()
462 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
463 relpath=self.relpath)
464 file_list = []
465 scm.revert(options, self.args, file_list)
466 self.assertEquals(file_list, [])
msb@chromium.org0f282062009-11-06 20:14:02 +0000467 self.assertEquals(scm.revinfo(options, self.args, None),
msb@chromium.orge28e4982009-09-25 20:51:45 +0000468 '069c602044c5388d2d15c3f875b057c852003458')
469
470
471 def testRevertModified(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000472 if not self.enabled:
473 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000474 options = self.Options()
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000475 file_path = gclient_scm.os.path.join(self.base_path, 'a')
msb@chromium.orge28e4982009-09-25 20:51:45 +0000476 open(file_path, 'a').writelines('touched\n')
477 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
478 relpath=self.relpath)
479 file_list = []
480 scm.revert(options, self.args, file_list)
481 self.assertEquals(file_list, [file_path])
482 file_list = []
483 scm.diff(options, self.args, file_list)
484 self.assertEquals(file_list, [])
msb@chromium.org0f282062009-11-06 20:14:02 +0000485 self.assertEquals(scm.revinfo(options, self.args, None),
msb@chromium.orge28e4982009-09-25 20:51:45 +0000486 '069c602044c5388d2d15c3f875b057c852003458')
487
488 def testRevertNew(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000489 if not self.enabled:
490 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000491 options = self.Options()
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000492 file_path = gclient_scm.os.path.join(self.base_path, 'c')
msb@chromium.orge28e4982009-09-25 20:51:45 +0000493 f = open(file_path, 'w')
494 f.writelines('new\n')
495 f.close()
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000496 Popen(['git', 'add', 'c'], stdout=PIPE,
497 stderr=STDOUT, cwd=self.base_path).communicate()
msb@chromium.orge28e4982009-09-25 20:51:45 +0000498 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
499 relpath=self.relpath)
500 file_list = []
501 scm.revert(options, self.args, file_list)
502 self.assertEquals(file_list, [file_path])
503 file_list = []
504 scm.diff(options, self.args, file_list)
505 self.assertEquals(file_list, [])
msb@chromium.org0f282062009-11-06 20:14:02 +0000506 self.assertEquals(scm.revinfo(options, self.args, None),
msb@chromium.orge28e4982009-09-25 20:51:45 +0000507 '069c602044c5388d2d15c3f875b057c852003458')
508
509 def testStatusNew(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000510 if not self.enabled:
511 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000512 options = self.Options()
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000513 file_path = gclient_scm.os.path.join(self.base_path, 'a')
msb@chromium.orge28e4982009-09-25 20:51:45 +0000514 open(file_path, 'a').writelines('touched\n')
515 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
516 relpath=self.relpath)
517 file_list = []
518 scm.status(options, self.args, file_list)
519 self.assertEquals(file_list, [file_path])
520
521 def testStatus2New(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000522 if not self.enabled:
523 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000524 options = self.Options()
525 expected_file_list = []
526 for f in ['a', 'b']:
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000527 file_path = gclient_scm.os.path.join(self.base_path, f)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000528 open(file_path, 'a').writelines('touched\n')
529 expected_file_list.extend([file_path])
530 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
531 relpath=self.relpath)
532 file_list = []
533 scm.status(options, self.args, file_list)
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000534 expected_file_list = [gclient_scm.os.path.join(self.base_path, x)
535 for x in ['a', 'b']]
msb@chromium.orge28e4982009-09-25 20:51:45 +0000536 self.assertEquals(sorted(file_list), expected_file_list)
537
538 def testUpdateCheckout(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000539 if not self.enabled:
540 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000541 options = self.Options(verbose=True)
542 root_dir = tempfile.mkdtemp()
543 relpath = 'foo'
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000544 base_path = gclient_scm.os.path.join(root_dir, relpath)
545 url = gclient_scm.os.path.join(self.root_dir, self.relpath, '.git')
msb@chromium.orge28e4982009-09-25 20:51:45 +0000546 try:
547 scm = gclient_scm.CreateSCM(url=url, root_dir=root_dir,
548 relpath=relpath)
549 file_list = []
550 scm.update(options, (), file_list)
551 self.assertEquals(len(file_list), 2)
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000552 self.assert_(gclient_scm.os.path.isfile(
553 gclient_scm.os.path.join(base_path, 'a')))
msb@chromium.org0f282062009-11-06 20:14:02 +0000554 self.assertEquals(scm.revinfo(options, (), None),
msb@chromium.orge28e4982009-09-25 20:51:45 +0000555 '069c602044c5388d2d15c3f875b057c852003458')
556 finally:
maruel@chromium.orgba551772010-02-03 18:21:42 +0000557 rmtree(root_dir)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000558
559 def testUpdateUpdate(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000560 if not self.enabled:
561 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000562 options = self.Options()
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000563 expected_file_list = [gclient_scm.os.path.join(self.base_path, x)
564 for x in ['a', 'b']]
msb@chromium.orge28e4982009-09-25 20:51:45 +0000565 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
566 relpath=self.relpath)
567 file_list = []
568 scm.update(options, (), file_list)
msb@chromium.org0f282062009-11-06 20:14:02 +0000569 self.assertEquals(file_list, expected_file_list)
570 self.assertEquals(scm.revinfo(options, (), None),
msb@chromium.orge28e4982009-09-25 20:51:45 +0000571 'a7142dc9f0009350b96a11f372b6ea658592aa95')
572
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000573 def testUpdateUnstagedConflict(self):
574 if not self.enabled:
575 return
576 options = self.Options()
577 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
578 relpath=self.relpath)
579 file_path = gclient_scm.os.path.join(self.base_path, 'b')
580 f = open(file_path, 'w').writelines('conflict\n')
581 exception = (
582 "error: Your local changes to 'b' would be overwritten by merge. "
583 "Aborting.\n"
584 "Please, commit your changes or stash them before you can merge.\n")
585 self.assertRaisesError(exception, scm.update, options, (), [])
586
msb@chromium.org5bde4852009-12-14 16:47:12 +0000587 def testUpdateConflict(self):
588 if not self.enabled:
589 return
590 options = self.Options()
591 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
592 relpath=self.relpath)
593 file_path = gclient_scm.os.path.join(self.base_path, 'b')
594 f = open(file_path, 'w').writelines('conflict\n')
595 scm._Run(['commit', '-am', 'test'])
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000596 self.mox.StubOutWithMock(__builtin__, 'raw_input')
597 __builtin__.raw_input.__call__(mox.StrContains('Cannot fast-forward merge, '
598 'attempt to rebase? (y)es / '
599 '(q)uit / (s)kip : ')
600 ).AndReturn('y')
601 self.mox.ReplayAll()
msb@chromium.org5bde4852009-12-14 16:47:12 +0000602 exception = \
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000603 'Conflict while rebasing this branch.\n' \
msb@chromium.org5bde4852009-12-14 16:47:12 +0000604 'Fix the conflict and run gclient again.\n' \
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000605 "See 'man git-rebase' for details.\n"
msb@chromium.org5bde4852009-12-14 16:47:12 +0000606 self.assertRaisesError(exception, scm.update, options, (), [])
607 exception = \
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000608 '\n____ . at refs/heads/master\n' \
msb@chromium.org5bde4852009-12-14 16:47:12 +0000609 '\tAlready in a conflict, i.e. (no branch).\n' \
610 '\tFix the conflict and run gclient again.\n' \
611 '\tOr to abort run:\n\t\tgit-rebase --abort\n' \
612 '\tSee man git-rebase for details.\n'
613 self.assertRaisesError(exception, scm.update, options, (), [])
614
msb@chromium.orge4af1ab2010-01-13 21:26:09 +0000615 def testUpdateNotGit(self):
616 if not self.enabled:
617 return
618 options = self.Options()
619 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
620 relpath=self.relpath)
621 git_path = gclient_scm.os.path.join(self.base_path, '.git')
maruel@chromium.orgba551772010-02-03 18:21:42 +0000622 rename(git_path, git_path + 'foo')
msb@chromium.orge4af1ab2010-01-13 21:26:09 +0000623 exception = \
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000624 '\n____ . at refs/heads/master\n' \
msb@chromium.orge4af1ab2010-01-13 21:26:09 +0000625 '\tPath is not a git repo. No .git dir.\n' \
626 '\tTo resolve:\n' \
627 '\t\trm -rf .\n' \
628 '\tAnd run gclient sync again\n'
629 self.assertRaisesError(exception, scm.update, options, (), [])
630
msb@chromium.org0f282062009-11-06 20:14:02 +0000631 def testRevinfo(self):
632 if not self.enabled:
633 return
634 options = self.Options()
635 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
636 relpath=self.relpath)
637 rev_info = scm.revinfo(options, (), None)
638 self.assertEquals(rev_info, '069c602044c5388d2d15c3f875b057c852003458')
639
msb@chromium.orge28e4982009-09-25 20:51:45 +0000640
msb@chromium.orge28e4982009-09-25 20:51:45 +0000641if __name__ == '__main__':
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000642 import unittest
msb@chromium.orge28e4982009-09-25 20:51:45 +0000643 unittest.main()
644
645# vim: ts=2:sw=2:tw=80:et: