from elasticsearch import Elasticsearch
from elasticsearch.client.indices import IndicesClient
from elasticsearch.client.cluster import ClusterClient


def get_document(base_document):
    base_document['_version'] = 1
    base_document['found'] = True

    return base_document


class ElasticsearchMock(Elasticsearch):
    def __init__(self, *args, **kwargs):
        self.indices = IndicesClient(self)
        self.cluster = ClusterClient(self)