fix tests

Signed-off-by: Luka Peschke <luka.peschke@toucantoco.com>
ToucanToco · Jan 23, 2024 · 20ada48 · 20ada48
1 parent c72bfcd
commit 20ada48
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/tests/elasticsearch/test_elasticsearch.py b/tests/elasticsearch/test_elasticsearch.py
@@ -39,13 +39,10 @@ def pytest_generate_tests(metafunc):
 
 
 def test_connector(mocker):
-    class ElasticsearchMock:
-        def search(self, index, body):
-            return {'hits': {'hits': [{'_source': {'yo': 'la'}}]}}
 
     module = 'toucan_connectors.elasticsearch.elasticsearch_connector'
     mock_es = mocker.patch(f'{module}.Elasticsearch')
-    mock_es.return_value = ElasticsearchMock()
+    mock_es.return_value.search.return_value = {'hits': {'hits': [{'_source': {'yo': 'la'}}]}}
 
     con = ElasticsearchConnector(
         name='test',