blob: d7d93b6789e6ff0b72e260584a20d047070346c4 [file] [log] [blame]
maruel@chromium.org45d8db02011-03-31 20:43:56 +00001#!/usr/bin/env python
2# Copyright (c) 2011 The Chromium Authors. All rights reserved.
maruel@chromium.orgba551772010-02-03 18:21:42 +00003# 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.orgbf38a7e2010-12-14 18:15:54 +00008# pylint: disable=E1101,E1103,W0403
9
maruel@chromium.orgba551772010-02-03 18:21:42 +000010# Import before super_mox to keep valid references.
11from os import rename
12from shutil import rmtree
maruel@chromium.org8ef5f542009-11-12 02:05:02 +000013from subprocess import Popen, PIPE, STDOUT
msb@chromium.orge28e4982009-09-25 20:51:45 +000014import tempfile
maruel@chromium.org389d6de2010-09-09 14:14:37 +000015import unittest
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +000016import __builtin__
msb@chromium.orge28e4982009-09-25 20:51:45 +000017
maruel@chromium.orgba551772010-02-03 18:21:42 +000018# Fixes include path.
maruel@chromium.orgdd057f32010-09-20 19:33:31 +000019from super_mox import mox, StdoutCheck, TestCaseUtils, SuperMoxTestBase
maruel@chromium.orgba551772010-02-03 18:21:42 +000020
maruel@chromium.orgcdbecc42011-02-09 04:31:47 +000021import logging
maruel@chromium.orgcb2985f2010-11-03 14:08:31 +000022import sys
msb@chromium.orge28e4982009-09-25 20:51:45 +000023import gclient_scm
maruel@chromium.org96913eb2010-06-01 16:22:47 +000024
maruel@chromium.org795a8c12010-10-05 19:54:29 +000025# Shortcut since this function is used often
26join = gclient_scm.os.path.join
27
maruel@chromium.org96913eb2010-06-01 16:22:47 +000028
maruel@chromium.org389d6de2010-09-09 14:14:37 +000029class GCBaseTestCase(object):
maruel@chromium.org96913eb2010-06-01 16:22:47 +000030 def assertRaisesError(self, msg, fn, *args, **kwargs):
maruel@chromium.org389d6de2010-09-09 14:14:37 +000031 """Like unittest's assertRaises() but checks for Gclient.Error."""
maruel@chromium.org96913eb2010-06-01 16:22:47 +000032 try:
33 fn(*args, **kwargs)
34 except gclient_scm.gclient_utils.Error, e:
35 self.assertEquals(e.args[0], msg)
36 else:
37 self.fail('%s not raised' % msg)
msb@chromium.orge28e4982009-09-25 20:51:45 +000038
maruel@chromium.org8ef5f542009-11-12 02:05:02 +000039
maruel@chromium.org389d6de2010-09-09 14:14:37 +000040class BaseTestCase(GCBaseTestCase, SuperMoxTestBase):
41 def setUp(self):
maruel@chromium.org389d6de2010-09-09 14:14:37 +000042 SuperMoxTestBase.setUp(self)
43 self.mox.StubOutWithMock(gclient_scm.gclient_utils, 'CheckCall')
44 self.mox.StubOutWithMock(gclient_scm.gclient_utils, 'CheckCallAndFilter')
45 self.mox.StubOutWithMock(gclient_scm.gclient_utils,
46 'CheckCallAndFilterAndHeader')
47 self.mox.StubOutWithMock(gclient_scm.gclient_utils, 'FileRead')
48 self.mox.StubOutWithMock(gclient_scm.gclient_utils, 'FileWrite')
49 self.mox.StubOutWithMock(gclient_scm.gclient_utils, 'Popen')
50 self.mox.StubOutWithMock(gclient_scm.gclient_utils, 'RemoveDirectory')
51 self._CaptureSVNInfo = gclient_scm.scm.SVN.CaptureInfo
52 self.mox.StubOutWithMock(gclient_scm.scm.SVN, 'Capture')
53 self.mox.StubOutWithMock(gclient_scm.scm.SVN, 'CaptureInfo')
54 self.mox.StubOutWithMock(gclient_scm.scm.SVN, 'CaptureStatus')
55 self.mox.StubOutWithMock(gclient_scm.scm.SVN, 'RunAndGetFileList')
56 self._scm_wrapper = gclient_scm.CreateSCM
maruel@chromium.org389d6de2010-09-09 14:14:37 +000057 gclient_scm.scm.SVN.current_version = None
maruel@chromium.org795a8c12010-10-05 19:54:29 +000058 # Absolute path of the fake checkout directory.
59 self.base_path = join(self.root_dir, self.relpath)
maruel@chromium.org389d6de2010-09-09 14:14:37 +000060
61 def tearDown(self):
maruel@chromium.org389d6de2010-09-09 14:14:37 +000062 SuperMoxTestBase.tearDown(self)
63
64
maruel@chromium.org8ef5f542009-11-12 02:05:02 +000065class SVNWrapperTestCase(BaseTestCase):
msb@chromium.orge28e4982009-09-25 20:51:45 +000066 class OptionsObject(object):
maruel@chromium.orgbf38a7e2010-12-14 18:15:54 +000067 def __init__(self, verbose=False, revision=None):
msb@chromium.orge28e4982009-09-25 20:51:45 +000068 self.verbose = verbose
69 self.revision = revision
70 self.manually_grab_svn_rev = True
71 self.deps_os = None
72 self.force = False
davemoore@chromium.org8bf27312010-02-19 17:29:44 +000073 self.reset = False
msb@chromium.orge28e4982009-09-25 20:51:45 +000074 self.nohooks = False
75
maruel@chromium.org8ef5f542009-11-12 02:05:02 +000076 def Options(self, *args, **kwargs):
maruel@chromium.org8071c282010-09-20 19:44:19 +000077 return self.OptionsObject(*args, **kwargs)
maruel@chromium.org8ef5f542009-11-12 02:05:02 +000078
msb@chromium.orge28e4982009-09-25 20:51:45 +000079 def setUp(self):
maruel@chromium.org8ef5f542009-11-12 02:05:02 +000080 BaseTestCase.setUp(self)
maruel@chromium.org389d6de2010-09-09 14:14:37 +000081 self.url = self.SvnUrl()
msb@chromium.orge28e4982009-09-25 20:51:45 +000082
83 def testDir(self):
84 members = [
maruel@chromium.org2b9aa8e2010-08-25 20:01:42 +000085 'FullUrlForRelativeUrl', 'RunCommand',
maruel@chromium.org55e724e2010-03-11 19:36:49 +000086 'cleanup', 'diff', 'export', 'pack', 'relpath', 'revert',
maruel@chromium.org9eda4112010-06-11 18:56:10 +000087 'revinfo', 'runhooks', 'status', 'update',
tony@chromium.org4b5b1772010-04-08 01:52:56 +000088 'updatesingle', 'url',
msb@chromium.orge28e4982009-09-25 20:51:45 +000089 ]
90
91 # If you add a member, be sure to add the relevant test!
maruel@chromium.org9eda4112010-06-11 18:56:10 +000092 self.compareMembers(self._scm_wrapper('svn://a'), members)
msb@chromium.orge28e4982009-09-25 20:51:45 +000093
94 def testUnsupportedSCM(self):
maruel@chromium.org9eda4112010-06-11 18:56:10 +000095 args = ['gopher://foo', self.root_dir, self.relpath]
96 exception_msg = 'No SCM found for url gopher://foo'
97 self.assertRaisesError(exception_msg, self._scm_wrapper, *args)
msb@chromium.orge28e4982009-09-25 20:51:45 +000098
msb@chromium.orge6f78352010-01-13 17:05:33 +000099 def testSVNFullUrlForRelativeUrl(self):
100 self.url = 'svn://a/b/c/d'
101
102 self.mox.ReplayAll()
103 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
104 relpath=self.relpath)
105 self.assertEqual(scm.FullUrlForRelativeUrl('/crap'), 'svn://a/b/crap')
106
107 def testGITFullUrlForRelativeUrl(self):
108 self.url = 'git://a/b/c/d'
109
110 self.mox.ReplayAll()
111 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
112 relpath=self.relpath)
113 self.assertEqual(scm.FullUrlForRelativeUrl('/crap'), 'git://a/b/c/crap')
114
msb@chromium.orge28e4982009-09-25 20:51:45 +0000115 def testRunCommandException(self):
116 options = self.Options(verbose=False)
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000117 gclient_scm.os.path.exists(join(self.base_path, '.git')).AndReturn(False)
118 gclient_scm.os.path.exists(join(self.base_path, '.hg')).AndReturn(False)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000119
120 self.mox.ReplayAll()
121 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
122 relpath=self.relpath)
123 exception = "Unsupported argument(s): %s" % ','.join(self.args)
124 self.assertRaisesError(exception, scm.RunCommand,
125 'update', options, self.args)
126
127 def testRunCommandUnknown(self):
128 # TODO(maruel): if ever used.
maruel@chromium.orgdd057f32010-09-20 19:33:31 +0000129 pass
msb@chromium.orge28e4982009-09-25 20:51:45 +0000130
131 def testRevertMissing(self):
132 options = self.Options(verbose=True)
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000133 gclient_scm.os.path.isdir(self.base_path).AndReturn(False)
maruel@chromium.org2b9aa8e2010-08-25 20:01:42 +0000134 gclient_scm.scm.SVN.Capture(['--version']
135 ).AndReturn('svn, version 1.5.1 (r32289)')
msb@chromium.orge28e4982009-09-25 20:51:45 +0000136 # It'll to a checkout instead.
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000137 gclient_scm.os.path.exists(join(self.base_path, '.git')).AndReturn(False)
138 gclient_scm.os.path.exists(join(self.base_path, '.hg')).AndReturn(False)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000139 # Checkout.
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000140 gclient_scm.os.path.exists(self.base_path).AndReturn(False)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000141 files_list = self.mox.CreateMockAnything()
maruel@chromium.orgce117f62011-01-17 20:04:25 +0000142 gclient_scm.scm.SVN.RunAndGetFileList(
143 options.verbose,
144 ['checkout', self.url, self.base_path, '--force', '--ignore-externals'],
145 cwd=self.root_dir,
146 file_list=files_list)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000147
148 self.mox.ReplayAll()
149 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
150 relpath=self.relpath)
151 scm.revert(options, self.args, files_list)
maruel@chromium.org54019f32010-09-09 13:50:11 +0000152 self.checkstdout(
153 ('\n_____ %s is missing, synching instead\n' % self.relpath))
msb@chromium.orge28e4982009-09-25 20:51:45 +0000154
155 def testRevertNone(self):
156 options = self.Options(verbose=True)
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000157 gclient_scm.os.path.isdir(self.base_path).AndReturn(True)
158 gclient_scm.scm.SVN.CaptureStatus(self.base_path).AndReturn([])
maruel@chromium.orgce117f62011-01-17 20:04:25 +0000159 gclient_scm.scm.SVN.RunAndGetFileList(
160 options.verbose,
161 ['update', '--revision', 'BASE', '--ignore-externals'],
162 cwd=self.base_path,
163 file_list=mox.IgnoreArg())
msb@chromium.orge28e4982009-09-25 20:51:45 +0000164
165 self.mox.ReplayAll()
166 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
167 relpath=self.relpath)
168 file_list = []
169 scm.revert(options, self.args, file_list)
170
msb@chromium.orge28e4982009-09-25 20:51:45 +0000171 def testRevertDirectory(self):
172 options = self.Options(verbose=True)
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000173 gclient_scm.os.path.isdir(self.base_path).AndReturn(True)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000174 items = [
175 ('~ ', 'a'),
176 ]
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000177 gclient_scm.scm.SVN.CaptureStatus(self.base_path).AndReturn(items)
178 file_path = join(self.base_path, 'a')
msb@chromium.orge28e4982009-09-25 20:51:45 +0000179 gclient_scm.os.path.exists(file_path).AndReturn(True)
180 gclient_scm.os.path.isfile(file_path).AndReturn(False)
msb@chromium.org32906d12010-01-12 18:22:09 +0000181 gclient_scm.os.path.islink(file_path).AndReturn(False)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000182 gclient_scm.os.path.isdir(file_path).AndReturn(True)
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000183 gclient_scm.gclient_utils.RemoveDirectory(file_path)
maruel@chromium.orgce117f62011-01-17 20:04:25 +0000184 gclient_scm.scm.SVN.RunAndGetFileList(
185 options.verbose,
186 ['update', '--revision', 'BASE', '--ignore-externals'],
187 cwd=self.base_path,
188 file_list=mox.IgnoreArg())
msb@chromium.orge28e4982009-09-25 20:51:45 +0000189
190 self.mox.ReplayAll()
191 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
192 relpath=self.relpath)
193 file_list2 = []
194 scm.revert(options, self.args, file_list2)
maruel@chromium.org54019f32010-09-09 13:50:11 +0000195 self.checkstdout(('%s\n' % file_path))
msb@chromium.orge28e4982009-09-25 20:51:45 +0000196
197 def testStatus(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)
maruel@chromium.org77e4eca2010-09-21 13:23:07 +0000200 gclient_scm.scm.SVN.RunAndGetFileList(
maruel@chromium.orgce117f62011-01-17 20:04:25 +0000201 options.verbose,
202 ['status'] + self.args + ['--ignore-externals'],
203 cwd=self.base_path,
204 file_list=[]).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)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000211
212 # TODO(maruel): TEST REVISIONS!!!
213 # TODO(maruel): TEST RELOCATE!!!
214 def testUpdateCheckout(self):
215 options = self.Options(verbose=True)
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000216 file_info = gclient_scm.gclient_utils.PrintableObject()
msb@chromium.orge28e4982009-09-25 20:51:45 +0000217 file_info.root = 'blah'
218 file_info.url = self.url
219 file_info.uuid = 'ABC'
220 file_info.revision = 42
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000221 gclient_scm.os.path.exists(join(self.base_path, '.git')).AndReturn(False)
222 gclient_scm.os.path.exists(join(self.base_path, '.hg')).AndReturn(False)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000223 # Checkout.
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000224 gclient_scm.os.path.exists(self.base_path).AndReturn(False)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000225 files_list = self.mox.CreateMockAnything()
maruel@chromium.org669600d2010-09-01 19:06:31 +0000226 gclient_scm.scm.SVN.Capture(['--version']
227 ).AndReturn('svn, version 1.5.1 (r32289)')
maruel@chromium.orgce117f62011-01-17 20:04:25 +0000228 gclient_scm.scm.SVN.RunAndGetFileList(
229 options.verbose,
230 ['checkout', self.url, self.base_path, '--force', '--ignore-externals'],
231 cwd=self.root_dir,
232 file_list=files_list)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000233 self.mox.ReplayAll()
234 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
235 relpath=self.relpath)
236 scm.update(options, (), files_list)
237
238 def testUpdateUpdate(self):
239 options = self.Options(verbose=True)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000240 options.force = True
241 options.nohooks = False
242 file_info = {
243 'Repository Root': 'blah',
244 'URL': self.url,
245 'UUID': 'ABC',
246 'Revision': 42,
247 }
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000248 gclient_scm.os.path.exists(join(self.base_path, '.git')).AndReturn(False)
249 gclient_scm.os.path.exists(join(self.base_path, '.hg')).AndReturn(False)
maruel@chromium.orge407c9a2010-08-09 19:11:37 +0000250
251 # Verify no locked files.
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000252 dotted_path = join(self.base_path, '.')
253 gclient_scm.scm.SVN.CaptureStatus(dotted_path).AndReturn([])
maruel@chromium.orge407c9a2010-08-09 19:11:37 +0000254
msb@chromium.orge28e4982009-09-25 20:51:45 +0000255 # Checkout or update.
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000256 gclient_scm.os.path.exists(self.base_path).AndReturn(True)
257 gclient_scm.scm.SVN.CaptureInfo(dotted_path).AndReturn(file_info)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000258 # Cheat a bit here.
maruel@chromium.org54019f32010-09-09 13:50:11 +0000259 gclient_scm.scm.SVN.CaptureInfo(file_info['URL']).AndReturn(file_info)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000260 additional_args = []
261 if options.manually_grab_svn_rev:
262 additional_args = ['--revision', str(file_info['Revision'])]
maruel@chromium.org669600d2010-09-01 19:06:31 +0000263 gclient_scm.scm.SVN.Capture(['--version']
264 ).AndReturn('svn, version 1.5.1 (r32289)')
maruel@chromium.orgce117f62011-01-17 20:04:25 +0000265 additional_args.extend(['--force', '--ignore-externals'])
msb@chromium.orge28e4982009-09-25 20:51:45 +0000266 files_list = []
maruel@chromium.org5aeb7dd2009-11-17 18:09:01 +0000267 gclient_scm.scm.SVN.RunAndGetFileList(
maruel@chromium.org03807072010-08-16 17:18:44 +0000268 options.verbose,
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000269 ['update', self.base_path] + additional_args,
maruel@chromium.org77e4eca2010-09-21 13:23:07 +0000270 cwd=self.root_dir, file_list=files_list)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000271
272 self.mox.ReplayAll()
273 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
274 relpath=self.relpath)
275 scm.update(options, (), files_list)
276
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000277 def testUpdateSingleCheckout(self):
278 options = self.Options(verbose=True)
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000279 file_info = {
280 'URL': self.url,
281 'Revision': 42,
282 }
tony@chromium.org57564662010-04-14 02:35:12 +0000283
284 # Checks to make sure that we support svn co --depth.
285 gclient_scm.scm.SVN.current_version = None
286 gclient_scm.scm.SVN.Capture(['--version']
287 ).AndReturn('svn, version 1.5.1 (r32289)')
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000288 gclient_scm.os.path.exists(join(self.base_path, '.svn')).AndReturn(False)
289 gclient_scm.os.path.exists(join(self.base_path, 'DEPS')).AndReturn(False)
tony@chromium.org57564662010-04-14 02:35:12 +0000290
maruel@chromium.orge407c9a2010-08-09 19:11:37 +0000291 # Verify no locked files.
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000292 dotted_path = join(self.base_path, '.')
293 gclient_scm.scm.SVN.CaptureStatus(dotted_path).AndReturn([])
maruel@chromium.orge407c9a2010-08-09 19:11:37 +0000294
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000295 # When checking out a single file, we issue an svn checkout and svn update.
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000296 files_list = self.mox.CreateMockAnything()
maruel@chromium.org669600d2010-09-01 19:06:31 +0000297 gclient_scm.gclient_utils.CheckCallAndFilterAndHeader(
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000298 ['svn', 'checkout', '--depth', 'empty', self.url, self.base_path],
maruel@chromium.orgce117f62011-01-17 20:04:25 +0000299 always=True,
300 cwd=self.root_dir)
301 gclient_scm.scm.SVN.RunAndGetFileList(
302 options.verbose,
303 ['update', 'DEPS', '--ignore-externals'],
304 cwd=self.base_path,
305 file_list=files_list)
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000306
307 # Now we fall back on scm.update().
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000308 gclient_scm.os.path.exists(join(self.base_path, '.git')).AndReturn(False)
309 gclient_scm.os.path.exists(join(self.base_path, '.hg')).AndReturn(False)
310 gclient_scm.os.path.exists(self.base_path).AndReturn(True)
311 gclient_scm.scm.SVN.CaptureInfo(dotted_path).AndReturn(file_info)
maruel@chromium.org54019f32010-09-09 13:50:11 +0000312 gclient_scm.scm.SVN.CaptureInfo(file_info['URL']).AndReturn(file_info)
tony@chromium.org57564662010-04-14 02:35:12 +0000313
314 self.mox.ReplayAll()
315 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
316 relpath=self.relpath)
317 scm.updatesingle(options, ['DEPS'], files_list)
maruel@chromium.orgdd057f32010-09-20 19:33:31 +0000318 self.checkstdout('\n_____ %s at 42\n' % self.relpath)
tony@chromium.org57564662010-04-14 02:35:12 +0000319
320 def testUpdateSingleCheckoutSVN14(self):
321 options = self.Options(verbose=True)
tony@chromium.org57564662010-04-14 02:35:12 +0000322
323 # Checks to make sure that we support svn co --depth.
324 gclient_scm.scm.SVN.current_version = None
325 gclient_scm.scm.SVN.Capture(['--version']
326 ).AndReturn('svn, version 1.4.4 (r25188)')
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000327 gclient_scm.os.path.exists(self.base_path).AndReturn(True)
tony@chromium.org57564662010-04-14 02:35:12 +0000328
329 # When checking out a single file with svn 1.4, we use svn export
330 files_list = self.mox.CreateMockAnything()
maruel@chromium.org669600d2010-09-01 19:06:31 +0000331 gclient_scm.gclient_utils.CheckCallAndFilterAndHeader(
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000332 ['svn', 'export', join(self.url, 'DEPS'), join(self.base_path, 'DEPS')],
maruel@chromium.org77e4eca2010-09-21 13:23:07 +0000333 always=True, cwd=self.root_dir)
tony@chromium.org57564662010-04-14 02:35:12 +0000334
335 self.mox.ReplayAll()
336 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
337 relpath=self.relpath)
338 scm.updatesingle(options, ['DEPS'], files_list)
339
340 def testUpdateSingleCheckoutSVNUpgrade(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.5.1 (r32289)')
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000351 gclient_scm.os.path.exists(join(self.base_path, '.svn')).AndReturn(False)
tony@chromium.org57564662010-04-14 02:35:12 +0000352 # If DEPS already exists, assume we're upgrading from svn1.4, so delete
353 # the old DEPS file.
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000354 gclient_scm.os.path.exists(join(self.base_path, 'DEPS')).AndReturn(True)
355 gclient_scm.os.remove(join(self.base_path, 'DEPS'))
tony@chromium.org57564662010-04-14 02:35:12 +0000356
maruel@chromium.orge407c9a2010-08-09 19:11:37 +0000357 # Verify no locked files.
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000358 gclient_scm.scm.SVN.CaptureStatus(join(self.base_path, '.')).AndReturn([])
maruel@chromium.orge407c9a2010-08-09 19:11:37 +0000359
tony@chromium.org57564662010-04-14 02:35:12 +0000360 # When checking out a single file, we issue an svn checkout and svn update.
361 files_list = self.mox.CreateMockAnything()
maruel@chromium.org669600d2010-09-01 19:06:31 +0000362 gclient_scm.gclient_utils.CheckCallAndFilterAndHeader(
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000363 ['svn', 'checkout', '--depth', 'empty', self.url, self.base_path],
maruel@chromium.orgce117f62011-01-17 20:04:25 +0000364 always=True,
365 cwd=self.root_dir)
366 gclient_scm.scm.SVN.RunAndGetFileList(
367 options.verbose,
368 ['update', 'DEPS', '--ignore-externals'],
369 cwd=self.base_path,
370 file_list=files_list)
tony@chromium.org57564662010-04-14 02:35:12 +0000371
372 # Now we fall back on scm.update().
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000373 gclient_scm.os.path.exists(join(self.base_path, '.git')).AndReturn(False)
374 gclient_scm.os.path.exists(join(self.base_path, '.hg')).AndReturn(False)
375 gclient_scm.os.path.exists(self.base_path).AndReturn(True)
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000376 gclient_scm.scm.SVN.CaptureInfo(
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000377 join(self.base_path, ".")).AndReturn(file_info)
maruel@chromium.org54019f32010-09-09 13:50:11 +0000378 gclient_scm.scm.SVN.CaptureInfo(file_info['URL']).AndReturn(file_info)
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000379
380 self.mox.ReplayAll()
381 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
382 relpath=self.relpath)
383 scm.updatesingle(options, ['DEPS'], files_list)
maruel@chromium.org54019f32010-09-09 13:50:11 +0000384 self.checkstdout(
385 ('\n_____ %s at 42\n' % self.relpath))
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000386
387 def testUpdateSingleUpdate(self):
388 options = self.Options(verbose=True)
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000389 file_info = {
390 'URL': self.url,
391 'Revision': 42,
392 }
tony@chromium.org57564662010-04-14 02:35:12 +0000393 # Checks to make sure that we support svn co --depth.
394 gclient_scm.scm.SVN.current_version = None
395 gclient_scm.scm.SVN.Capture(['--version']
396 ).AndReturn('svn, version 1.5.1 (r32289)')
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000397 gclient_scm.os.path.exists(join(self.base_path, '.svn')).AndReturn(True)
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000398
maruel@chromium.orge407c9a2010-08-09 19:11:37 +0000399 # Verify no locked files.
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000400 gclient_scm.scm.SVN.CaptureStatus(join(self.base_path, '.')).AndReturn([])
maruel@chromium.orge407c9a2010-08-09 19:11:37 +0000401
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000402 # Now we fall back on scm.update().
403 files_list = self.mox.CreateMockAnything()
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000404 gclient_scm.os.path.exists(join(self.base_path, '.git')).AndReturn(False)
405 gclient_scm.os.path.exists(join(self.base_path, '.hg')).AndReturn(False)
406 gclient_scm.os.path.exists(self.base_path).AndReturn(True)
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000407 gclient_scm.scm.SVN.CaptureInfo(
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000408 join(self.base_path, '.')).AndReturn(file_info)
maruel@chromium.org54019f32010-09-09 13:50:11 +0000409 gclient_scm.scm.SVN.CaptureInfo(file_info['URL']).AndReturn(file_info)
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000410
411 self.mox.ReplayAll()
412 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
413 relpath=self.relpath)
414 scm.updatesingle(options, ['DEPS'], files_list)
maruel@chromium.org54019f32010-09-09 13:50:11 +0000415 self.checkstdout('\n_____ %s at 42\n' % self.relpath)
tony@chromium.org4b5b1772010-04-08 01:52:56 +0000416
msb@chromium.orge28e4982009-09-25 20:51:45 +0000417 def testUpdateGit(self):
418 options = self.Options(verbose=True)
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000419 file_path = gclient_scm.os.path.join(self.root_dir, self.relpath, '.git')
420 gclient_scm.os.path.exists(file_path).AndReturn(True)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000421
422 self.mox.ReplayAll()
423 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
424 relpath=self.relpath)
425 file_list = []
426 scm.update(options, self.args, file_list)
maruel@chromium.org54019f32010-09-09 13:50:11 +0000427 self.checkstdout(
428 ('________ found .git directory; skipping %s\n' % self.relpath))
msb@chromium.orge28e4982009-09-25 20:51:45 +0000429
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000430 def testUpdateHg(self):
431 options = self.Options(verbose=True)
432 gclient_scm.os.path.exists(join(self.base_path, '.git')).AndReturn(False)
433 gclient_scm.os.path.exists(join(self.base_path, '.hg')).AndReturn(True)
434
435 self.mox.ReplayAll()
436 scm = self._scm_wrapper(url=self.url, root_dir=self.root_dir,
437 relpath=self.relpath)
438 file_list = []
439 scm.update(options, self.args, file_list)
440 self.checkstdout(
441 ('________ found .hg directory; skipping %s\n' % self.relpath))
442
msb@chromium.orge28e4982009-09-25 20:51:45 +0000443
maruel@chromium.orgdd057f32010-09-20 19:33:31 +0000444class GitWrapperTestCase(GCBaseTestCase, StdoutCheck, TestCaseUtils,
445 unittest.TestCase):
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000446 """This class doesn't use pymox."""
msb@chromium.orge28e4982009-09-25 20:51:45 +0000447 class OptionsObject(object):
maruel@chromium.orgbf38a7e2010-12-14 18:15:54 +0000448 def __init__(self, verbose=False, revision=None):
msb@chromium.orge28e4982009-09-25 20:51:45 +0000449 self.verbose = verbose
450 self.revision = revision
451 self.manually_grab_svn_rev = True
452 self.deps_os = None
453 self.force = False
davemoore@chromium.org8bf27312010-02-19 17:29:44 +0000454 self.reset = False
msb@chromium.orge28e4982009-09-25 20:51:45 +0000455 self.nohooks = False
456
457 sample_git_import = """blob
458mark :1
459data 6
460Hello
461
462blob
463mark :2
464data 4
465Bye
466
467reset refs/heads/master
468commit refs/heads/master
469mark :3
470author Bob <bob@example.com> 1253744361 -0700
471committer Bob <bob@example.com> 1253744361 -0700
472data 8
473A and B
474M 100644 :1 a
475M 100644 :2 b
476
477blob
478mark :4
479data 10
480Hello
481You
482
483blob
484mark :5
485data 8
486Bye
487You
488
489commit refs/heads/origin
490mark :6
491author Alice <alice@example.com> 1253744424 -0700
492committer Alice <alice@example.com> 1253744424 -0700
493data 13
494Personalized
495from :3
496M 100644 :4 a
497M 100644 :5 b
498
499reset refs/heads/master
500from :3
501"""
msb@chromium.orge28e4982009-09-25 20:51:45 +0000502 def Options(self, *args, **kwargs):
maruel@chromium.org8071c282010-09-20 19:44:19 +0000503 return self.OptionsObject(*args, **kwargs)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000504
maruel@chromium.orgbf38a7e2010-12-14 18:15:54 +0000505 @staticmethod
506 def CreateGitRepo(git_import, path):
maruel@chromium.orgd5800f12009-11-12 20:03:43 +0000507 """Do it for real."""
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000508 try:
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000509 Popen(['git', 'init', '-q'], stdout=PIPE, stderr=STDOUT,
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000510 cwd=path).communicate()
511 except OSError:
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000512 # git is not available, skip this test.
513 return False
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000514 Popen(['git', 'fast-import', '--quiet'], stdin=PIPE, stdout=PIPE,
515 stderr=STDOUT, cwd=path).communicate(input=git_import)
516 Popen(['git', 'checkout', '-q'], stdout=PIPE, stderr=STDOUT,
517 cwd=path).communicate()
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000518 Popen(['git', 'remote', 'add', '-f', 'origin', '.'], stdout=PIPE,
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000519 stderr=STDOUT, cwd=path).communicate()
520 Popen(['git', 'checkout', '-b', 'new', 'origin/master', '-q'], stdout=PIPE,
521 stderr=STDOUT, cwd=path).communicate()
522 Popen(['git', 'push', 'origin', 'origin/origin:origin/master', '-q'],
523 stdout=PIPE, stderr=STDOUT, cwd=path).communicate()
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000524 Popen(['git', 'config', '--unset', 'remote.origin.fetch'], stdout=PIPE,
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000525 stderr=STDOUT, cwd=path).communicate()
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000526 return True
msb@chromium.orge28e4982009-09-25 20:51:45 +0000527
msb@chromium.orge28e4982009-09-25 20:51:45 +0000528 def setUp(self):
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000529 TestCaseUtils.setUp(self)
530 unittest.TestCase.setUp(self)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000531 self.url = 'git://foo'
532 self.root_dir = tempfile.mkdtemp()
533 self.relpath = '.'
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000534 self.base_path = join(self.root_dir, self.relpath)
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000535 self.enabled = self.CreateGitRepo(self.sample_git_import, self.base_path)
maruel@chromium.orgdd057f32010-09-20 19:33:31 +0000536 StdoutCheck.setUp(self)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000537
538 def tearDown(self):
maruel@chromium.orgdd057f32010-09-20 19:33:31 +0000539 StdoutCheck.tearDown(self)
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000540 TestCaseUtils.tearDown(self)
541 unittest.TestCase.tearDown(self)
maruel@chromium.orgba551772010-02-03 18:21:42 +0000542 rmtree(self.root_dir)
msb@chromium.orge28e4982009-09-25 20:51:45 +0000543
544 def testDir(self):
545 members = [
maruel@chromium.org55e724e2010-03-11 19:36:49 +0000546 'FullUrlForRelativeUrl', 'RunCommand',
msb@chromium.orge6f78352010-01-13 17:05:33 +0000547 'cleanup', 'diff', 'export', 'pack', 'relpath', 'revert',
maruel@chromium.org9eda4112010-06-11 18:56:10 +0000548 'revinfo', 'runhooks', 'status', 'update', 'url',
msb@chromium.orge28e4982009-09-25 20:51:45 +0000549 ]
550
551 # If you add a member, be sure to add the relevant test!
552 self.compareMembers(gclient_scm.CreateSCM(url=self.url), members)
553
554 def testRevertMissing(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000555 if not self.enabled:
556 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000557 options = self.Options()
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000558 file_path = join(self.base_path, 'a')
msb@chromium.orge28e4982009-09-25 20:51:45 +0000559 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
560 relpath=self.relpath)
561 file_list = []
nasser@codeaurora.orgb2b46312010-04-30 20:58:03 +0000562 scm.update(options, None, file_list)
563 gclient_scm.os.remove(file_path)
564 file_list = []
msb@chromium.orge28e4982009-09-25 20:51:45 +0000565 scm.revert(options, self.args, file_list)
566 self.assertEquals(file_list, [file_path])
567 file_list = []
568 scm.diff(options, self.args, file_list)
569 self.assertEquals(file_list, [])
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000570 self.checkstdout(
maruel@chromium.orgad80e3b2010-09-09 14:18:28 +0000571 ('\n_____ . at refs/heads/master\nUpdating 069c602..a7142dc\n'
572 'Fast-forward\n a | 1 +\n b | 1 +\n'
573 ' 2 files changed, 2 insertions(+), 0 deletions(-)\n\n\n'
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000574 '________ running \'git reset --hard origin/master\' in \'%s\'\n'
575 'HEAD is now at a7142dc Personalized\n') %
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000576 join(self.root_dir, '.'))
msb@chromium.orge28e4982009-09-25 20:51:45 +0000577
578 def testRevertNone(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000579 if not self.enabled:
580 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000581 options = self.Options()
582 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
583 relpath=self.relpath)
584 file_list = []
nasser@codeaurora.orgb2b46312010-04-30 20:58:03 +0000585 scm.update(options, None, file_list)
586 file_list = []
msb@chromium.orge28e4982009-09-25 20:51:45 +0000587 scm.revert(options, self.args, file_list)
588 self.assertEquals(file_list, [])
msb@chromium.org0f282062009-11-06 20:14:02 +0000589 self.assertEquals(scm.revinfo(options, self.args, None),
nasser@codeaurora.orgb2b46312010-04-30 20:58:03 +0000590 'a7142dc9f0009350b96a11f372b6ea658592aa95')
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000591 self.checkstdout(
maruel@chromium.orgad80e3b2010-09-09 14:18:28 +0000592 ('\n_____ . at refs/heads/master\nUpdating 069c602..a7142dc\n'
593 'Fast-forward\n a | 1 +\n b | 1 +\n'
594 ' 2 files changed, 2 insertions(+), 0 deletions(-)\n\n\n'
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000595 '________ running \'git reset --hard origin/master\' in \'%s\'\n'
596 'HEAD is now at a7142dc Personalized\n') %
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000597 join(self.root_dir, '.'))
msb@chromium.orge28e4982009-09-25 20:51:45 +0000598
599 def testRevertModified(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000600 if not self.enabled:
601 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000602 options = self.Options()
msb@chromium.orge28e4982009-09-25 20:51:45 +0000603 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
604 relpath=self.relpath)
605 file_list = []
nasser@codeaurora.orgb2b46312010-04-30 20:58:03 +0000606 scm.update(options, None, file_list)
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000607 file_path = join(self.base_path, 'a')
nasser@codeaurora.orgb2b46312010-04-30 20:58:03 +0000608 open(file_path, 'a').writelines('touched\n')
609 file_list = []
msb@chromium.orge28e4982009-09-25 20:51:45 +0000610 scm.revert(options, self.args, file_list)
611 self.assertEquals(file_list, [file_path])
612 file_list = []
613 scm.diff(options, self.args, file_list)
614 self.assertEquals(file_list, [])
msb@chromium.org0f282062009-11-06 20:14:02 +0000615 self.assertEquals(scm.revinfo(options, self.args, None),
nasser@codeaurora.orgb2b46312010-04-30 20:58:03 +0000616 'a7142dc9f0009350b96a11f372b6ea658592aa95')
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000617 self.checkstdout(
maruel@chromium.orgad80e3b2010-09-09 14:18:28 +0000618 ('\n_____ . at refs/heads/master\nUpdating 069c602..a7142dc\n'
619 'Fast-forward\n a | 1 +\n b | 1 +\n'
620 ' 2 files changed, 2 insertions(+), 0 deletions(-)\n\n\n'
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000621 '________ running \'git reset --hard origin/master\' in \'%s\'\n'
622 'HEAD is now at a7142dc Personalized\n') %
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000623 join(self.root_dir, '.'))
msb@chromium.orge28e4982009-09-25 20:51:45 +0000624
625 def testRevertNew(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000626 if not self.enabled:
627 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000628 options = self.Options()
nasser@codeaurora.orgb2b46312010-04-30 20:58:03 +0000629 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
630 relpath=self.relpath)
631 file_list = []
632 scm.update(options, None, file_list)
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000633 file_path = join(self.base_path, 'c')
msb@chromium.orge28e4982009-09-25 20:51:45 +0000634 f = open(file_path, 'w')
635 f.writelines('new\n')
636 f.close()
maruel@chromium.org8ef5f542009-11-12 02:05:02 +0000637 Popen(['git', 'add', 'c'], stdout=PIPE,
638 stderr=STDOUT, cwd=self.base_path).communicate()
msb@chromium.orge28e4982009-09-25 20:51:45 +0000639 file_list = []
640 scm.revert(options, self.args, file_list)
641 self.assertEquals(file_list, [file_path])
642 file_list = []
643 scm.diff(options, self.args, file_list)
644 self.assertEquals(file_list, [])
msb@chromium.org0f282062009-11-06 20:14:02 +0000645 self.assertEquals(scm.revinfo(options, self.args, None),
nasser@codeaurora.orgb2b46312010-04-30 20:58:03 +0000646 'a7142dc9f0009350b96a11f372b6ea658592aa95')
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000647 self.checkstdout(
maruel@chromium.orgad80e3b2010-09-09 14:18:28 +0000648 ('\n_____ . at refs/heads/master\nUpdating 069c602..a7142dc\n'
649 'Fast-forward\n a | 1 +\n b | 1 +\n'
650 ' 2 files changed, 2 insertions(+), 0 deletions(-)\n\n\n'
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000651 '________ running \'git reset --hard origin/master\' in \'%s\'\n'
652 'HEAD is now at a7142dc Personalized\n') %
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000653 join(self.root_dir, '.'))
msb@chromium.orge28e4982009-09-25 20:51:45 +0000654
655 def testStatusNew(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000656 if not self.enabled:
657 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000658 options = self.Options()
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000659 file_path = join(self.base_path, 'a')
msb@chromium.orge28e4982009-09-25 20:51:45 +0000660 open(file_path, 'a').writelines('touched\n')
661 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
662 relpath=self.relpath)
663 file_list = []
664 scm.status(options, self.args, file_list)
665 self.assertEquals(file_list, [file_path])
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000666 self.checkstdout(
667 ('\n________ running \'git diff --name-status '
668 '069c602044c5388d2d15c3f875b057c852003458\' in \'%s\'\nM\ta\n') %
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000669 join(self.root_dir, '.'))
msb@chromium.orge28e4982009-09-25 20:51:45 +0000670
671 def testStatus2New(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000672 if not self.enabled:
673 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000674 options = self.Options()
675 expected_file_list = []
676 for f in ['a', 'b']:
maruel@chromium.orgbf38a7e2010-12-14 18:15:54 +0000677 file_path = join(self.base_path, f)
678 open(file_path, 'a').writelines('touched\n')
679 expected_file_list.extend([file_path])
msb@chromium.orge28e4982009-09-25 20:51:45 +0000680 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)
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000684 expected_file_list = [join(self.base_path, x) for x in ['a', 'b']]
msb@chromium.orge28e4982009-09-25 20:51:45 +0000685 self.assertEquals(sorted(file_list), expected_file_list)
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000686 self.checkstdout(
687 ('\n________ running \'git diff --name-status '
688 '069c602044c5388d2d15c3f875b057c852003458\' in \'%s\'\nM\ta\nM\tb\n') %
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000689 join(self.root_dir, '.'))
msb@chromium.orge28e4982009-09-25 20:51:45 +0000690
691 def testUpdateCheckout(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000692 if not self.enabled:
693 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000694 options = self.Options(verbose=True)
dpranke@chromium.org330f5342011-03-13 02:56:43 +0000695 root_dir = gclient_scm.os.path.realpath(tempfile.mkdtemp())
msb@chromium.orge28e4982009-09-25 20:51:45 +0000696 relpath = 'foo'
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000697 base_path = join(root_dir, relpath)
698 url = join(self.base_path, '.git')
msb@chromium.orge28e4982009-09-25 20:51:45 +0000699 try:
700 scm = gclient_scm.CreateSCM(url=url, root_dir=root_dir,
701 relpath=relpath)
702 file_list = []
703 scm.update(options, (), file_list)
704 self.assertEquals(len(file_list), 2)
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000705 self.assert_(gclient_scm.os.path.isfile(join(base_path, 'a')))
msb@chromium.org0f282062009-11-06 20:14:02 +0000706 self.assertEquals(scm.revinfo(options, (), None),
msb@chromium.orge28e4982009-09-25 20:51:45 +0000707 '069c602044c5388d2d15c3f875b057c852003458')
708 finally:
maruel@chromium.orgba551772010-02-03 18:21:42 +0000709 rmtree(root_dir)
maruel@chromium.orgcdbecc42011-02-09 04:31:47 +0000710 msg1 = (
711 "\n_____ foo at refs/heads/master\n\n"
712 "________ running 'git clone -b master --verbose %s %s' in '%s'\n"
713 "Initialized empty Git repository in %s\n") % (
714 join(self.root_dir, '.', '.git'),
715 join(root_dir, 'foo'),
716 root_dir,
717 join(gclient_scm.os.path.realpath(root_dir), 'foo', '.git') + '/')
718 msg2 = (
719 "\n_____ foo at refs/heads/master\n\n"
720 "________ running 'git clone -b master --verbose %s %s' in '%s'\n"
721 "Cloning into %s...\ndone.\n") % (
722 join(self.root_dir, '.', '.git'),
723 join(root_dir, 'foo'),
724 root_dir,
725 join(gclient_scm.os.path.realpath(root_dir), 'foo'))
726 out = sys.stdout.getvalue()
727 sys.stdout.close()
dpranke@chromium.org330f5342011-03-13 02:56:43 +0000728 sys.stdout = self._old_stdout
maruel@chromium.orgcdbecc42011-02-09 04:31:47 +0000729 self.assertTrue(out in (msg1, msg2), (out, msg1, msg2))
msb@chromium.orge28e4982009-09-25 20:51:45 +0000730
731 def testUpdateUpdate(self):
maruel@chromium.orgea6c2c52009-10-09 20:38:14 +0000732 if not self.enabled:
733 return
msb@chromium.orge28e4982009-09-25 20:51:45 +0000734 options = self.Options()
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000735 expected_file_list = [join(self.base_path, x) for x in ['a', 'b']]
msb@chromium.orge28e4982009-09-25 20:51:45 +0000736 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
737 relpath=self.relpath)
738 file_list = []
739 scm.update(options, (), file_list)
msb@chromium.org0f282062009-11-06 20:14:02 +0000740 self.assertEquals(file_list, expected_file_list)
741 self.assertEquals(scm.revinfo(options, (), None),
msb@chromium.orge28e4982009-09-25 20:51:45 +0000742 'a7142dc9f0009350b96a11f372b6ea658592aa95')
maruel@chromium.orgad80e3b2010-09-09 14:18:28 +0000743 self.checkstdout(
744 '\n_____ . at refs/heads/master\n'
745 'Updating 069c602..a7142dc\nFast-forward\n a | 1 +\n b | 1 +\n'
746 ' 2 files changed, 2 insertions(+), 0 deletions(-)\n\n')
msb@chromium.orge28e4982009-09-25 20:51:45 +0000747
nasser@codeaurora.orgd90ba3f2010-02-23 14:42:57 +0000748 def testUpdateUnstagedConflict(self):
749 if not self.enabled:
750 return
751 options = self.Options()
752 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
753 relpath=self.relpath)
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000754 file_path = join(self.base_path, 'b')
maruel@chromium.orgbf38a7e2010-12-14 18:15:54 +0000755 open(file_path, 'w').writelines('conflict\n')
maruel@chromium.orgcdbecc42011-02-09 04:31:47 +0000756 try:
757 scm.update(options, (), [])
758 self.fail()
759 except gclient_scm.gclient_utils.CheckCallError:
760 # The exact exception text varies across git versions so it's not worth
761 # verifying it. It's fine as long as it throws.
762 pass
763 # Manually flush stdout since we can't verify it's content accurately across
764 # git versions.
765 sys.stdout.getvalue()
766 sys.stdout.close()
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')
maruel@chromium.orgbf38a7e2010-12-14 18:15:54 +0000775 open(file_path, 'w').writelines('conflict\n')
776 # pylint: disable=W0212
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000777 scm._Run(['commit', '-am', 'test'], options)
778 __builtin__.raw_input = lambda x: 'y'
779 exception = ('Conflict while rebasing this branch.\n'
780 'Fix the conflict and run gclient again.\n'
781 'See \'man git-rebase\' for details.\n')
msb@chromium.org5bde4852009-12-14 16:47:12 +0000782 self.assertRaisesError(exception, scm.update, options, (), [])
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000783 exception = ('\n____ . at refs/heads/master\n'
784 '\tYou have unstaged changes.\n'
785 '\tPlease commit, stash, or reset.\n')
msb@chromium.org5bde4852009-12-14 16:47:12 +0000786 self.assertRaisesError(exception, scm.update, options, (), [])
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000787 # The hash always changes. Use a cheap trick.
788 start = ('\n________ running \'git commit -am test\' in \'%s\'\n'
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000789 '[new ') % join(self.root_dir, '.')
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000790 end = ('] test\n 1 files changed, 1 insertions(+), '
791 '1 deletions(-)\n\n_____ . at refs/heads/master\n'
792 'Attempting rebase onto refs/remotes/origin/master...\n')
maruel@chromium.orgcb2985f2010-11-03 14:08:31 +0000793 self.assertTrue(sys.stdout.getvalue().startswith(start))
794 self.assertTrue(sys.stdout.getvalue().endswith(end))
795 self.assertEquals(len(sys.stdout.getvalue()),
maruel@chromium.orgdd057f32010-09-20 19:33:31 +0000796 len(start) + len(end) + 7)
maruel@chromium.orgcb2985f2010-11-03 14:08:31 +0000797 sys.stdout.close()
msb@chromium.org5bde4852009-12-14 16:47:12 +0000798
msb@chromium.orge4af1ab2010-01-13 21:26:09 +0000799 def testUpdateNotGit(self):
800 if not self.enabled:
801 return
802 options = self.Options()
803 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
804 relpath=self.relpath)
maruel@chromium.org795a8c12010-10-05 19:54:29 +0000805 git_path = join(self.base_path, '.git')
maruel@chromium.orgba551772010-02-03 18:21:42 +0000806 rename(git_path, git_path + 'foo')
maruel@chromium.org389d6de2010-09-09 14:14:37 +0000807 exception = ('\n____ . at refs/heads/master\n'
808 '\tPath is not a git repo. No .git dir.\n'
809 '\tTo resolve:\n'
810 '\t\trm -rf .\n'
811 '\tAnd run gclient sync again\n')
msb@chromium.orge4af1ab2010-01-13 21:26:09 +0000812 self.assertRaisesError(exception, scm.update, options, (), [])
813
msb@chromium.org0f282062009-11-06 20:14:02 +0000814 def testRevinfo(self):
815 if not self.enabled:
816 return
817 options = self.Options()
818 scm = gclient_scm.CreateSCM(url=self.url, root_dir=self.root_dir,
819 relpath=self.relpath)
820 rev_info = scm.revinfo(options, (), None)
821 self.assertEquals(rev_info, '069c602044c5388d2d15c3f875b057c852003458')
822
msb@chromium.orge28e4982009-09-25 20:51:45 +0000823
msb@chromium.orge28e4982009-09-25 20:51:45 +0000824if __name__ == '__main__':
maruel@chromium.orgcdbecc42011-02-09 04:31:47 +0000825 if '-v' in sys.argv:
826 logging.basicConfig(
827 level=logging.DEBUG,
828 format='%(asctime).19s %(levelname)s %(filename)s:'
829 '%(lineno)s %(message)s')
msb@chromium.orge28e4982009-09-25 20:51:45 +0000830 unittest.main()
831
832# vim: ts=2:sw=2:tw=80:et: