def setUp(self): super(TestAgent, self).setUp() if hasattr(yamllib, 'CSafeLoader'): self.yaml_loader = yamllib.CSafeLoader else: self.yaml_loader = yamllib.SafeLoader self.override_config('disable_murano_agent', False, group='engine') mock_host = mock.MagicMock() mock_host.id = '1234' mock_host.find_owner = lambda *args, **kwargs: mock_host mock_host().getRegion.return_value = mock.MagicMock( __class__=dsl.MuranoObjectInterface) self.rabbit_mq_settings = { 'agentRabbitMq': {'login': '******', 'password': '******', 'host': 'test_host', 'port': 123, 'virtual_host': 'test_virtual_host'} } mock_host().getRegion()().getConfig.return_value =\ self.rabbit_mq_settings self.agent = agent.Agent(mock_host) self.resources = mock.Mock(spec=resource_manager.ResourceManager) self.resources.string.return_value = 'text' self.addCleanup(mock.patch.stopall)
def test_agent_disabled(self): self.override_config('disable_murano_agent', True, 'engine') a = agent.Agent(mock.MagicMock()) self.assertFalse(a.enabled) self.assertRaises(exc.PolicyViolationException, a.call, {}, None) self.assertRaises(exc.PolicyViolationException, a.send, {}, None) self.assertRaises(exc.PolicyViolationException, a.call_raw, {}) self.assertRaises(exc.PolicyViolationException, a.send_raw, {})
def test_agent_enabled(self): self.override_config('disable_murano_agent', False, 'engine') agent_cls = 'murano.engine.system.agent.Agent' a = agent.Agent(mock.MagicMock()) self.assertTrue(a.enabled) with mock.patch(agent_cls + '._send') as s: s.return_value = mock.MagicMock() a.send_raw({}) s.assert_called_with({}, False, 0)
def setUp(self): super(TestExecutionPlan, self).setUp() if hasattr(yamllib, 'CSafeLoader'): self.yaml_loader = yamllib.CSafeLoader else: self.yaml_loader = yamllib.SafeLoader self.mock_murano_class = mock.Mock(spec=murano_class.MuranoClass) self.mock_murano_class.name = 'io.murano.system.Agent' self.mock_murano_class.parents = [] self.mock_object_store = mock.Mock(spec=object_store.ObjectStore) self.agent = agent.Agent(self.mock_murano_class, None, self.mock_object_store, None) self.resources = mock.Mock(spec=resource.ResourceManager) self.resources.string.return_value = 'text' self.uuids = ['ID1', 'ID2', 'ID3', 'ID4'] self.mock_uuid = self._stub_uuid(self.uuids) self.addCleanup(mock.patch.stopall)
def setUp(self): super(TestExecutionPlan, self).setUp() if hasattr(yamllib, 'CSafeLoader'): self.yaml_loader = yamllib.CSafeLoader else: self.yaml_loader = yamllib.SafeLoader self.mock_murano_class = mock.Mock(spec=murano_type.MuranoClass) self.mock_murano_class.name = 'io.murano.system.Agent' self.mock_murano_class.declared_parents = [] self.mock_object_store = mock.Mock(spec=object_store.ObjectStore) object_interface = mock.Mock(spec=murano_object.MuranoObject) object_interface.id = '1234' object_interface.find_owner = lambda *args, **kwargs: object_interface self.agent = agent.Agent(object_interface) self.resources = mock.Mock(spec=resource_manager.ResourceManager) self.resources.string.return_value = 'text' self.uuids = ['ID1', 'ID2', 'ID3', 'ID4'] self.mock_uuid = self._stub_uuid(self.uuids) self.addCleanup(mock.patch.stopall)