def test_index_calls_index_or_create_method_with_correct_args(self):
bulk = self.ss.bulk_operation()
body = dict(key1='val1')
bulk._index_or_create = Mock()
bulk.index(index='test_bulk', doc_type='test_bulk_doc_type', body=body,
timeout=200)
self.assertTrue(bulk._index_or_create)
self.assertEquals(bulk._index_or_create.call_args[0][0], 'index')
assertDictEquals(bulk._index_or_create.call_args[0][1], body)
self.assertEquals(bulk._index_or_create.call_args[0][2], None)
self.assertEquals(bulk._index_or_create.call_args[1]['timeout'], 200)
self.assertEquals(bulk._index_or_create.call_args[1]['index'],
'test_bulk')
self.assertEquals(bulk._index_or_create.call_args[1]['doc_type'],
'test_bulk_doc_type')
def test_create_calls_index_or_create_method_with_correct_args(self):
bulk = self.ss.bulk_operation()
body = dict(key1='val1')
bulk._index_or_create = Mock()
bulk.create(doc_type='test_bulk_doc_type', body=body,
id=4, timeout=200, routing='abcd')
self.assertTrue(bulk._index_or_create)
self.assertEquals(bulk._index_or_create.call_args[0][0], 'create')
assertDictEquals(bulk._index_or_create.call_args[0][1], body)
self.assertEquals(bulk._index_or_create.call_args[0][2], 4)
self.assertEquals(bulk._index_or_create.call_args[1]['timeout'], 200)
self.assertEquals(bulk._index_or_create.call_args[1]['doc_type'],
'test_bulk_doc_type')
self.assertEquals(bulk._index_or_create.call_args[1]['routing'],
'abcd')
def test_update_must_push_correct_action(self):
bulk = self.ss.bulk_operation()
body = dict(key1='val1')
# Without params
bulk.update(id=123, body=body)
action = bulk._actions[-1]
self.assertEquals(action.type, 'update')
assertDictEquals(action.body, body)
assertDictEquals(action.params, dict(_id=123))
# With params
bulk.update(index='test_index', doc_type='test_doc_type', body=body,
id=123, consistency='sync', ttl=200)
action = bulk._actions[-1]
self.assertEquals(action.type, 'update')
assertDictEquals(action.body, body)
assertDictEquals(action.params, {
'_index': 'test_index',
'_type': 'test_doc_type',
'_id': 123,
'consistency': 'sync',
'ttl': '200'
})
def test_delete_must_push_correct_action(self):
bulk = self.ss.bulk_operation()
body = dict(key1='val1')
# Without params
bulk.delete(id=123)
action = bulk._actions[-1]
self.assertEquals(action.type, 'delete')
assertDictEquals(action.body, None)
assertDictEquals(action.params, dict(_id=123))
# With params
bulk.delete(index='test_index', doc_type='test_doc_type',
id=123, consistency='sync', parent=1)
action = bulk._actions[-1]
self.assertEquals(action.type, 'delete')
assertDictEquals(action.body, None)
assertDictEquals(action.params, {
'_index': 'test_index',
'_type': 'test_doc_type',
'_id': 123,
'consistency': 'sync',
'parent': '1',
})
# Make sure delete does not push body even if passed
bulk.delete(id=123, body=body)
action = bulk._actions[-1]
self.assertEquals(action.type, 'delete')
assertDictEquals(action.body, None)
assertDictEquals(action.params, dict(_id=123))
def test_index_or_create_must_push_correct_action(self):
bulk = self.ss.bulk_operation()
body = dict(key1='val1')
# Without params
bulk._index_or_create('index', body)
action = bulk._actions[-1]
self.assertEquals(action.type, 'index')
assertDictEquals(action.body, body)
assertDictEquals(action.params, {})
# With params
bulk._index_or_create('create', doc_type='test_doc_type', body=body,
id=1, consistency='sync', ttl=200)
action = bulk._actions[-1]
self.assertEquals(action.type, 'create')
assertDictEquals(action.body, body)
assertDictEquals(action.params, {
'_type': 'test_doc_type',
'_id': 1,
'consistency': 'sync',
'ttl': '200'
})
bulk._index_or_create('create', index='test_bulk',
doc_type='test_doc_type', body=body,
routing='abcd', refresh=True)
action = bulk._actions[-1]
self.assertEquals(action.type, 'create')
assertDictEquals(action.body, body)
assertDictEquals(action.params, {
'_index': 'test_bulk',
'_type': 'test_doc_type',
'routing': 'abcd',
'refresh': 'true',
})
