Skip to content

Commit

Permalink
test: add a test for saved search kwargs in _do_search() method
Browse files Browse the repository at this point in the history
  • Loading branch information
anesson-cs committed Jul 11, 2024
1 parent c58a262 commit 654017c
Showing 1 changed file with 21 additions and 0 deletions.
21 changes: 21 additions & 0 deletions tests/units/test_core.py
Original file line number Diff line number Diff line change
Expand Up @@ -2394,6 +2394,27 @@ class DummyConfig:
self.assertIsNone(sr.number_matched)
self.assertEqual(len(sr), self.search_results_size)

@mock.patch("eodag.plugins.search.qssearch.QueryStringSearch", autospec=True)
def test__do_search_must_save_search_kwargs(self, search_plugin):
"""_do_search must return a "SearchResult" instance with its search kwargs"""
search_plugin.provider = "peps"
search_plugin.query.return_value = (
self.search_results.data,
None, # .query must return None if count is False
)

class DummyConfig:
pagination = {}

search_plugin.config = DummyConfig()

kwargs = {"page": 1, "items_per_page": 2, "cloud_cover": 80}
sr = self.dag._do_search(
search_plugin=search_plugin, count=False, raise_errors=False, **kwargs
)
self.assertIsNotNone(sr.search_kwargs)
self.assertDictEqual(sr.search_kwargs, kwargs)

@mock.patch("eodag.plugins.search.qssearch.QueryStringSearch", autospec=True)
def test__do_search_paginated_handle_no_count_returned(self, search_plugin):
"""_do_search must return None as count if provider does not return the count"""
Expand Down

0 comments on commit 654017c

Please sign in to comment.