def test_bulk_action_must_generate_valid_bulk_op_for_es(self):
body = dict(key1='val1')
action = _BulkAction('index', params={}, body=body)
self.assertEquals(action.es_op,
(json.dumps({ 'index': {} }) + '\n' +
json.dumps(body)))
action = _BulkAction('create', params=dict(routing='123', refresh=True),
body=body)
self.assertEquals(action.es_op,
(json.dumps({ 'create': dict(routing='123',
refresh=True) }) +
'\n' + json.dumps(body)))
# make sure that body is ignored when the operation does not require it
action = _BulkAction('delete', params=dict(routing='123', refresh=True),
body=body)
self.assertEquals(action.es_op,
(json.dumps({ 'delete': dict(routing='123',
refresh=True) })))
def test_bulk_action_must_throw_exception_when_missing_expected_body(self):
self.assertRaises(Exception, _BulkAction, 'index', params={})
_BulkAction('index', params={}, body=dict(key1='val1'))
def test_bulk_action_must_accept_valid_actions(self):
_BulkAction('index', params={}, body=dict(key1='val1'))
_BulkAction('create', params={}, body=dict(key1='val1'))
_BulkAction('update', params={}, body=dict(key1='val1'))
_BulkAction('delete', params={})
