Revision 8bb2df7d test/py/ganeti.cmdlib_unittest.py
b/test/py/ganeti.cmdlib_unittest.py | ||
---|---|---|
1533 | 1533 |
cmdlib._CheckOpportunisticLocking(op) |
1534 | 1534 |
|
1535 | 1535 |
|
1536 |
class _OpTestVerifyErrors(opcodes.OpCode): |
|
1537 |
OP_PARAMS = [ |
|
1538 |
opcodes._PDebugSimulateErrors, |
|
1539 |
opcodes._PErrorCodes, |
|
1540 |
opcodes._PIgnoreErrors, |
|
1541 |
] |
|
1542 |
|
|
1543 |
|
|
1544 |
class _LuTestVerifyErrors(cmdlib._VerifyErrors): |
|
1545 |
def __init__(self, **kwargs): |
|
1546 |
cmdlib._VerifyErrors.__init__(self) |
|
1547 |
self.op = _OpTestVerifyErrors(**kwargs) |
|
1548 |
self.op.Validate(True) |
|
1549 |
self.msglist = [] |
|
1550 |
self._feedback_fn = self.msglist.append |
|
1551 |
self.bad = False |
|
1552 |
|
|
1553 |
def DispatchCallError(self, which, *args, **kwargs): |
|
1554 |
if which: |
|
1555 |
self._Error(*args, **kwargs) |
|
1556 |
else: |
|
1557 |
self._ErrorIf(True, *args, **kwargs) |
|
1558 |
|
|
1559 |
def CallErrorIf(self, c, *args, **kwargs): |
|
1560 |
self._ErrorIf(c, *args, **kwargs) |
|
1561 |
|
|
1562 |
|
|
1563 |
class TestVerifyErrors(unittest.TestCase): |
|
1564 |
# Fake cluster-verify error code structures; we use two arbitary real error |
|
1565 |
# codes to pass validation of ignore_errors |
|
1566 |
(_, _ERR1ID, _) = constants.CV_ECLUSTERCFG |
|
1567 |
_NODESTR = "node" |
|
1568 |
_NODENAME = "mynode" |
|
1569 |
_ERR1CODE = (_NODESTR, _ERR1ID, "Error one") |
|
1570 |
(_, _ERR2ID, _) = constants.CV_ECLUSTERCERT |
|
1571 |
_INSTSTR = "instance" |
|
1572 |
_INSTNAME = "myinstance" |
|
1573 |
_ERR2CODE = (_INSTSTR, _ERR2ID, "Error two") |
|
1574 |
# Arguments used to call _Error() or _ErrorIf() |
|
1575 |
_ERR1ARGS = (_ERR1CODE, _NODENAME, "Error1 is %s", "an error") |
|
1576 |
_ERR2ARGS = (_ERR2CODE, _INSTNAME, "Error2 has no argument") |
|
1577 |
# Expected error messages |
|
1578 |
_ERR1MSG = _ERR1ARGS[2] % _ERR1ARGS[3] |
|
1579 |
_ERR2MSG = _ERR2ARGS[2] |
|
1580 |
|
|
1581 |
def testNoError(self): |
|
1582 |
lu = _LuTestVerifyErrors() |
|
1583 |
lu.CallErrorIf(False, self._ERR1CODE, *self._ERR1ARGS) |
|
1584 |
self.assertFalse(lu.bad) |
|
1585 |
self.assertFalse(lu.msglist) |
|
1586 |
|
|
1587 |
def _InitTest(self, **kwargs): |
|
1588 |
self.lu1 = _LuTestVerifyErrors(**kwargs) |
|
1589 |
self.lu2 = _LuTestVerifyErrors(**kwargs) |
|
1590 |
|
|
1591 |
def _CallError(self, *args, **kwargs): |
|
1592 |
# Check that _Error() and _ErrorIf() produce the same results |
|
1593 |
self.lu1.DispatchCallError(True, *args, **kwargs) |
|
1594 |
self.lu2.DispatchCallError(False, *args, **kwargs) |
|
1595 |
self.assertEqual(self.lu1.bad, self.lu2.bad) |
|
1596 |
self.assertEqual(self.lu1.msglist, self.lu2.msglist) |
|
1597 |
# Test-specific checks are made on one LU |
|
1598 |
return self.lu1 |
|
1599 |
|
|
1600 |
def _checkMsgCommon(self, logstr, errmsg, itype, item, warning): |
|
1601 |
self.assertTrue(errmsg in logstr) |
|
1602 |
if warning: |
|
1603 |
self.assertTrue("WARNING" in logstr) |
|
1604 |
else: |
|
1605 |
self.assertTrue("ERROR" in logstr) |
|
1606 |
self.assertTrue(itype in logstr) |
|
1607 |
self.assertTrue(item in logstr) |
|
1608 |
|
|
1609 |
def _checkMsg1(self, logstr, warning=False): |
|
1610 |
self._checkMsgCommon(logstr, self._ERR1MSG, self._NODESTR, |
|
1611 |
self._NODENAME, warning) |
|
1612 |
|
|
1613 |
def _checkMsg2(self, logstr, warning=False): |
|
1614 |
self._checkMsgCommon(logstr, self._ERR2MSG, self._INSTSTR, |
|
1615 |
self._INSTNAME, warning) |
|
1616 |
|
|
1617 |
def testPlain(self): |
|
1618 |
self._InitTest() |
|
1619 |
lu = self._CallError(*self._ERR1ARGS) |
|
1620 |
self.assertTrue(lu.bad) |
|
1621 |
self.assertEqual(len(lu.msglist), 1) |
|
1622 |
self._checkMsg1(lu.msglist[0]) |
|
1623 |
|
|
1624 |
def testMultiple(self): |
|
1625 |
self._InitTest() |
|
1626 |
self._CallError(*self._ERR1ARGS) |
|
1627 |
lu = self._CallError(*self._ERR2ARGS) |
|
1628 |
self.assertTrue(lu.bad) |
|
1629 |
self.assertEqual(len(lu.msglist), 2) |
|
1630 |
self._checkMsg1(lu.msglist[0]) |
|
1631 |
self._checkMsg2(lu.msglist[1]) |
|
1632 |
|
|
1633 |
def testIgnore(self): |
|
1634 |
self._InitTest(ignore_errors=[self._ERR1ID]) |
|
1635 |
lu = self._CallError(*self._ERR1ARGS) |
|
1636 |
self.assertFalse(lu.bad) |
|
1637 |
self.assertEqual(len(lu.msglist), 1) |
|
1638 |
self._checkMsg1(lu.msglist[0], warning=True) |
|
1639 |
|
|
1640 |
def testWarning(self): |
|
1641 |
self._InitTest() |
|
1642 |
lu = self._CallError(*self._ERR1ARGS, |
|
1643 |
code=_LuTestVerifyErrors.ETYPE_WARNING) |
|
1644 |
self.assertFalse(lu.bad) |
|
1645 |
self.assertEqual(len(lu.msglist), 1) |
|
1646 |
self._checkMsg1(lu.msglist[0], warning=True) |
|
1647 |
|
|
1648 |
def testWarning2(self): |
|
1649 |
self._InitTest() |
|
1650 |
self._CallError(*self._ERR1ARGS) |
|
1651 |
lu = self._CallError(*self._ERR2ARGS, |
|
1652 |
code=_LuTestVerifyErrors.ETYPE_WARNING) |
|
1653 |
self.assertTrue(lu.bad) |
|
1654 |
self.assertEqual(len(lu.msglist), 2) |
|
1655 |
self._checkMsg1(lu.msglist[0]) |
|
1656 |
self._checkMsg2(lu.msglist[1], warning=True) |
|
1657 |
|
|
1658 |
def testDebugSimulate(self): |
|
1659 |
lu = _LuTestVerifyErrors(debug_simulate_errors=True) |
|
1660 |
lu.CallErrorIf(False, *self._ERR1ARGS) |
|
1661 |
self.assertTrue(lu.bad) |
|
1662 |
self.assertEqual(len(lu.msglist), 1) |
|
1663 |
self._checkMsg1(lu.msglist[0]) |
|
1664 |
|
|
1665 |
def testErrCodes(self): |
|
1666 |
self._InitTest(error_codes=True) |
|
1667 |
lu = self._CallError(*self._ERR1ARGS) |
|
1668 |
self.assertTrue(lu.bad) |
|
1669 |
self.assertEqual(len(lu.msglist), 1) |
|
1670 |
self._checkMsg1(lu.msglist[0]) |
|
1671 |
self.assertTrue(self._ERR1ID in lu.msglist[0]) |
|
1672 |
|
|
1673 |
|
|
1536 | 1674 |
if __name__ == "__main__": |
1537 | 1675 |
testutils.GanetiTestProgram() |
Also available in: Unified diff