burnin: Add unittest for supported disk templates

When new a new disk template is added, burnin should be updated.

Signed-off-by: Michael Hanselmann <hansmi@google.com>
Reviewed-by: Iustin Pop <iustin@google.com>

diff --git a/Makefile.am b/Makefile.am
index 7363ead..4150e8d 100644 (file)
--- a/Makefile.am
+++ b/Makefile.am
@@ -1090,6 +1090,7 @@ python_tests = \
        test/py/ganeti.ssconf_unittest.py \
        test/py/ganeti.ssh_unittest.py \
        test/py/ganeti.storage_unittest.py \
+       test/py/ganeti.tools.burnin_unittest.py \
        test/py/ganeti.tools.ensure_dirs_unittest.py \
        test/py/ganeti.tools.node_daemon_setup_unittest.py \
        test/py/ganeti.tools.prepare_node_join_unittest.py \

diff --git a/test/py/ganeti.tools.burnin_unittest.py b/test/py/ganeti.tools.burnin_unittest.py
new file mode 100755 (executable)
index 0000000..2f37d9e
--- /dev/null
+++ b/test/py/ganeti.tools.burnin_unittest.py
@@ -0,0 +1,43 @@
+#!/usr/bin/python
+#
+
+# Copyright (C) 2013 Google Inc.
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty 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.
+
+
+"""Script for testing ganeti.tools.burnin"""
+
+import unittest
+
+from ganeti import constants
+from ganeti.tools import burnin
+
+import testutils
+
+
+class TestConstants(unittest.TestCase):
+  def testSupportedDiskTemplates(self):
+    # Ignore disk templates not supported by burnin
+    supported = (constants.DISK_TEMPLATES - frozenset([
+      constants.DT_BLOCK,
+      ]))
+
+    self.assertEqual(burnin._SUPPORTED_DISK_TEMPLATES, supported)
+
+
+if __name__ == "__main__":
+  testutils.GanetiTestProgram()