#!/usr/bin/env python # ClusterShell.NodeSet.NodeSet error handling test suite # Written by S. Thiell """Unit test for RangeSet errors""" import sys import unittest sys.path.insert(0, '../lib') from ClusterShell.RangeSet import RangeSet, RangeSetND from ClusterShell.NodeSet import NodeSet from ClusterShell.NodeSet import NodeSetBase from ClusterShell.NodeSet import NodeSetError from ClusterShell.NodeSet import NodeSetParseError from ClusterShell.NodeSet import NodeSetParseRangeError class NodeSetErrorTest(unittest.TestCase): def _testNS(self, pattern, expected_exc): try: nodeset = NodeSet(pattern) print nodeset except NodeSetParseError, e: self.assertEqual(e.__class__, expected_exc) return except: raise self.assert_(0, "error not detected/no exception raised [pattern=%s]" % pattern) def testBadRangeUsages(self): """test NodeSet parse errors in range""" self._testNS("nova[]", NodeSetParseRangeError) self._testNS("nova[-]", NodeSetParseRangeError) self._testNS("nova[A]", NodeSetParseRangeError) self._testNS("nova[2-5/a]", NodeSetParseRangeError) self._testNS("nova[3/2]", NodeSetParseRangeError) self._testNS("nova[3-/2]", NodeSetParseRangeError) self._testNS("nova[-3/2]", NodeSetParseRangeError) self._testNS("nova[-/2]", NodeSetParseRangeError) self._testNS("nova[4-a/2]", NodeSetParseRangeError) self._testNS("nova[4-3/2]", NodeSetParseRangeError) self._testNS("nova[4-5/-2]", NodeSetParseRangeError) self._testNS("nova[4-2/-2]", NodeSetParseRangeError) self._testNS("nova[004-002]", NodeSetParseRangeError) self._testNS("nova[3-59/2,102a]", NodeSetParseRangeError) self._testNS("nova[3-59/2,,102]", NodeSetParseRangeError) self._testNS("nova%s" % ("3" * 101), NodeSetParseRangeError) # nD self._testNS("nova[]p0", NodeSetParseRangeError) self._testNS("nova[-]p0", NodeSetParseRangeError) self._testNS("nova[A]p0", NodeSetParseRangeError) self._testNS("nova[2-5/a]p0", NodeSetParseRangeError) self._testNS("nova[3/2]p0", NodeSetParseRangeError) self._testNS("nova[3-/2]p0", NodeSetParseRangeError) self._testNS("nova[-3/2]p0", NodeSetParseRangeError) self._testNS("nova[-/2]p0", NodeSetParseRangeError) self._testNS("nova[4-a/2]p0", NodeSetParseRangeError) self._testNS("nova[4-3/2]p0", NodeSetParseRangeError) self._testNS("nova[4-5/-2]p0", NodeSetParseRangeError) self._testNS("nova[4-2/-2]p0", NodeSetParseRangeError) self._testNS("nova[004-002]p0", NodeSetParseRangeError) self._testNS("nova[3-59/2,102a]p0", NodeSetParseRangeError) self._testNS("nova[3-59/2,,102]p0", NodeSetParseRangeError) self._testNS("nova%sp0" % ("3" * 101), NodeSetParseRangeError) self._testNS("x4nova[]p0", NodeSetParseRangeError) self._testNS("x4nova[-]p0", NodeSetParseRangeError) self._testNS("x4nova[A]p0", NodeSetParseRangeError) self._testNS("x4nova[2-5/a]p0", NodeSetParseRangeError) self._testNS("x4nova[3/2]p0", NodeSetParseRangeError) self._testNS("x4nova[3-/2]p0", NodeSetParseRangeError) self._testNS("x4nova[-3/2]p0", NodeSetParseRangeError) self._testNS("x4nova[-/2]p0", NodeSetParseRangeError) self._testNS("x4nova[4-a/2]p0", NodeSetParseRangeError) self._testNS("x4nova[4-3/2]p0", NodeSetParseRangeError) self._testNS("x4nova[4-5/-2]p0", NodeSetParseRangeError) self._testNS("x4nova[4-2/-2]p0", NodeSetParseRangeError) self._testNS("x4nova[004-002]p0", NodeSetParseRangeError) self._testNS("x4nova[3-59/2,102a]p0", NodeSetParseRangeError) self._testNS("x4nova[3-59/2,,102]p0", NodeSetParseRangeError) self._testNS("x4nova%sp0" % ("3" * 101), NodeSetParseRangeError) def testBadUsages(self): """test NodeSet other parse errors""" self._testNS("nova[3-59/2,102", NodeSetParseError) self._testNS("nova3,nova4,,nova6", NodeSetParseError) self._testNS("nova3,nova4,5,nova6", NodeSetParseError) self._testNS("nova3,nova4,[5-8],nova6", NodeSetParseError) self._testNS("nova6,", NodeSetParseError) self._testNS("nova6[", NodeSetParseError) self._testNS("nova6]", NodeSetParseError) self._testNS("n6[1-4]]", NodeSetParseError) # reopening bracket: no pfx/sfx between delimited ranges self._testNS("n[1-4]0[3-4]", NodeSetParseError) self._testNS("n6[1-4][3-4]", NodeSetParseError) self._testNS("n6[1-4]56[3-4]", NodeSetParseError) # illegal numerical bracket folding with /step syntax self._testNS("prod-0[01-06/2]0", NodeSetParseError) self._testNS("prod-0[1-7/2,9]0", NodeSetParseError) self._testNS("prod-0[1-5/2,7-9]0", NodeSetParseError) self._testNS("prod-00[1-6/2]0", NodeSetParseError) # and not NodeSetParseRangeError # nD more self._testNS("[1-30][4-9]", NodeSetParseError) self._testNS("[1-30][4-9]p", NodeSetParseError) self._testNS("x[1-30][4-9]p", NodeSetParseError) self._testNS("x[1-30]p4-9]", NodeSetParseError) self._testNS("xazer][1-30]p[4-9]", NodeSetParseError) self._testNS("xa[[zer[1-30]p[4-9]", NodeSetParseRangeError) # entirely numeric hostnames are not allowed self._testNS("[0-10]", NodeSetParseError) self._testNS("0[0-10]", NodeSetParseError) self._testNS("[0-10]0", NodeSetParseError) self._testNS("0[0-10]0", NodeSetParseError) def testTypeSanityCheck(self): """test NodeSet input type sanity check""" self.assertRaises(TypeError, NodeSet, dict()) self.assertRaises(TypeError, NodeSet, list()) self.assertRaises(ValueError, NodeSetBase, None, RangeSet("1-10")) def testRangeSetEntryMismatch(self): """test NodeSet RangeSet entry mismatch""" nodeset = NodeSet("toto%s") rangeset = RangeSet("5") self.assertRaises(NodeSetError, nodeset._add, "toto%%s", rangeset) def test_bad_slices(self): nodeset = NodeSet("cluster[1-30]c[1-2]") self.assertRaises(TypeError, nodeset.__getitem__, "zz") self.assertRaises(TypeError, nodeset.__getitem__, slice(1, 'foo')) def test_binary_bad_object_type(self): nodeset = NodeSet("cluster[1-30]c[1-2]") class Dummy: pass dummy = Dummy() self.assertRaises(TypeError, nodeset.add, dummy) def test_internal_mismatch(self): nodeset = NodeSet("cluster[1-30]c[1-2]") self.assertTrue("cluster%sc%s" in nodeset._patterns) nodeset._patterns["cluster%sc%s"] = RangeSetND([[1]]) self.assertRaises(NodeSetParseError, str, nodeset) nodeset._patterns["cluster%sc%s"] = RangeSetND([[1, 1]]) self.assertEqual(str(nodeset), "cluster1c1") nodeset._patterns["cluster%sc%s"] = RangeSetND([[1, 1, 1]]) self.assertRaises(NodeSetParseError, str, nodeset) def test_empty_operand(self): # right self.assertRaises(NodeSetParseError, NodeSet, "foo!") self.assertRaises(NodeSetParseError, NodeSet, "foo,") self.assertRaises(NodeSetParseError, NodeSet, "foo&") self.assertRaises(NodeSetParseError, NodeSet, "foo^") self.assertRaises(NodeSetParseError, NodeSet, "c[1-30]c[1-2]!") # left self.assertRaises(NodeSetParseError, NodeSet, "!foo") self.assertRaises(NodeSetParseError, NodeSet, ",foo") self.assertRaises(NodeSetParseError, NodeSet, "&foo") self.assertRaises(NodeSetParseError, NodeSet, "^foo") self.assertRaises(NodeSetParseError, NodeSet, "!c[1-30]c[1-2]") # other self.assertRaises(NodeSetParseError, NodeSet, "!") self.assertRaises(NodeSetParseError, NodeSet, ",") self.assertRaises(NodeSetParseError, NodeSet, "&") self.assertRaises(NodeSetParseError, NodeSet, "^") self.assertRaises(NodeSetParseError, NodeSet, ",,,") self.assertRaises(NodeSetParseError, NodeSet, "foo,,bar") def test_nd_fold_axis_errors(self): """test NodeSet fold_axis errors""" n1 = NodeSet("a3b2c0,a2b3c1,a2b4c1,a1b2c0,a1b2c1,a3b2c1,a2b5c1") n1.fold_axis = 0 self.assertRaises(NodeSetParseError, str, n1) n1.fold_axis = 1 self.assertRaises(NodeSetParseError, str, n1) n1.fold_axis = "0-1" # nok self.assertRaises(NodeSetParseError, str, n1) n1.fold_axis = range(2) # ok self.assertEqual(str(n1), "a[1,3]b2c0,a[1,3]b2c1,a2b[3-5]c1") n1.fold_axis = RangeSet("0-1") # ok self.assertEqual(str(n1), "a[1,3]b2c0,a[1,3]b2c1,a2b[3-5]c1")