def test_company_dbmodels_to_es_documents(self, setup_es):
"""Tests conversion of db models to Elasticsearch documents."""
companies = CompanyFactory.create_batch(2)
result = ESCompany.db_objects_to_es_documents(companies)
assert len(list(result)) == len(companies)
def test_company_dbmodels_to_es_documents(self, es):
"""Tests conversion of db models to Elasticsearch documents."""
companies = CompanyFactory.create_batch(2)
app = get_search_app('company')
companies_qs = app.queryset.all()
result = ESCompany.db_objects_to_es_documents(companies_qs)
assert len(list(result)) == len(companies)