def test_client_configuration(self, settings, patched_client, key, value,
settingkey):
settings[settingkey] = value
get_client(settings)
_, kwargs = patched_client.call_args
assert kwargs[key] == value
def test_ignores_aws_auth_when_settings_missing(self, settings,
patched_client,
aws_settings):
settings.update(aws_settings)
get_client(settings)
_, kwargs = patched_client.call_args
assert 'auth' not in kwargs
def test_initialises_aws_auth(self, settings, patched_aws_auth):
settings.update({
'es.aws.access_key_id': 'foo',
'es.aws.secret_access_key': 'bar',
'es.aws.region': 'baz'
})
get_client(settings)
patched_aws_auth.assert_called_once_with('foo', 'bar', 'baz', 'es')
def test_sets_aws_auth(self, settings, patched_client, patched_aws_auth):
settings.update({
'es.aws.access_key_id': 'foo',
'es.aws.secret_access_key': 'bar',
'es.aws.region': 'baz'
})
get_client(settings)
_, kwargs = patched_client.call_args
assert kwargs['http_auth'] == patched_aws_auth.return_value
def test_sets_connection_class_for_aws_auth(self, settings,
patched_client):
settings.update({
'es.aws.access_key_id': 'foo',
'es.aws.secret_access_key': 'bar',
'es.aws.region': 'baz'
})
get_client(settings)
_, kwargs = patched_client.call_args
assert kwargs['connection_class'] == RequestsHttpConnection
def includeme(config):
settings = config.registry.settings
settings.setdefault('es.host', 'http://localhost:9200')
settings.setdefault('es.index', 'hypothesis')
# Allow users of this module to register additional search filter and
# search matcher factories.
config.registry[FILTERS_KEY] = []
config.registry[MATCHERS_KEY] = []
config.add_directive(
'memex_add_search_filter',
lambda c, f: c.registry[FILTERS_KEY].append(config.maybe_dotted(f)))
config.add_directive('memex_get_search_filters',
lambda c: c.registry[FILTERS_KEY])
config.add_directive(
'memex_add_search_matcher',
lambda c, m: c.registry[MATCHERS_KEY].append(config.maybe_dotted(m)))
config.add_directive('memex_get_search_matchers',
lambda c: c.registry[MATCHERS_KEY])
# Add a property to all requests for easy access to the elasticsearch
# client. This can be used for direct or bulk access without having to
# reread the settings.
config.registry['es.client'] = get_client(settings)
config.add_request_method(lambda r: r.registry['es.client'],
name='es',
reify=True)
def test_initializes_client_with_index(self, settings, patched_client):
get_client(settings)
args, _ = patched_client.call_args
assert args[1] == 'my-index'
def test_initializes_client_with_host(self, settings, patched_client):
get_client(settings)
args, _ = patched_client.call_args
assert args[0] == 'search.svc'