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}