def test_add(self):
"""
Test adding a beacon
"""
comm1 = "Added beacon: ps."
event_returns = [
{
"complete": True,
"tag": "/salt/minion/minion_beacons_list_complete",
"beacons": {},
},
{
"complete": True,
"tag": "/salt/minion/minion_beacons_list_available_complete",
"beacons": ["ps"],
},
{
"complete": True,
"valid": True,
"vcomment": "",
"tag": "/salt/minion/minion_beacons_list_complete",
},
{
"complete": True,
"tag": "/salt/minion/minion_beacon_add_complete",
"beacons": {
"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.add(
"ps",
[{
"processes": {
"salt-master": "stopped",
"apache2": "stopped",
}
}],
),
{
"comment": comm1,
"result": True
},
)
def test_add(self):
'''
Test adding a beacon
'''
comm1 = 'Added beacon: ps.'
event_returns = [{'complete': True,
'tag': '/salt/minion/minion_beacons_list_complete',
'beacons': {}},
{'complete': True,
'tag': '/salt/minion/minion_beacons_list_available_complete',
'beacons': ['ps']},
{'complete': True,
'valid': True,
'vcomment': '',
'tag': '/salt/minion/minion_beacons_list_complete'},
{'complete': True,
'tag': '/salt/minion/minion_beacon_add_complete',
'beacons': {'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.add('ps', [{'processes': {'salt-master': 'stopped', 'apache2': 'stopped'}}]),
{'comment': comm1, 'result': True})
def test_add_beacon_module():
"""
Test adding a beacon
"""
comm1 = "Added beacon: watch_salt_master."
event_returns = [
{
"complete": True,
"tag": "/salt/minion/minion_beacons_list_complete",
"beacons": {},
},
{
"complete": True,
"tag": "/salt/minion/minion_beacons_list_available_complete",
"beacons": ["ps"],
},
{
"complete": True,
"valid": True,
"vcomment": "",
"tag": "/salt/minion/minion_beacons_list_complete",
},
{
"complete": True,
"tag": "/salt/minion/minion_beacon_add_complete",
"beacons": {
"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.add(
"watch_salt_master",
[
{
"processes": {
"salt-master": "stopped"
}
},
{
"beacon_module": "ps"
},
],
) == {
"comment": comm1,
"result": True
}
