From e41c4073103c8df070843af54541127ee9f55981 Mon Sep 17 00:00:00 2001
From: Bryan Newbold <bnewbold@robocracy.org>
Date: Tue, 28 Jul 2020 11:34:09 -0700
Subject: coverage search: 'recent' endpoint test (minimal)

---
 python/tests/web_coverage.py | 33 ++++++++++++++++++++++++++++++++-
 1 file changed, 32 insertions(+), 1 deletion(-)

(limited to 'python/tests')

diff --git a/python/tests/web_coverage.py b/python/tests/web_coverage.py
index d446fba6..4407934f 100644
--- a/python/tests/web_coverage.py
+++ b/python/tests/web_coverage.py
@@ -1,6 +1,7 @@
 
 import json
 import pytest
+import datetime
 
 from fatcat_openapi_client.rest import ApiException
 from fixtures import *
@@ -123,18 +124,48 @@ def test_coverage_search(app, mocker):
         },
     }
 
+    # preservation by date histogram
+    today = str(datetime.date.today())
+    elastic_resp3 = {
+        'took': 294,
+        'timed_out': False,
+        '_shards': {'total': 5, 'successful': 5, 'skipped': 0, 'failed': 0},
+        'hits': {'total': 4327, 'max_score': 0.0, 'hits': []},
+        'aggregations': {
+            'date_preservation': {
+              'buckets': [
+                {'key': {'date': f'{today}T00:00.000Z', 'preservation': 'bright'}, 'doc_count': 444},
+                {'key': {'date': f'{today}T00:00.000Z', 'preservation': 'dark'}, 'doc_count': 111},
+              ],
+              'sum_other_doc_count': 0,
+            },
+        },
+    }
+
 
     es_raw = mocker.patch('elasticsearch.connection.Urllib3HttpConnection.perform_request')
     es_raw.side_effect = [
         # counts summary
         (200, {}, json.dumps(ES_CONTAINER_STATS_RESP)),
+        # by type
+        (200, {}, json.dumps(elastic_resp2)),
         # by year
         (200, {}, json.dumps(elastic_resp1)),
+    ]
+
+    rv = app.get('/coverage/search?q=*')
+    assert rv.status_code == 200
+
+    es_raw.side_effect = [
+        # counts summary
+        (200, {}, json.dumps(ES_CONTAINER_STATS_RESP)),
         # by type
         (200, {}, json.dumps(elastic_resp2)),
+        # by date
+        (200, {}, json.dumps(elastic_resp3)),
     ]
 
-    rv = app.get('/coverage/search?q=*')
+    rv = app.get('/coverage/search?recent=1&q=*')
     assert rv.status_code == 200
 
 
-- 
cgit v1.2.3