speed up tests by saving 1 request

StabiBerlin · Jul 18, 2023 · f7ed8c9 · f7ed8c9
1 parent ab37221
commit f7ed8c9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/oai_status/list_records.py b/oai_status/list_records.py
@@ -80,6 +80,7 @@ def yield_records(doc: Document) -> Iterable[TagNode]:
     yield from yield_records(doc)
     if not counter.left:
         return
+
     while resumption_token := get_resumption_token(doc):
         doc = request_list_records(setSpec, metadata_prefix, resumption_token)
         yield from yield_records(doc)

diff --git a/tests/test_list_records.py b/tests/test_list_records.py
@@ -35,8 +35,9 @@ def test_list_records_limited():
 
 
 def test_list_records_limited_paging():
-    records = list(list_records('illustrierte.liedflugschriften', limit=125))
-    assert len(records) == 125
+    assert len(
+        list(list_records('illustrierte.liedflugschriften', limit=75))
+    ) == 75
 
 
 def test_entrypoint():