| Index: bin/cros_mark_chrome_as_stable_unittest.py
|
| diff --git a/bin/cros_mark_chrome_as_stable_unittest.py b/bin/cros_mark_chrome_as_stable_unittest.py
|
| deleted file mode 100755
|
| index dfc981de8b31a5184b06ddd0cf8473b0e9835fb4..0000000000000000000000000000000000000000
|
| --- a/bin/cros_mark_chrome_as_stable_unittest.py
|
| +++ /dev/null
|
| @@ -1,270 +0,0 @@
|
| -#!/usr/bin/python
|
| -
|
| -# Copyright (c) 2010 The Chromium OS 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 cros_mark_chrome_as_stable.py."""
|
| -
|
| -import cros_mark_chrome_as_stable
|
| -import mox
|
| -import os
|
| -import shutil
|
| -import sys
|
| -import tempfile
|
| -import unittest
|
| -import urllib
|
| -
|
| -sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
|
| -import cros_mark_as_stable
|
| -
|
| -unstable_data = 'KEYWORDS=~x86 ~arm'
|
| -stable_data = 'KEYWORDS=x86 arm'
|
| -
|
| -def _TouchAndWrite(path, data=None):
|
| - """Writes data (if it exists) to the file specified by the path."""
|
| - fh = open(path, 'w')
|
| - if data:
|
| - fh.write(data)
|
| -
|
| - fh.close()
|
| -
|
| -
|
| -class CrosMarkChromeAsStable(mox.MoxTestBase):
|
| -
|
| - def setUp(self):
|
| - """Setup vars and create mock dir."""
|
| - mox.MoxTestBase.setUp(self)
|
| - self.tmp_overlay = tempfile.mkdtemp(prefix='chromiumos-overlay')
|
| - self.mock_chrome_dir = os.path.join(self.tmp_overlay, 'chromeos-base',
|
| - 'chromeos-chrome')
|
| - os.makedirs(self.mock_chrome_dir)
|
| -
|
| - self.unstable = os.path.join(self.mock_chrome_dir,
|
| - 'chromeos-chrome-9999.ebuild')
|
| - self.sticky_branch = '8.0.224'
|
| - self.sticky_version = '%s.503' % self.sticky_branch
|
| - self.sticky = os.path.join(self.mock_chrome_dir,
|
| - 'chromeos-chrome-%s.ebuild' %
|
| - self.sticky_version)
|
| - self.sticky_rc_version = '%s.504' % self.sticky_branch
|
| - self.sticky_rc = os.path.join(self.mock_chrome_dir,
|
| - 'chromeos-chrome-%s_rc-r1.ebuild' %
|
| - self.sticky_rc_version)
|
| - self.latest_stable_version = '8.0.300.1'
|
| - self.latest_stable = os.path.join(self.mock_chrome_dir,
|
| - 'chromeos-chrome-%s_rc-r2.ebuild' %
|
| - self.latest_stable_version)
|
| - self.tot_stable_version = '9.0.305.0'
|
| - self.tot_stable = os.path.join(self.mock_chrome_dir,
|
| - 'chromeos-chrome-%s_alpha-r1.ebuild' %
|
| - self.tot_stable_version)
|
| -
|
| - self.sticky_new_rc_version = '%s.520' % self.sticky_branch
|
| - self.sticky_new_rc = os.path.join(self.mock_chrome_dir,
|
| - 'chromeos-chrome-%s_rc-r1.ebuild' %
|
| - self.sticky_new_rc_version)
|
| - self.latest_new_version = '9.0.305.1'
|
| - self.latest_new = os.path.join(self.mock_chrome_dir,
|
| - 'chromeos-chrome-%s_rc-r1.ebuild' %
|
| - self.latest_new_version)
|
| - self.tot_new_version = '9.0.306.0'
|
| - self.tot_new = os.path.join(self.mock_chrome_dir,
|
| - 'chromeos-chrome-%s_alpha-r1.ebuild' %
|
| - self.tot_new_version)
|
| -
|
| - _TouchAndWrite(self.unstable, unstable_data)
|
| - _TouchAndWrite(self.sticky, stable_data)
|
| - _TouchAndWrite(self.sticky_rc, stable_data)
|
| - _TouchAndWrite(self.latest_stable, stable_data)
|
| - _TouchAndWrite(self.tot_stable, stable_data)
|
| -
|
| - def tearDown(self):
|
| - """Cleans up mock dir."""
|
| - shutil.rmtree(self.tmp_overlay)
|
| -
|
| - def testFindChromeCandidates(self):
|
| - """Test creation of stable ebuilds from mock dir."""
|
| - unstable, stable_ebuilds = cros_mark_chrome_as_stable.FindChromeCandidates(
|
| - self.mock_chrome_dir)
|
| -
|
| - self.assertEqual(unstable.ebuild_path, self.unstable)
|
| - self.assertEqual(len(stable_ebuilds), 4)
|
| - self.assertTrue(cros_mark_chrome_as_stable.ChromeEBuild(self.sticky) in
|
| - stable_ebuilds)
|
| - self.assertTrue(cros_mark_chrome_as_stable.ChromeEBuild(self.sticky_rc) in
|
| - stable_ebuilds)
|
| - self.assertTrue(cros_mark_chrome_as_stable.ChromeEBuild(self.latest_stable)
|
| - in stable_ebuilds)
|
| - self.assertTrue(cros_mark_chrome_as_stable.ChromeEBuild(self.tot_stable) in
|
| - stable_ebuilds)
|
| -
|
| - def _GetStableEBuilds(self):
|
| - """Common helper to create a list of stable ebuilds."""
|
| - return [
|
| - cros_mark_chrome_as_stable.ChromeEBuild(self.sticky),
|
| - cros_mark_chrome_as_stable.ChromeEBuild(self.sticky_rc),
|
| - cros_mark_chrome_as_stable.ChromeEBuild(self.latest_stable),
|
| - cros_mark_chrome_as_stable.ChromeEBuild(self.tot_stable),
|
| - ]
|
| -
|
| - def testTOTFindChromeUprevCandidate(self):
|
| - """Tests if we can find tot uprev candidate from our mock dir data."""
|
| - stable_ebuilds = self._GetStableEBuilds()
|
| -
|
| - candidate = cros_mark_chrome_as_stable.FindChromeUprevCandidate(
|
| - stable_ebuilds, cros_mark_chrome_as_stable.TIP_OF_TRUNK,
|
| - self.sticky_branch)
|
| -
|
| - self.assertEqual(candidate.ebuild_path, self.tot_stable)
|
| -
|
| - def testLatestFindChromeUprevCandidate(self):
|
| - """Tests if we can find latest uprev candidate from our mock dir data."""
|
| - stable_ebuilds = self._GetStableEBuilds()
|
| -
|
| - candidate = cros_mark_chrome_as_stable.FindChromeUprevCandidate(
|
| - stable_ebuilds, cros_mark_chrome_as_stable.LATEST_RELEASE,
|
| - self.sticky_branch)
|
| -
|
| - self.assertEqual(candidate.ebuild_path, self.latest_stable)
|
| -
|
| - def testStickyFindChromeUprevCandidate(self):
|
| - """Tests if we can find sticky uprev candidate from our mock dir data."""
|
| - stable_ebuilds = self._GetStableEBuilds()
|
| -
|
| - candidate = cros_mark_chrome_as_stable.FindChromeUprevCandidate(
|
| - stable_ebuilds, cros_mark_chrome_as_stable.STICKY,
|
| - self.sticky_branch)
|
| -
|
| - self.assertEqual(candidate.ebuild_path, self.sticky_rc)
|
| -
|
| - def testGetTipOfTrunkSvnRevision(self):
|
| - """Tests if we can get the latest svn revision from TOT."""
|
| - self.mox.StubOutWithMock(cros_mark_chrome_as_stable, 'RunCommand')
|
| - cros_mark_chrome_as_stable.RunCommand(
|
| - ['svn', 'info', cros_mark_chrome_as_stable._GetSvnUrl()],
|
| - redirect_stdout=True).AndReturn(
|
| - 'Some Junk 2134\nRevision: 12345\nOtherInfo: test_data')
|
| - self.mox.ReplayAll()
|
| - revision = cros_mark_chrome_as_stable._GetTipOfTrunkSvnRevision()
|
| - self.mox.VerifyAll()
|
| - self.assertEquals(revision, '12345')
|
| -
|
| - def testGetTipOfTrunkVersion(self):
|
| - """Tests if we get the latest version from TOT."""
|
| - self.mox.StubOutWithMock(urllib, 'urlopen')
|
| - mock_file = self.mox.CreateMock(file)
|
| - urllib.urlopen(os.path.join(cros_mark_chrome_as_stable._GetSvnUrl(), 'src',
|
| - 'chrome', 'VERSION')).AndReturn(mock_file)
|
| - mock_file.read().AndReturn('A=8\nB=0\nC=256\nD=0')
|
| - mock_file.close()
|
| -
|
| - self.mox.ReplayAll()
|
| - version = cros_mark_chrome_as_stable._GetTipOfTrunkVersion()
|
| - self.mox.VerifyAll()
|
| - self.assertEquals(version, '8.0.256.0')
|
| -
|
| - def testGetLatestRelease(self):
|
| - """Tests if we can find the latest release from our mock url data."""
|
| - test_data = '\n'.join(['7.0.224.1/',
|
| - '7.0.224.2/',
|
| - '8.0.365.5/',
|
| - 'LATEST.txt'])
|
| - self.mox.StubOutWithMock(cros_mark_chrome_as_stable, 'RunCommand')
|
| - cros_mark_chrome_as_stable.RunCommand(
|
| - ['svn', 'ls', 'http://src.chromium.org/svn/releases'],
|
| - redirect_stdout=True).AndReturn('some_data')
|
| - cros_mark_chrome_as_stable.RunCommand(
|
| - ['sort', '--version-sort'], input='some_data',
|
| - redirect_stdout=True).AndReturn(test_data)
|
| - self.mox.ReplayAll()
|
| - release = cros_mark_chrome_as_stable._GetLatestRelease()
|
| - self.mox.VerifyAll()
|
| - self.assertEqual('8.0.365.5', release)
|
| -
|
| - def testGetLatestStickyRelease(self):
|
| - """Tests if we can find the latest sticky release from our mock url data."""
|
| - test_data = '\n'.join(['7.0.222.1/',
|
| - '8.0.224.2/',
|
| - '8.0.365.5/',
|
| - 'LATEST.txt'])
|
| - self.mox.StubOutWithMock(cros_mark_chrome_as_stable, 'RunCommand')
|
| - cros_mark_chrome_as_stable.RunCommand(
|
| - ['svn', 'ls', 'http://src.chromium.org/svn/releases'],
|
| - redirect_stdout=True).AndReturn('some_data')
|
| - cros_mark_chrome_as_stable.RunCommand(
|
| - ['sort', '--version-sort'], input='some_data',
|
| - redirect_stdout=True).AndReturn(test_data)
|
| - self.mox.ReplayAll()
|
| - release = cros_mark_chrome_as_stable._GetLatestRelease(self.sticky_branch)
|
| - self.mox.VerifyAll()
|
| - self.assertEqual('8.0.224.2', release)
|
| -
|
| - def testStickyEBuild(self):
|
| - """Tests if we can find the sticky ebuild from our mock directories."""
|
| - stable_ebuilds = self._GetStableEBuilds()
|
| - sticky_ebuild = cros_mark_chrome_as_stable._GetStickyEBuild(
|
| - stable_ebuilds)
|
| - self.assertEqual(sticky_ebuild.chrome_version, self.sticky_version)
|
| -
|
| - def testChromeEBuildInit(self):
|
| - """Tests if the chrome_version is set correctly in a ChromeEBuild."""
|
| - ebuild = cros_mark_chrome_as_stable.ChromeEBuild(self.sticky)
|
| - self.assertEqual(ebuild.chrome_version, self.sticky_version)
|
| -
|
| - def _CommonMarkAsStableTest(self, chrome_rev, new_version, old_ebuild_path,
|
| - new_ebuild_path, commit_string_indicator):
|
| - """Common function used for test functions for MarkChromeEBuildAsStable.
|
| -
|
| - This function stubs out others calls, and runs MarkChromeEBuildAsStable
|
| - with the specified args.
|
| -
|
| - Args:
|
| - chrome_rev: standard chrome_rev argument
|
| - new_version: version we are revving up to
|
| - old_ebuild_path: path to the stable ebuild
|
| - new_ebuild_path: path to the to be created path
|
| - commit_string_indicator: a string that the commit message must contain
|
| - """
|
| - self.mox.StubOutWithMock(cros_mark_chrome_as_stable, 'RunCommand')
|
| - self.mox.StubOutWithMock(cros_mark_as_stable.EBuildStableMarker,
|
| - 'CommitChange')
|
| - stable_candidate = cros_mark_chrome_as_stable.ChromeEBuild(old_ebuild_path)
|
| - unstable_ebuild = cros_mark_chrome_as_stable.ChromeEBuild(self.unstable)
|
| - sticky_ebuild = cros_mark_chrome_as_stable.ChromeEBuild(self.sticky)
|
| - chrome_version = new_version
|
| - commit = None
|
| - overlay_dir = self.mock_chrome_dir
|
| -
|
| - cros_mark_chrome_as_stable.RunCommand(['git', 'add', new_ebuild_path])
|
| - cros_mark_chrome_as_stable.RunCommand(['git', 'rm', old_ebuild_path])
|
| - cros_mark_as_stable.EBuildStableMarker.CommitChange(
|
| - mox.StrContains(commit_string_indicator))
|
| -
|
| - self.mox.ReplayAll()
|
| - cros_mark_chrome_as_stable.MarkChromeEBuildAsStable(
|
| - stable_candidate, unstable_ebuild, chrome_rev, chrome_version, commit,
|
| - overlay_dir, sticky_ebuild)
|
| - self.mox.VerifyAll()
|
| -
|
| - def testStickyMarkAsStable(self):
|
| - """Tests to see if we can mark chrome as stable for a new sticky release."""
|
| - self._CommonMarkAsStableTest(cros_mark_chrome_as_stable.STICKY,
|
| - self.sticky_new_rc_version, self.sticky_rc,
|
| - self.sticky_new_rc, 'stable_release')
|
| -
|
| - def testLatestMarkAsStable(self):
|
| - """Tests to see if we can mark chrome for a latest release."""
|
| - self._CommonMarkAsStableTest(cros_mark_chrome_as_stable.LATEST_RELEASE,
|
| - self.latest_new_version, self.latest_stable,
|
| - self.latest_new, 'latest_release')
|
| -
|
| - def testTotMarkAsStable(self):
|
| - """Tests to see if we can mark chrome for tot."""
|
| - self._CommonMarkAsStableTest(cros_mark_chrome_as_stable.TIP_OF_TRUNK,
|
| - self.tot_new_version, self.tot_stable,
|
| - self.tot_new, 'tot')
|
| -
|
| -
|
| -if __name__ == '__main__':
|
| - unittest.main()
|
|
|