def test_disable(self):
'''
Test disabling beacons
'''
comm1 = 'Disabled beacons on minion.'
event_returns = [{
'complete': True,
'tag': '/salt/minion/minion_beacons_disabled_complete',
'beacons': {
'enabled':
False,
'ps': [{
'processes': {
'salt-master': 'stopped',
'apache2': 'stopped'
}
}]
}
}]
with patch.dict(beacons.__opts__, {
'beacons': {},
'sock_dir': SOCK_DIR
}):
mock = MagicMock(return_value=True)
with patch.dict(beacons.__salt__, {'event.fire': mock}):
with patch.object(SaltEvent,
'get_event',
side_effect=event_returns):
self.assertDictEqual(beacons.disable(), {
'comment': comm1,
'result': True
})
def test_disable(self):
"""
Test disabling beacons
"""
comm1 = "Disabled beacons on minion."
event_returns = [{
"complete": True,
"tag": "/salt/minion/minion_beacons_disabled_complete",
"beacons": {
"enabled":
False,
"ps": [{
"processes": {
"salt-master": "stopped",
"apache2": "stopped"
}
}],
},
}]
with patch.dict(beacons.__opts__, {
"beacons": {},
"sock_dir": self.sock_dir
}):
mock = MagicMock(return_value=True)
with patch.dict(beacons.__salt__, {"event.fire": mock}):
with patch.object(SaltEvent,
"get_event",
side_effect=event_returns):
self.assertDictEqual(beacons.disable(), {
"comment": comm1,
"result": True
})
