def test_get_library_not_initialized():
self = create_autospec(Arctic, mongo_host=sentinel.host)
self._library_cache = {}
with pytest.raises(LibraryNotFoundException) as e, patch("arctic.arctic.ArcticLibraryBinding", autospec=True) as ML:
ML.return_value.get_library_type.return_value = None
Arctic.get_library(self, sentinel.lib_name)
assert "Library %s was not correctly initialized in %s." % (sentinel.lib_name, self) in str(e)
def test_get_library_auth_issue():
self = create_autospec(Arctic, mongo_host=sentinel.host)
self._library_cache = {}
with pytest.raises(LibraryNotFoundException) as e, \
patch('arctic.arctic.ArcticLibraryBinding', autospec=True) as ML:
ML.return_value.get_library_type.side_effect = OperationFailure('database error: not authorized for query on arctic_marketdata.index.ARCTIC')
Arctic.get_library(self, sentinel.lib_name)
assert "Library %s was not correctly initialized in %s." % (sentinel.lib_name, self) in str(e)
def test_get_library_auth_issue():
self = create_autospec(Arctic, mongo_host=sentinel.host)
self._library_cache = {}
with pytest.raises(LibraryNotFoundException) as e, \
patch('arctic.arctic.ArcticLibraryBinding', autospec=True) as ML:
ML.return_value.get_library_type.side_effect = OperationFailure('database error: not authorized for query on arctic_marketdata.index.ARCTIC')
Arctic.get_library(self, sentinel.lib_name)
assert "Library %s was not correctly initialized in %s." % (sentinel.lib_name, self) in str(e)
def test_get_library_not_initialized():
self = create_autospec(Arctic, mongo_host=sentinel.host)
self._library_cache = {}
with pytest.raises(LibraryNotFoundException) as e, \
patch('arctic.arctic.ArcticLibraryBinding', autospec=True) as ML:
ML.return_value.get_library_type.return_value = None
Arctic.get_library(self, sentinel.lib_name)
assert "Library %s was not correctly initialized in %s." % (
sentinel.lib_name, self) in str(e)
def test_get_library_not_registered():
self = create_autospec(Arctic)
self._library_cache = {}
with pytest.raises(LibraryNotFoundException) as e, \
patch('arctic.arctic.ArcticLibraryBinding', autospec=True) as ML:
ML.return_value.get_library_type.return_value = sentinel.lib_type
Arctic.get_library(self, sentinel.lib_name)
assert ("Couldn't load LibraryType '%s' for '%s' (has the class been registered?)" %
(sentinel.lib_type, sentinel.lib_name)
)in str(e)
def test_get_library_not_registered():
self = create_autospec(Arctic)
self._library_cache = {}
with pytest.raises(LibraryNotFoundException) as e, \
patch('arctic.arctic.ArcticLibraryBinding', autospec=True) as ML:
ML.return_value.get_library_type.return_value = sentinel.lib_type
Arctic.get_library(self, sentinel.lib_name)
assert (
"Couldn't load LibraryType '%s' for '%s' (has the class been registered?)"
% (sentinel.lib_type, sentinel.lib_name)) in str(e)
def test_get_library():
self = create_autospec(Arctic)
self._library_cache = {}
library_type = Mock()
register_library_type(sentinel.lib_type, library_type)
with patch('arctic.arctic.ArcticLibraryBinding', autospec=True) as ML:
ML.return_value.get_library_type.return_value = sentinel.lib_type
library = Arctic.get_library(self, sentinel.lib_name)
del LIBRARY_TYPES[sentinel.lib_type]
assert ML.call_args_list == [call(self, sentinel.lib_name)]
assert library_type.call_args_list == [call(ML.return_value)]
assert library == library_type.return_value
def test_get_library():
self = create_autospec(Arctic)
self._library_cache = {}
library_type = Mock()
register_library_type(sentinel.lib_type, library_type)
with patch('arctic.arctic.ArcticLibraryBinding', autospec=True) as ML:
ML.return_value.get_library_type.return_value = sentinel.lib_type
library = Arctic.get_library(self, sentinel.lib_name)
del LIBRARY_TYPES[sentinel.lib_type]
assert ML.call_args_list == [call(self, sentinel.lib_name)]
assert library_type.call_args_list == [call(ML.return_value)]
assert library == library_type.return_value
