def test_select_raise_warning(self):
# Check that the warning gets called upon if we select more than
# one node, or not if we don't
args = ['select', 'cluster=appservers', 'set/pooled=yes']
original_raise_warning = tool.ToolCliByLabel.raise_warning
tool.ToolCliByLabel.raise_warning = mock.MagicMock()
with self.assertRaises(SystemExit) as cm:
tool.main(cmdline=args)
self.assertEqual(cm.exception_code, 0)
self.assertEqual(tool.ToolCliByLabel.raise_warning.call_count, 1)
# now let's loop through the responses from conftool get
args = ['select', 'cluster=appservers', 'get']
for res in self.output_for(args):
_log.debug(res)
del res['tags']
k = list(res.keys())[0]
self.assertEqual(res[k]['pooled'], 'yes')
tool.ToolCliByLabel.raise_warning.reset_mock()
args = ['select', 'name=mw1018', 'set/pooled=inactive']
with self.assertRaises(SystemExit) as cm:
tool.main(cmdline=args)
self.assertEqual(cm.exception_code, 0)
tool.ToolCliByLabel.raise_warning.assert_not_called()
out = self.output_for(['select', 'name=mw1018', 'get'])
self.assertEqual(out[0]['mw1018']['pooled'], 'inactive')
tool.ToolCliByLabel.raise_warning = original_raise_warning
def test_create_returns_error(self):
"""
Test creation is not possible from confctl
"""
args = self.generate_args('set/pooled=yes cp1059')
with self.assertRaises(SystemExit) as cm:
tool.main(cmdline=args)
# In error, we exit with status 1
self.assertEqual(cm.exception_code, 1)
# This doesn't get announced to irc.
self.assertEqual(self.irc.emit.called, False)
n = node.Node('eqiad', 'cache_text', 'https', 'cp1059')
self.assertFalse(n.exists)
def test_change_node_regexp(self):
"""
Changing values according to a regexp
"""
args = self.generate_args('set/pooled=yes re:cp105.')
with self.assertRaises(SystemExit) as cm:
tool.main(cmdline=args)
self.assertEqual(cm.exception_code, 0)
self.assertEqual(self.irc.emit.called, True)
for hostname in ['cp1052', 'cp1053', 'cp1054', 'cp1055']:
n = node.Node('eqiad', 'cache_text', 'https', hostname)
self.assertTrue(n.exists)
self.assertEqual(n.pooled, "yes")
def output_for(self, args):
with captured_output() as (out, err):
with self.assertRaises(SystemExit) as cm:
tool.main(cmdline=args)
self.assertEqual(cm.exception_code, 0)
res = out.getvalue().strip()
output = []
if not res:
return []
try:
for line in res.split("\n"):
output.append(json.loads(line))
except:
raise ValueError(res)
return output