| # Copyright 2016 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. |
| |
| """Unittests for loman.py""" |
| |
| from __future__ import print_function |
| |
| import os |
| |
| from chromite.lib import cros_test_lib |
| from chromite.lib import osutils |
| from chromite.scripts import loman |
| |
| |
| class ParserTest(cros_test_lib.OutputTestCase): |
| """Tests for the CLI parser.""" |
| |
| def setUp(self): |
| self.parser = loman.GetParser() |
| |
| def testNoCommand(self): |
| """Require a command at least.""" |
| with self.OutputCapturer(): |
| self.assertRaises(SystemExit, self.parser.parse_args, []) |
| |
| def testBadCommand(self): |
| """Reject unknown commands.""" |
| with self.OutputCapturer(): |
| self.assertRaises(SystemExit, self.parser.parse_args, ['flyaway']) |
| |
| def testAddCommand(self): |
| """Verify basic add command behavior.""" |
| with self.OutputCapturer(): |
| self.parser.parse_args(['add', '--workon', 'project']) |
| self.parser.parse_args(['add', 'project', 'path', '--remote', 'foo']) |
| |
| def testUpgradeMinilayoutCommand(self): |
| """Verify basic upgrade command behavior.""" |
| with self.OutputCapturer(): |
| self.parser.parse_args(['upgrade-minilayout']) |
| |
| |
| class ManifestTest(cros_test_lib.TempDirTestCase): |
| """Tests that need a real .repo/ manifest layout.""" |
| |
| def setUp(self): |
| # The loman code looks for the repo root, so make one, and chdir there. |
| os.chdir(self.tempdir) |
| |
| for d in ('repo', 'manifests', 'manifests.git'): |
| osutils.SafeMakedirs(os.path.join('.repo', d)) |
| |
| for m in ('default.xml', 'full.xml', 'minilayout.xml'): |
| osutils.Touch(os.path.join('.repo', 'manifests', m)) |
| |
| self._SetManifest('default.xml') |
| |
| def _SetManifest(self, manifest): |
| """Set active manifest to point to |manifest|.""" |
| source = os.path.join('.repo', 'manifest.xml') |
| target = os.path.join('.repo', 'manifests', manifest) |
| osutils.SafeUnlink(source) |
| os.symlink(target, source) |
| |
| |
| class UpgradeMinilayoutTest(cros_test_lib.MockOutputTestCase, ManifestTest): |
| """Tests for the upgrade minilayout logic.""" |
| |
| def testNoUpgradeNeeded(self): |
| """Don't run update when it isn't needed.""" |
| with self.OutputCapturer(): |
| self.PatchObject(loman, '_UpgradeMinilayout', side_effect=Exception) |
| loman.main(['upgrade-minilayout']) |
| |
| def testUpgradeNeeded(self): |
| """Run update when it's needed.""" |
| class _Exception(Exception): |
| """Custom exception.""" |
| |
| self._SetManifest('minilayout.xml') |
| with self.OutputCapturer(): |
| self.PatchObject(loman, '_UpgradeMinilayout', side_effect=_Exception) |
| self.assertRaises(_Exception, loman.main, ['upgrade-minilayout']) |
| |
| def testAutoUpgrade(self): |
| """Run update automatically.""" |
| class _Exception(Exception): |
| """Custom exception.""" |
| |
| self._SetManifest('minilayout.xml') |
| with self.OutputCapturer(): |
| self.PatchObject(loman, '_UpgradeMinilayout', side_effect=_Exception) |
| self.assertRaises(_Exception, loman.main, ['add', '--workon', 'proj']) |
| |
| |
| class AddTest(cros_test_lib.MockOutputTestCase, ManifestTest): |
| """Tests for the add command.""" |
| |
| def testRejectBadCommands(self): |
| """Reject bad invocations.""" |
| bad_cmds = ( |
| # Missing path. |
| ['add'], |
| # Extra project. |
| ['add', '--workon', 'path', 'project'], |
| # Missing --remote. |
| ['add', 'path', 'project'], |
| # Missing project. |
| ['add', 'path', '--remote', 'remote'], |
| ) |
| with self.OutputCapturer(): |
| for cmd in bad_cmds: |
| self.assertRaises(SystemExit, loman.main, cmd) |