def test_get_integration(self):
"""Testing IntegrationManager.get_integration"""
manager = IntegrationManager(IntegrationConfig)
manager.register_integration_class(DummyIntegration1)
manager.register_integration_class(DummyIntegration2)
integration = manager.get_integration(DummyIntegration1.integration_id)
self.assertEqual(integration.__class__, DummyIntegration1)
integration = manager.get_integration(DummyIntegration2.integration_id)
self.assertEqual(integration.__class__, DummyIntegration2)
def test_get_integration(self):
"""Testing IntegrationManager.get_integration"""
manager = IntegrationManager(IntegrationConfig)
manager.register_integration_class(DummyIntegration1)
manager.register_integration_class(DummyIntegration2)
integration = manager.get_integration(DummyIntegration1.integration_id)
self.assertEqual(integration.__class__, DummyIntegration1)
integration = manager.get_integration(DummyIntegration2.integration_id)
self.assertEqual(integration.__class__, DummyIntegration2)
def test_get_integration_with_invalid_id(self):
"""Testing IntegrationManager.get_integration with invalid ID"""
manager = IntegrationManager(IntegrationConfig)
with self.assertRaises(IntegrationNotRegisteredError):
manager.get_integration('foobar')
def test_get_integration_with_invalid_id(self):
"""Testing IntegrationManager.get_integration with invalid ID"""
manager = IntegrationManager(IntegrationConfig)
with self.assertRaises(IntegrationNotRegisteredError):
manager.get_integration("foobar")
