def test_resolve_elasticsearch_with_multiple_hosts(self, elasticsearch):
result = resolve()
self.assertIs(elasticsearch.return_value, result)
elasticsearch.assert_called_with('test-index',
hosts=[
'http://test-host1:9200/',
'http://test-host2:9200/',
'http://test-host3:8080/'
])
def test_resolve_elasticsearch_extended(self, elasticsearch, yaml, f_open):
yaml.safe_load.return_value = {
'elasticsearch': {
'index': 'test-index',
'client': {
'foo': 'bar',
'a': 1
}
}
}
result = resolve()
self.assertIs(elasticsearch.return_value, result)
f_open.assert_called_with('/etc/opt/test.yaml')
elasticsearch.assert_called_with('test-index', foo='bar', a=1)
def test_resolve_elasticsearch(self, elasticsearch):
result = resolve()
self.assertIs(elasticsearch.return_value, result)
elasticsearch.assert_called_with('test-index',
hosts=['http://test-host:9200/'])
def test_resolve_dynamodb(self, dynamodb):
result = resolve()
self.assertIs(dynamodb.return_value, result)
dynamodb.assert_called_with('test-table')
def test_resolve_s3_with_prefix(self, s3):
result = resolve()
self.assertIs(s3.return_value, result)
s3.assert_called_with('test-bucket', prefix='test/prefix')
def test_resolve_sqlite(self, sqlite):
result = resolve()
self.assertIs(sqlite.return_value, result)
sqlite.assert_called_with('test.db')