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={})