mirror of
https://github.com/flatcar/scripts.git
synced 2025-08-08 13:36:58 +02:00
9f8e53b27b
Code was changed to add a CR. Fix tests to accomodate. BUG=5258 TEST=Unittests pass again. Change-Id: Iacdbffb14d14f60a43883db3ee12b99a5eeb9f65 Review URL: http://codereview.chromium.org/3014045
204 lines
7.3 KiB
Python
Executable File
204 lines
7.3 KiB
Python
Executable File
#!/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_as_stable.py."""
|
|
|
|
|
|
import mox
|
|
import os
|
|
import sys
|
|
import unittest
|
|
|
|
# Required to include '.' in the python path.
|
|
sys.path.append(os.path.dirname(__file__))
|
|
import cros_mark_as_stable
|
|
|
|
class GitBranchTest(mox.MoxTestBase):
|
|
|
|
def setUp(self):
|
|
mox.MoxTestBase.setUp(self)
|
|
# Always stub RunCommmand out as we use it in every method.
|
|
self.mox.StubOutWithMock(cros_mark_as_stable, '_RunCommand')
|
|
self._branch = 'test_branch'
|
|
|
|
def testCreateBranchNoPrevious(self):
|
|
# Test init with no previous branch existing.
|
|
branch = cros_mark_as_stable._GitBranch(self._branch)
|
|
self.mox.StubOutWithMock(branch, 'Exists')
|
|
self.mox.StubOutWithMock(branch, '_Checkout')
|
|
branch.Exists().AndReturn(False)
|
|
branch._Checkout(self._branch)
|
|
self.mox.ReplayAll()
|
|
branch.CreateBranch()
|
|
self.mox.VerifyAll()
|
|
|
|
def testCreateBranchWithPrevious(self):
|
|
# Test init with previous branch existing.
|
|
branch = cros_mark_as_stable._GitBranch(self._branch)
|
|
self.mox.StubOutWithMock(branch, 'Exists')
|
|
self.mox.StubOutWithMock(branch, 'Delete')
|
|
self.mox.StubOutWithMock(branch, '_Checkout')
|
|
branch.Exists().AndReturn(True)
|
|
branch.Delete()
|
|
branch._Checkout(self._branch)
|
|
self.mox.ReplayAll()
|
|
branch.CreateBranch()
|
|
self.mox.VerifyAll()
|
|
|
|
def testCheckoutCreate(self):
|
|
# Test init with no previous branch existing.
|
|
cros_mark_as_stable._RunCommand('git checkout -b %s origin' % self._branch)
|
|
self.mox.ReplayAll()
|
|
branch = cros_mark_as_stable._GitBranch(self._branch)
|
|
branch._Checkout(self._branch)
|
|
self.mox.VerifyAll()
|
|
|
|
def testCheckoutNoCreate(self):
|
|
# Test init with previous branch existing.
|
|
cros_mark_as_stable._RunCommand('git checkout master')
|
|
self.mox.ReplayAll()
|
|
branch = cros_mark_as_stable._GitBranch(self._branch)
|
|
branch._Checkout('master', False)
|
|
self.mox.VerifyAll()
|
|
|
|
def testDelete(self):
|
|
branch = cros_mark_as_stable._GitBranch(self._branch)
|
|
self.mox.StubOutWithMock(branch, '_Checkout')
|
|
branch._Checkout('master', create=False)
|
|
cros_mark_as_stable._RunCommand('git branch -D ' + self._branch)
|
|
self.mox.ReplayAll()
|
|
branch.Delete()
|
|
self.mox.VerifyAll()
|
|
|
|
def testExists(self):
|
|
branch = cros_mark_as_stable._GitBranch(self._branch)
|
|
|
|
# Test if branch exists that is created
|
|
cros_mark_as_stable._RunCommand('git branch').AndReturn(
|
|
'%s %s' % (self._branch, 'master'))
|
|
self.mox.ReplayAll()
|
|
self.assertTrue(branch.Exists())
|
|
self.mox.VerifyAll()
|
|
|
|
|
|
class EBuildTest(mox.MoxTestBase):
|
|
|
|
def setUp(self):
|
|
mox.MoxTestBase.setUp(self)
|
|
self.package = 'test_package'
|
|
self.ebuild_path = '/path/test_package-0.0.1-r1.ebuild'
|
|
self.ebuild_path_no_rev = '/path/test_package-0.0.1.ebuild'
|
|
|
|
def testInit(self):
|
|
self.mox.StubOutWithMock(cros_mark_as_stable._EBuild, '_FindEBuildPath')
|
|
self.mox.StubOutWithMock(cros_mark_as_stable._EBuild, '_ParseEBuildPath')
|
|
|
|
cros_mark_as_stable._EBuild._FindEBuildPath(
|
|
self.package).AndReturn(self.ebuild_path)
|
|
cros_mark_as_stable._EBuild._ParseEBuildPath(
|
|
self.ebuild_path).AndReturn(['/path/test_package-0.0.1',
|
|
'/path/test_package',
|
|
1])
|
|
self.mox.ReplayAll()
|
|
ebuild = cros_mark_as_stable._EBuild(self.package, 'my_id')
|
|
self.mox.VerifyAll()
|
|
self.assertEquals(ebuild.package, self.package)
|
|
self.assertEquals(ebuild.ebuild_path, self.ebuild_path)
|
|
self.assertEquals(ebuild.ebuild_path_no_revision,
|
|
'/path/test_package-0.0.1')
|
|
self.assertEquals(ebuild.ebuild_path_no_version, '/path/test_package')
|
|
self.assertEquals(ebuild.current_revision, 1)
|
|
self.assertEquals(ebuild.commit_id, 'my_id')
|
|
|
|
def testFindEBuildPath(self):
|
|
self.mox.StubOutWithMock(cros_mark_as_stable, '_RunCommand')
|
|
cros_mark_as_stable._RunCommand(
|
|
'equery-x86-generic which %s 2> /dev/null' % self.package).AndReturn(
|
|
self.ebuild_path)
|
|
self.mox.ReplayAll()
|
|
path = cros_mark_as_stable._EBuild._FindEBuildPath(self.package)
|
|
self.mox.VerifyAll()
|
|
self.assertEquals(path, self.ebuild_path)
|
|
|
|
def testParseEBuildPath(self):
|
|
# Test with ebuild with revision number.
|
|
no_rev, no_version, revision = cros_mark_as_stable._EBuild._ParseEBuildPath(
|
|
self.ebuild_path)
|
|
self.assertEquals(no_rev, '/path/test_package-0.0.1')
|
|
self.assertEquals(no_version, '/path/test_package')
|
|
self.assertEquals(revision, 1)
|
|
|
|
def testParseEBuildPathNoRevisionNumber(self):
|
|
# Test with ebuild without revision number.
|
|
no_rev, no_version, revision = cros_mark_as_stable._EBuild._ParseEBuildPath(
|
|
self.ebuild_path_no_rev)
|
|
self.assertEquals(no_rev, '/path/test_package-0.0.1')
|
|
self.assertEquals(no_version, '/path/test_package')
|
|
self.assertEquals(revision, 0)
|
|
|
|
|
|
class EBuildStableMarkerTest(mox.MoxTestBase):
|
|
|
|
def setUp(self):
|
|
mox.MoxTestBase.setUp(self)
|
|
self.mox.StubOutWithMock(cros_mark_as_stable, '_RunCommand')
|
|
self.m_ebuild = self.mox.CreateMock(cros_mark_as_stable._EBuild)
|
|
self.m_ebuild.package = 'test_package'
|
|
self.m_ebuild.current_revision = 1
|
|
self.m_ebuild.ebuild_path_no_revision = '/path/test_package-0.0.1'
|
|
self.m_ebuild.ebuild_path_no_version = '/path/test_package'
|
|
self.m_ebuild.ebuild_path = '/path/test_package-0.0.1-r1.ebuild'
|
|
self.revved_ebuild_path = '/path/test_package-0.0.1-r2.ebuild'
|
|
|
|
def testRevEBuild(self):
|
|
self.mox.StubOutWithMock(cros_mark_as_stable.fileinput, 'input')
|
|
self.mox.StubOutWithMock(cros_mark_as_stable.shutil, 'copyfile')
|
|
m_file = self.mox.CreateMock(file)
|
|
|
|
# Prepare mock fileinput. This tests to make sure both the commit id
|
|
# and keywords are changed correctly.
|
|
mock_file = ['EAPI=2', 'CROS_WORKON_COMMIT=old_id',
|
|
'KEYWORDS=\"~x86 ~arm\"', 'src_unpack(){}']
|
|
|
|
cros_mark_as_stable.shutil.copyfile(
|
|
self.m_ebuild.ebuild_path_no_version + '-9999.ebuild',
|
|
self.revved_ebuild_path)
|
|
cros_mark_as_stable.fileinput.input(self.revved_ebuild_path,
|
|
inplace=1).AndReturn(mock_file)
|
|
m_file.write('EAPI=2')
|
|
m_file.write('CROS_WORKON_COMMIT="my_id"\n')
|
|
m_file.write('KEYWORDS="x86 arm"')
|
|
m_file.write('src_unpack(){}')
|
|
cros_mark_as_stable._RunCommand('git add ' + self.revved_ebuild_path)
|
|
cros_mark_as_stable._RunCommand('git rm ' + self.m_ebuild.ebuild_path)
|
|
|
|
self.mox.ReplayAll()
|
|
marker = cros_mark_as_stable.EBuildStableMarker(self.m_ebuild)
|
|
marker.RevEBuild('my_id', redirect_file=m_file)
|
|
self.mox.VerifyAll()
|
|
|
|
|
|
def testCommitChange(self):
|
|
mock_message = 'Commit me'
|
|
cros_mark_as_stable._RunCommand(
|
|
'git commit -am "%s"' % mock_message)
|
|
self.mox.ReplayAll()
|
|
marker = cros_mark_as_stable.EBuildStableMarker(self.m_ebuild)
|
|
marker.CommitChange(mock_message)
|
|
self.mox.VerifyAll()
|
|
|
|
def testPushChange(self):
|
|
#cros_mark_as_stable._RunCommand('git push')
|
|
#self.mox.ReplayAll()
|
|
#marker = cros_mark_as_stable.EBuildStableMarker(self.m_ebuild)
|
|
#marker.PushChange()
|
|
#self.mox.VerifyAll()
|
|
pass
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|