Revision 23269c5b
| b/qa/ganeti-qa.py | ||
|---|---|---|
| 136 | 136 |
if qa_config.TestEnabled('instance-info'):
|
| 137 | 137 |
RunTest(qa_instance.TestInstanceInfo, instance) |
| 138 | 138 |
|
| 139 |
if qa_config.TestEnabled('instance-automatic-restart'):
|
|
| 140 |
RunTest(qa_daemon.TestInstanceAutomaticRestart, node, instance) |
|
| 139 |
automatic_restart = \ |
|
| 140 |
qa_config.TestEnabled('instance-automatic-restart')
|
|
| 141 |
consecutive_failures = \ |
|
| 142 |
qa_config.TestEnabled('instance-consecutive-failures')
|
|
| 141 | 143 |
|
| 142 |
if qa_config.TestEnabled('instance-consecutive-failures'):
|
|
| 143 |
RunTest(qa_daemon.TestInstanceConsecutiveFailures, node, instance) |
|
| 144 |
if automatic_restart or consecutive_failures: |
|
| 145 |
qa_daemon.PrintCronWarning() |
|
| 146 |
|
|
| 147 |
if automatic_restart: |
|
| 148 |
RunTest(qa_daemon.TestInstanceAutomaticRestart, node, instance) |
|
| 149 |
|
|
| 150 |
if consecutive_failures: |
|
| 151 |
RunTest(qa_daemon.TestInstanceConsecutiveFailures, node, instance) |
|
| 144 | 152 |
|
| 145 | 153 |
if qa_config.TestEnabled('instance-export'):
|
| 146 | 154 |
expnode = qa_config.AcquireNode(exclude=node) |
| b/qa/qa_daemon.py | ||
|---|---|---|
| 78 | 78 |
utils.ShellQuoteArgs(cmd)).wait(), 0) |
| 79 | 79 |
|
| 80 | 80 |
|
| 81 |
def PrintCronWarning(): |
|
| 82 |
"""Shows a warning about the required cron job. |
|
| 83 |
|
|
| 84 |
""" |
|
| 85 |
|
|
| 86 |
qa_utils.PrintWarning("The following tests require the cron script for"
|
|
| 87 |
" ganeti-watcher to be set up.") |
|
| 88 |
|
|
| 89 |
|
|
| 81 | 90 |
def TestInstanceAutomaticRestart(node, instance): |
| 82 | 91 |
"""Test automatic restart of instance by ganeti-watcher. |
| 83 | 92 |
|
| b/qa/qa_utils.py | ||
|---|---|---|
| 21 | 21 |
""" |
| 22 | 22 |
|
| 23 | 23 |
import os |
| 24 |
import sys |
|
| 24 | 25 |
import subprocess |
| 25 | 26 |
|
| 26 | 27 |
from ganeti import utils |
| ... | ... | |
| 29 | 30 |
import qa_error |
| 30 | 31 |
|
| 31 | 32 |
|
| 33 |
_INFO_SEQ = None |
|
| 34 |
_WARNING_SEQ = None |
|
| 35 |
_ERROR_SEQ = None |
|
| 36 |
_RESET_SEQ = None |
|
| 37 |
|
|
| 38 |
|
|
| 39 |
def _SetupColours(): |
|
| 40 |
"""Initializes the colour constants. |
|
| 41 |
|
|
| 42 |
""" |
|
| 43 |
global _INFO_SEQ, _WARNING_SEQ, _ERROR_SEQ, _RESET_SEQ |
|
| 44 |
|
|
| 45 |
try: |
|
| 46 |
import curses |
|
| 47 |
except ImportError: |
|
| 48 |
# Don't use colours if curses module can't be imported |
|
| 49 |
return |
|
| 50 |
|
|
| 51 |
curses.setupterm() |
|
| 52 |
|
|
| 53 |
_RESET_SEQ = curses.tigetstr("op")
|
|
| 54 |
|
|
| 55 |
setaf = curses.tigetstr("setaf")
|
|
| 56 |
_INFO_SEQ = curses.tparm(setaf, curses.COLOR_GREEN) |
|
| 57 |
_WARNING_SEQ = curses.tparm(setaf, curses.COLOR_YELLOW) |
|
| 58 |
_ERROR_SEQ = curses.tparm(setaf, curses.COLOR_RED) |
|
| 59 |
|
|
| 60 |
|
|
| 61 |
_SetupColours() |
|
| 62 |
|
|
| 63 |
|
|
| 32 | 64 |
def AssertEqual(first, second, msg=None): |
| 33 | 65 |
"""Raises an error when values aren't equal. |
| 34 | 66 |
|
| ... | ... | |
| 113 | 145 |
AssertEqual(p.wait(), 0) |
| 114 | 146 |
|
| 115 | 147 |
return p.stdout.read().strip() |
| 148 |
|
|
| 149 |
|
|
| 150 |
def _PrintWithColor(text, seq): |
|
| 151 |
f = sys.stdout |
|
| 152 |
|
|
| 153 |
if not f.isatty(): |
|
| 154 |
seq = None |
|
| 155 |
|
|
| 156 |
if seq: |
|
| 157 |
f.write(seq) |
|
| 158 |
|
|
| 159 |
f.write(text) |
|
| 160 |
f.write("\n")
|
|
| 161 |
|
|
| 162 |
if seq: |
|
| 163 |
f.write(_RESET_SEQ) |
|
| 164 |
|
|
| 165 |
|
|
| 166 |
def PrintWarning(text): |
|
| 167 |
return _PrintWithColor(text, _WARNING_SEQ) |
|
| 168 |
|
|
| 169 |
|
|
| 170 |
def PrintError(f, text): |
|
| 171 |
return _PrintWithColor(text, _ERROR_SEQ) |
|
| 172 |
|
|
| 173 |
|
|
| 174 |
def PrintInfo(f, text): |
|
| 175 |
return _PrintWithColor(text, _INFO_SEQ) |
|
Also available in: Unified diff