def test_disable_no_motd(self):
"""puppet:Disable with no motd update parses correctly"""
# This is the same as the test_disable test
expected = ["puppet agent --disable --color=false"]
params = {
'hosts': ['testhost.example.com'],
'command': 'puppet',
'subcommand': 'Disable',
'motd': False
}
(update, cmd) = ppt._parse_Disable(params, self.app_logger)
self.assertEqual(cmd, expected)
def test_disable_custom_motd(self):
"""puppet:Disable with custom motd update parses correctly"""
# FIXME
#expected = ["""echo "custom message" >> /etc/motd && puppet agent --disable --color=false"""]
expected = ["""puppet agent --disable --color=false"""]
params = {
'hosts': ['testhost.example.com'],
'command': 'puppet',
'subcommand': 'Disable',
'motd': 'custom message'
}
(update, cmd) = ppt._parse_Disable(params, self.app_logger)
self.assertEqual(cmd, expected)
def test_disable_custom_motd_with_date(self):
"""puppet:Disable with custom motd with date update parses correctly"""
with mock.patch('replugin.funcworker.puppet.dt') as (
dt):
dt.now.return_value = NOW
# FIXME
#expected = ["""echo "custom message %s" >> /etc/motd && puppet agent --disable --color=false""" % NOW]
expected = ["""puppet agent --disable --color=false"""]
params = {
'hosts': ['testhost.example.com'],
'command': 'puppet',
'subcommand': 'Disable',
'motd': 'custom message %s'
}
(update, cmd) = ppt._parse_Disable(params, self.app_logger)
self.assertEqual(cmd, expected)