def test_delete(self):
'''
Test deleting a beacon.
'''
comm1 = 'Deleted beacon: ps.'
event_returns = [
{
'complete': True,
'tag': '/salt/minion/minion_beacons_delete_complete',
'beacons': {}
},
]
with patch.dict(
beacons.__opts__, {
'beacons': {
'ps': [{
'processes': {
'salt-master': 'stopped',
'apache2': 'stopped'
}
}]
},
'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.delete('ps'), {
'comment': comm1,
'result': True
})
def test_delete_beacon_module(self):
"""
Test deleting a beacon.
"""
comm1 = "Deleted beacon: watch_salt_master."
event_returns = [
{
"complete": True,
"tag": "/salt/minion/minion_beacons_delete_complete",
"beacons": {},
},
]
with patch.dict(
beacons.__opts__,
{
"beacons": {
"watch_salt_master": [
{"processes": {"salt-master": "stopped"}},
{"beacon_module": "ps"},
]
},
"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.delete("watch_salt_master"),
{"comment": comm1, "result": True},
)
def test_delete():
"""
Test deleting a beacon.
"""
comm1 = "Deleted beacon: ps."
event_returns = [
{
"complete": True,
"tag": "/salt/minion/minion_beacons_delete_complete",
"beacons": {},
},
]
with patch.dict(
beacons.__opts__,
{
"beacons": {
"ps": [{
"processes": {
"salt-master": "stopped",
"apache2": "stopped"
}
}]
},
},
):
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.delete("ps") == {
"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}
)
@skipIf(True, "SLOWTEST skip")
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_add_beacon_module(self):
'''
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'}]}}]
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('watch_salt_master', [{'processes': {'salt-master': 'stopped'}}, {'beacon_module': 'ps'}]),
{'comment': comm1, 'result': True})
def test_enable_beacon_module(self):
'''
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'}]}}]
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_delete_beacon_module(self):
'''
Test deleting a beacon.
'''
comm1 = 'Deleted beacon: watch_salt_master.'
event_returns = [
{'complete': True,
'tag': '/salt/minion/minion_beacons_delete_complete',
'beacons': {}},
]
with patch.dict(beacons.__opts__, {'beacons': {'watch_salt_master': [{'processes': {'salt-master': 'stopped'}}, {'beacon_module': 'ps'}]}, '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.delete('watch_salt_master'),
{'comment': comm1, 'result': True})
