def test_get_manager_endpoint(self, mock_iter_entry_points):
loader = mock.Mock()
mock_iter_entry_points.return_value = [
mock.Mock(**{"load.return_value": loader}),
]
manager.get_manager('test', ('127.0.0.1', 8080))
mock_iter_entry_points.assert_called_once_with('tendril.manager',
'test')
for m in mock_iter_entry_points.return_value:
m.load.assert_called_once_with()
loader.assert_called_once_with(('127.0.0.1', 8080))
def test_get_manager(self, mock_iter_entry_points):
loader = mock.Mock()
mock_iter_entry_points.return_value = [
mock.Mock(**{"load.side_effect": ImportError}),
mock.Mock(**{"load.side_effect": pkg_resources.UnknownExtra}),
mock.Mock(**{"load.return_value": loader}),
]
manager.get_manager('test')
mock_iter_entry_points.assert_called_once_with('tendril.manager',
'test')
for m in mock_iter_entry_points.return_value:
m.load.assert_called_once_with()
loader.assert_called_once_with(('', 0))
def test_get_manager_existing(self, mock_iter_entry_points):
mock_manager = mock.Mock()
manager.TendrilManager._managers[('test', ('', 0))] = mock_manager
result = manager.get_manager('TEST')
self.assertEqual(id(result), id(mock_manager))
self.assertFalse(mock_iter_entry_points.called)
