[PATCH] Create reboot unittest

[Paul W. Frields]

---
 tests/commands/reboot.py |   46 ++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 46 insertions(+), 0 deletions(-)
 create mode 100644 tests/commands/reboot.py

diff --git a/tests/commands/reboot.py b/tests/commands/reboot.py
new file mode 100644
index 0000000..1e20ca9
--- /dev/null
+++ b/tests/commands/reboot.py
@@ -0,0 +1,46 @@
+#
+# Paul W. Frields <pfrields at redhat.com>
+#
+# Copyright 2009 Red Hat, Inc.
+#
+# This copyrighted material is made available to anyone wishing to use, modify,
+# copy, or redistribute it subject to the terms and conditions of the GNU
+# General Public License v.2.  This program is distributed in the hope that it
+# will be useful, but WITHOUT ANY WARRANTY expressed or implied, including the
+# implied warranties of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
+# See the GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along with
+# this program; if not, write to the Free Software Foundation, Inc., 51
+# Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.  Any Red Hat
+# trademarks that are incorporated in the source code or documentation are not
+# subject to the GNU General Public License and may only be used or replicated
+# with the express permission of Red Hat, Inc. 
+#
+import unittest, shlex
+import warnings
+from tests.baseclass import *
+
+from pykickstart.errors import *
+from pykickstart.commands.reboot import *
+
+class FC3_TestCase(CommandTest):
+	def runTest(self):
+		# pass
+		self.assert_parse("reboot", "reboot\n")
+
+
+class FC6_TestCase(FC3_TestCase):
+	def runTest(self):
+		FC3_TestCase.runTest(self)
+	
+		# pass
+		self.assert_parse("reboot --eject", "reboot --eject\n")
+
+		# fail, bad option
+		self.assert_parse_error("reboot --idontexist")
+		# fail, no argument
+		self.assert_parse_error("reboot --eject=foo")
+
+if __name__ == "__main__":
+    unittest.main()
-- 
1.6.0.6