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)
