depot_tools/tests/scm_unittest.py

#!/usr/bin/python
# Copyright (c) 2006-2009 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Unit tests for scm.py."""

from gclient_test import BaseTestCase
import scm
from super_mox import mox


class BaseSCMTestCase(BaseTestCase):
  def setUp(self):
    BaseTestCase.setUp(self)
    self.mox.StubOutWithMock(scm.gclient_utils, 'SubprocessCall')
    self.mox.StubOutWithMock(scm.gclient_utils, 'SubprocessCallAndFilter')


class RootTestCase(BaseSCMTestCase):
  def testMembersChanged(self):
    self.mox.ReplayAll()
    members = [
        'CaptureGit', 'CaptureGitStatus', 'GIT_COMMAND',
        'CaptureSVN', 'CaptureSVNHeadRevision', 'CaptureSVNInfo',
        'CaptureSVNStatus', 'RunSVN', 'RunSVNAndFilterOutput',
        'RunSVNAndGetFileList', 'SVN_COMMAND',
        'gclient_utils', 'os', 're', 'subprocess', 'sys', 'xml',
    ]
    # If this test fails, you should add the relevant test.
    self.compareMembers(scm, members)


class GitWrapperTestCase(BaseSCMTestCase):
  sample_git_import = """blob
mark :1
data 6
Hello

blob
mark :2
data 4
Bye

reset refs/heads/master
commit refs/heads/master
mark :3
author Bob <bob@example.com> 1253744361 -0700
committer Bob <bob@example.com> 1253744361 -0700
data 8
A and B
M 100644 :1 a
M 100644 :2 b

blob
mark :4
data 10
Hello
You

blob
mark :5
data 8
Bye
You

commit refs/heads/origin
mark :6
author Alice <alice@example.com> 1253744424 -0700
committer Alice <alice@example.com> 1253744424 -0700
data 13
Personalized
from :3
M 100644 :4 a
M 100644 :5 b

reset refs/heads/master
from :3
"""

  def CreateGitRepo(self, git_import, path):
    try:
      subprocess.Popen(['git', 'init'], stdout=subprocess.PIPE,
                       stderr=subprocess.STDOUT, cwd=path).communicate()
    except WindowsError:
      # git is not available, skip this test.
      return False
    subprocess.Popen(['git', 'fast-import'], stdin=subprocess.PIPE,
                     stdout=subprocess.PIPE, stderr=subprocess.STDOUT,
                     cwd=path).communicate(input=git_import)
    subprocess.Popen(['git', 'checkout'], stdout=subprocess.PIPE,
                     stderr=subprocess.STDOUT, cwd=path).communicate()
    return True

  def setUp(self):
    BaseSCMTestCase.setUp(self)
    self.args = self.Args()
    self.url = 'git://foo'
    self.root_dir = tempfile.mkdtemp()
    self.relpath = '.'
    self.base_path = os.path.join(self.root_dir, self.relpath)
    self.enabled = self.CreateGitRepo(self.sample_git_import, self.base_path)

  def tearDown(self):
    shutil.rmtree(self.root_dir)
    gclient_test.BaseTestCase.tearDown(self)


class SVNTestCase(BaseSCMTestCase):
  def setUp(self):
    BaseSCMTestCase.setUp(self)
    self.root_dir = self.Dir()
    self.args = self.Args()
    self.url = self.Url()
    self.relpath = 'asf'

  def testGetSVNFileInfo(self):
    xml_text = r"""<?xml version="1.0"?>
<info>
<entry kind="file" path="%s" revision="14628">
<url>http://src.chromium.org/svn/trunk/src/chrome/app/d</url>
<repository><root>http://src.chromium.org/svn</root></repository>
<wc-info>
<schedule>add</schedule>
<depth>infinity</depth>
<copy-from-url>http://src.chromium.org/svn/trunk/src/chrome/app/DEPS</copy-from-url>
<copy-from-rev>14628</copy-from-rev>
<checksum>369f59057ba0e6d9017e28f8bdfb1f43</checksum>
</wc-info>
</entry>
</info>
""" % self.url
    self.mox.StubOutWithMock(scm, 'CaptureSVN')
    scm.CaptureSVN(['info', '--xml', self.url], '.', True).AndReturn(xml_text)
    expected = {
      'URL': 'http://src.chromium.org/svn/trunk/src/chrome/app/d',
      'UUID': None,
      'Repository Root': 'http://src.chromium.org/svn',
      'Schedule': 'add',
      'Copied From URL':
        'http://src.chromium.org/svn/trunk/src/chrome/app/DEPS',
      'Copied From Rev': '14628',
      'Path': self.url,
      'Revision': 14628,
      'Node Kind': 'file',
    }
    self.mox.ReplayAll()
    file_info = scm.CaptureSVNInfo(self.url, '.', True)
    self.assertEquals(sorted(file_info.items()), sorted(expected.items()))

  def testCaptureSvnInfo(self):
    xml_text = """<?xml version="1.0"?>
<info>
<entry
   kind="dir"
   path="."
   revision="35">
<url>%s</url>
<repository>
<root>%s</root>
<uuid>7b9385f5-0452-0410-af26-ad4892b7a1fb</uuid>
</repository>
<wc-info>
<schedule>normal</schedule>
<depth>infinity</depth>
</wc-info>
<commit
   revision="35">
<author>maruel</author>
<date>2008-12-04T20:12:19.685120Z</date>
</commit>
</entry>
</info>
""" % (self.url, self.root_dir)
    self.mox.StubOutWithMock(scm, 'CaptureSVN')
    scm.CaptureSVN(['info', '--xml', self.url], '.', True).AndReturn(xml_text)
    self.mox.ReplayAll()
    file_info = scm.CaptureSVNInfo(self.url, '.', True)
    expected = {
      'URL': self.url,
      'UUID': '7b9385f5-0452-0410-af26-ad4892b7a1fb',
      'Revision': 35,
      'Repository Root': self.root_dir,
      'Schedule': 'normal',
      'Copied From URL': None,
      'Copied From Rev': None,
      'Path': '.',
      'Node Kind': 'dir',
    }
    self.assertEqual(file_info, expected)

  def testCaptureSVNStatus(self):
    text =r"""<?xml version="1.0"?>
<status>
<target path=".">
<entry path="unversionned_file.txt">
<wc-status props="none" item="unversioned"></wc-status>
</entry>
<entry path="build\internal\essential.vsprops">
<wc-status props="normal" item="modified" revision="14628">
<commit revision="13818">
<author>ajwong@chromium.org</author>
<date>2009-04-16T00:42:06.872358Z</date>
</commit>
</wc-status>
</entry>
<entry path="chrome\app\d">
<wc-status props="none" copied="true" tree-conflicted="true" item="added">
</wc-status>
</entry>
<entry path="chrome\app\DEPS">
<wc-status props="modified" item="modified" revision="14628">
<commit revision="1279">
<author>brettw@google.com</author>
<date>2008-08-23T17:16:42.090152Z</date>
</commit>
</wc-status>
</entry>
<entry path="scripts\master\factory\gclient_factory.py">
<wc-status props="normal" item="conflicted" revision="14725">
<commit revision="14633">
<author>nsylvain@chromium.org</author>
<date>2009-04-27T19:37:17.977400Z</date>
</commit>
</wc-status>
</entry>
</target>
</status>
"""
    proc = self.mox.CreateMockAnything()
    scm.subprocess.Popen(['svn', 'status', '--xml', '.'],
                         cwd=None,
                         shell=scm.sys.platform.startswith('win'),
                         stderr=None,
                         stdout=scm.subprocess.PIPE).AndReturn(proc)
    proc.communicate().AndReturn((text, 0))

    self.mox.ReplayAll()
    info = scm.CaptureSVNStatus('.')
    expected = [
      ('?      ', 'unversionned_file.txt'),
      ('M      ', 'build\\internal\\essential.vsprops'),
      ('A  +   ', 'chrome\\app\\d'),
      ('MM     ', 'chrome\\app\\DEPS'),
      ('C      ', 'scripts\\master\\factory\\gclient_factory.py'),
    ]
    self.assertEquals(sorted(info), sorted(expected))

  def testRunSVN(self):
    param2 = 'bleh'
    scm.gclient_utils.SubprocessCall(['svn', 'foo', 'bar'],
                                     param2).AndReturn(None)
    self.mox.ReplayAll()
    scm.RunSVN(['foo', 'bar'], param2)

  def testCaptureSVNStatusEmpty(self):
    text = r"""<?xml version="1.0"?>
    <status>
    <target
       path="perf">
       </target>
       </status>"""
    proc = self.mox.CreateMockAnything()
    scm.subprocess.Popen(['svn', 'status', '--xml'],
                         cwd=None,
                         shell=scm.sys.platform.startswith('win'),
                         stderr=None,
                         stdout=scm.subprocess.PIPE).AndReturn(proc)
    proc.communicate().AndReturn((text, 0))
    self.mox.ReplayAll()
    info = scm.CaptureSVNStatus(None)
    self.assertEquals(info, [])


if __name__ == '__main__':
  import unittest
  unittest.main()

# vim: ts=2:sw=2:tw=80:et: