def __init__(self, index, model, messages, counter, size=500):
self._counter = counter
self._index = index
self._messages = messages
self._model = model
self._size = 500
if index not in settings.ELASTICSEARCH['INDEXES']:
overrides = None
else:
overrides = settings.ELASTICSEARCH['INDEXES'][index].get('FETCHERS')
self._fetcher = fetcher_for(model, overrides)
def __init__(self, index, model, messages, counter):
self._counter = counter
self._index = index
self._messages = messages
self._model = model
self._size = settings.ELASTICSEARCH['CHUNK_SIZE']
if index not in settings.ELASTICSEARCH['INDEXES']:
overrides = None
else:
overrides = settings.ELASTICSEARCH['INDEXES'][index].get(
'FETCHERS')
self._fetcher = fetcher_for(model, overrides)
def test_fetcher_not_found(self):
with pytest.raises(ValueError) as e:
fetchers.fetcher_for(models.AgentIdentifier)
assert e.value.args == (
'No fetcher exists for <class \'share.models.identifiers.AgentIdentifier\'>',
)
def test_fetcher_for(self, model, fetcher):
assert isinstance(fetchers.fetcher_for(model), fetcher)
def test_fetcher_not_found(self):
with pytest.raises(ValueError) as e:
fetchers.fetcher_for(models.AgentIdentifier)
assert e.value.args == ('No fetcher exists for <class \'share.models.identifiers.AgentIdentifier\'>', )
def test_fetcher_for(self, model, fetcher):
assert isinstance(fetchers.fetcher_for(model), fetcher)