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 }