Align vcrpy usage with VCRTestCase

See https://vcrpy.readthedocs.io/en/latest/usage.html#unittest-integration
nsidc · Apr 9, 2024 · fae6b73 · fae6b73
1 parent 9669377
commit fae6b73
Show file tree

Hide file tree

Showing 11 changed files with 553 additions and 102 deletions.
diff --git a/earthaccess/search.py b/earthaccess/search.py
@@ -3,14 +3,14 @@
 
 import dateutil.parser as parser
 import requests
+
 from cmr import CollectionQuery, GranuleQuery
 
 from .auth import Auth
 from .daac import find_provider, find_provider_by_shortname
 from .results import DataCollection, DataGranule
 from .typing_ import (
     Any,
-    Dict,
     List,
     Never,
     Optional,
@@ -54,7 +54,7 @@ def get_results(
     while more_results:
         response = requests.get(url, headers=headers, params={"page_size": page_size})
 
-        if cmr_search_after := query.headers.get("cmr-search-after"):
+        if cmr_search_after := response.headers.get("cmr-search-after"):
             headers["cmr-search-after"] = cmr_search_after
 
         try:

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -110,6 +110,7 @@ module = [
   "pqdm.*",
   "s3fs",
   "tinynetrc.*",  # TODO: generate stubs for tinynetrc and remove this line
+  "vcr.unittest",  # TODO: generate stubs for vcr and remove this line
 ]
 ignore_missing_imports = true
 

diff --git a/...s/unit/fixtures/vcr_cassettes/PODAAC.yaml → ...esults.test_collections_less_than_2k.yaml b/...s/unit/fixtures/vcr_cassettes/PODAAC.yaml → ...esults.test_collections_less_than_2k.yaml
diff --git a/tests/unit/fixtures/vcr_cassettes/ALL.yaml → ...esults.test_collections_more_than_2k.yaml b/tests/unit/fixtures/vcr_cassettes/ALL.yaml → ...esults.test_collections_more_than_2k.yaml