def test_mechanism_driver_by_agent_type_found(self):
self.agents = plugin.PlacementReporterAgents(ml2_plugin=self.plugin)
mech_driver = self.agents.mechanism_driver_by_agent_type(
'test_mechanism_driver_agent')
self.assertIsInstance(mech_driver, mechanism_test.TestMechanismDriver)
def test_mechanism_driver_by_agent_type_not_found(self):
self.agents = plugin.PlacementReporterAgents(ml2_plugin=self.plugin)
self.assertRaises(
Exception, # noqa
self.agents.mechanism_driver_by_agent_type,
'agent_not_belonging_to_any_mechanism_driver')
def test_supported_agent_types(self):
self.agents = plugin.PlacementReporterAgents(ml2_plugin=self.plugin)
self.assertEqual(
['test_mechanism_driver_agent'],
self.agents.supported_agent_types)