loman: Add support for adding elements from default.xml

bin/loman_unittest.py

 #!/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
@@ skipping 20 matching lines @@
   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 testGetProject(self):
+    ptree = loman.LocalManifest('<manifest>\n</manifest>')
+    ptree.Parse()
+    ptree.AddProject('foo', 'path/to/foo')
+    project = ptree.GetProject('foo')
+    self.assertEqual(project.attrib['name'], 'foo')
+
   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 testAddNewElement(self):
+    dtree = loman.LocalManifest('<manifest>\n</manifest>')
+    dtree.Parse()
+    dtree.AddProject('foo', 'path/to/foo')
+    ptree = loman.LocalManifest('<manifest>\n</manifest>')
+    ptree.Parse()
+    ptree.AddWorkonProjectElement(dtree.GetProject('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(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):
 
@@ skipping 15 matching lines @@
     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):
+    default = tempfile.NamedTemporaryFile('w')
+    print >> default, '<manifest>\n' \
+      '<project name="foo" path="path/to/foo" />\n' \
+      '</manifest>\n'
+    default.flush()
+    os.fsync(default.fileno())
     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'])
+    loman.main(['loman', 'add', '--workon', '-f', temp.name,
+                '-d', default.name, '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):
+  def testIgnoredPath(self):
+    default = tempfile.NamedTemporaryFile('w')
+    print >> default, '<manifest>\n' \
+      '<project name="foo" path="path/to/foo" />\n' \
+      '</manifest>\n'
+    default.flush()
+    os.fsync(default.fileno())
     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'])
-    loman.main(['loman', 'add', '--workon', '-f',
-                temp.name, 'foo', 'path/to/foo'])
+    loman.main(['loman', 'add', '--workon', '-f', temp.name,
+                '-d', default.name, 'foo'])
+
+  def testAddDup(self):
+    default = tempfile.NamedTemporaryFile('w')
+    print >> default, '<manifest>\n' \
+      '<project name="foo" path="path/to/foo" />\n' \
+      '</manifest>\n'
+    default.flush()
+    os.fsync(default.fileno())
+    temp = tempfile.NamedTemporaryFile('w')
+    print >> temp, '<manifest>\n' \
+      '<project name="foo" path="bad/path/to/foo" />\n' \
+      '</manifest>\n'
+    temp.flush()
+    os.fsync(temp.fileno())
     self.assertRaises(SystemExit, loman.main,
-                      ['loman', 'add', '--workon', '-f',
-                       temp.name, 'foo', 'path/foo'])
+                      ['loman', 'add', '--workon', '-f', temp.name,
+                       '-d', default.name, 'foo'])
+
+  def testAddNonexistant(self):
+    default = tempfile.NamedTemporaryFile('w')
+    print >> default, '<manifest>\n' \
+      '<project name="foo" path="path/to/foo" />\n' \
+      '</manifest>\n'
+    default.flush()
+    os.fsync(default.fileno())
+    temp = tempfile.NamedTemporaryFile('w')
+    print >> temp, '<manifest>\n</manifest>'
+    temp.flush()
+    os.fsync(temp.fileno())
+    self.assertRaises(SystemExit, loman.main,
+                      ['loman', 'add', '--workon', '-f', temp.name,
+                       '-d', default.name, 'bar'])
 
 
 if __name__ == '__main__':
   unittest.main()