def create_view(self, es_client):
"""Create and return test view class instance
Args:
es_client (Elasticsearch): elasticsearch connection client
Returns:
ElasticAPIView: test view instance
"""
view = ElasticAPIView()
view.es_client = es_client
view.es_model = DataDocType
view.es_filter_backends = (ElasticSearchFilter, ElasticFieldsFilter,
ElasticFieldsRangeFilter,
ElasticOrderingFilter)
return view
def test_get_es_client(self, es_data_client):
view = ElasticAPIView()
view.es_client = es_data_client
assert view.get_es_client() == es_data_client
def test_get_es_client_with_wrong_es_client(self):
view = ElasticAPIView()
view.es_client = str
with pytest.raises(ValueError):
view.get_es_client()