mirror of
https://github.com/flatcar/scripts.git
synced 2025-08-12 15:36:58 +02:00
865a42edee
BUG=5787 TEST=Ran unittests. Change-Id: I6b8cc37eb2bf5f81f803f61b886106b6f08f3315 Review URL: http://codereview.chromium.org/3331013
112 lines
3.5 KiB
Python
Executable File
112 lines
3.5 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.
|
|
|
|
"""Unittests for loman."""
|
|
|
|
import os
|
|
import StringIO
|
|
import sys
|
|
import tempfile
|
|
import unittest
|
|
|
|
import loman
|
|
|
|
_TEST_MANIFEST1 = """<manifest>
|
|
<project name="foo" path="path/to/foo" workon="True" />
|
|
</manifest>"""
|
|
|
|
class LocalManifestTest(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.utf8 = "<?xml version='1.0' encoding='UTF-8'?>\n"
|
|
self.tiny_manifest = '<manifest>\n</manifest>'
|
|
|
|
def testSimpleParse(self):
|
|
ptree = loman.LocalManifest()
|
|
ptree.Parse()
|
|
|
|
def testParse(self):
|
|
ptree = loman.LocalManifest(self.tiny_manifest)
|
|
ptree.Parse()
|
|
self.assertEqual(ptree.ToString(), self.utf8 + self.tiny_manifest)
|
|
|
|
def testUTF8Parse(self):
|
|
ptree = loman.LocalManifest(self.utf8 + self.tiny_manifest)
|
|
ptree.Parse()
|
|
self.assertEqual(ptree.ToString(), self.utf8 + self.tiny_manifest)
|
|
|
|
def testAddNew(self):
|
|
ptree = loman.LocalManifest('<manifest>\n</manifest>')
|
|
ptree.Parse()
|
|
self.assertTrue(ptree.AddWorkonProject('foo', 'path/to/foo'))
|
|
self.assertEqual(
|
|
ptree.ToString(),
|
|
self.utf8 + '<manifest>\n'
|
|
'<project name="foo" path="path/to/foo" workon="True" />\n'
|
|
'</manifest>')
|
|
|
|
def testAddDup(self):
|
|
ptree = loman.LocalManifest('<manifest>\n</manifest>')
|
|
ptree.Parse()
|
|
ptree.AddWorkonProject('foo', 'path/to/foo')
|
|
self.assertTrue(not ptree.AddWorkonProject('foo', 'path/to/foo'))
|
|
self.assertTrue(not ptree.AddWorkonProject('foo', 'path/foo'))
|
|
self.assertTrue(not ptree.AddWorkonProject('foobar', 'path/to/foo'))
|
|
|
|
class MainTest(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.utf8 = "<?xml version='1.0' encoding='UTF-8'?>\n"
|
|
self.tiny_manifest = '<manifest>\n</manifest>'
|
|
self.stderr = sys.stderr
|
|
sys.stderr = StringIO.StringIO()
|
|
|
|
def tearDown(self):
|
|
sys.stderr = self.stderr
|
|
|
|
def testNotEnoughArgs(self):
|
|
err_msg = 'Not enough arguments\n'
|
|
self.assertRaises(SystemExit, loman.main, ['loman'])
|
|
self.assertTrue(sys.stderr.getvalue().endswith(err_msg))
|
|
|
|
def testNotWorkon(self):
|
|
err_msg = 'Adding of non-workon projects is currently unsupported.\n'
|
|
self.assertRaises(SystemExit, loman.main, ['loman', 'add', 'foo', 'path'])
|
|
self.assertTrue(sys.stderr.getvalue().endswith(err_msg))
|
|
|
|
def testBadCommand(self):
|
|
err_msg = 'Unsupported command: bad.\n'
|
|
self.assertRaises(SystemExit, loman.main, ['loman', 'bad', 'foo', 'path'])
|
|
self.assertTrue(sys.stderr.getvalue().endswith(err_msg))
|
|
|
|
def testSimpleAdd(self):
|
|
temp = tempfile.NamedTemporaryFile('w')
|
|
print >> temp, '<manifest>\n</manifest>'
|
|
temp.flush()
|
|
os.fsync(temp.fileno())
|
|
loman.main(['loman', 'add', '--workon', '-f',
|
|
temp.name, 'foo', 'path/to/foo'])
|
|
self.assertEqual(
|
|
open(temp.name, 'r').read(),
|
|
self.utf8 + '<manifest>\n'
|
|
'<project name="foo" path="path/to/foo" workon="True" />\n'
|
|
'</manifest>\n')
|
|
|
|
def testAddDup(self):
|
|
temp = tempfile.NamedTemporaryFile('w')
|
|
print >> temp, '<manifest>\n</manifest>'
|
|
temp.flush()
|
|
os.fsync(temp.fileno())
|
|
loman.main(['loman', 'add', '--workon', '-f',
|
|
temp.name, 'foo', 'path/to/foo'])
|
|
self.assertRaises(SystemExit, loman.main,
|
|
['loman', 'add', '--workon', '-f',
|
|
temp.name, 'foo', 'path/to/foo'])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|