def test_enable(self):
"""
Test enabling beacons
"""
comm1 = "Enabled beacons on minion."
event_returns = [{
"complete": True,
"tag": "/salt/minion/minion_beacon_enabled_complete",
"beacons": {
"enabled":
True,
"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.enable(), {
"comment": comm1,
"result": True
})
def test_enable(self):
'''
Test enabling beacons
'''
comm1 = 'Enabled beacons on minion.'
event_returns = [{
'complete': True,
'tag': '/salt/minion/minion_beacon_enabled_complete',
'beacons': {
'enabled':
True,
'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.enable(), {
'comment': comm1,
'result': True
})
def test_enable_beacon_module():
"""
Test enabling beacons
"""
comm1 = "Enabled beacons on minion."
event_returns = [{
"complete": True,
"tag": "/salt/minion/minion_beacon_enabled_complete",
"beacons": {
"enabled":
True,
"watch_salt_master": [
{
"processes": {
"salt-master": "stopped"
}
},
{
"beacon_module": "ps"
},
],
},
}]
mock = MagicMock(return_value=True)
with patch.dict(beacons.__salt__, {"event.fire": mock}):
with patch.object(SaltEvent, "get_event", side_effect=event_returns):
assert beacons.enable() == {"comment": comm1, "result": True}
