From ddf54c9670c6d4adb500e08342d831cf1214d641 Mon Sep 17 00:00:00 2001
From: Bryan Newbold <bnewbold@archive.org>
Date: Wed, 6 Apr 2022 17:40:30 -0700
Subject: polish RSS feed generation, and add basic test coverage

---
 tests/test_web.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

(limited to 'tests')

diff --git a/tests/test_web.py b/tests/test_web.py
index 3ffab87..31f7982 100644
--- a/tests/test_web.py
+++ b/tests/test_web.py
@@ -104,6 +104,28 @@ def test_basic_search(client: Any, mocker: Any) -> None:
     assert rv.status_code == 200
 
 
+def test_basic_rss_feed(client: Any, mocker: Any) -> None:
+
+    with open("tests/files/elastic_fulltext_search.json") as f:
+        elastic_resp = json.loads(f.read())
+
+    es_raw = mocker.patch(
+        "elasticsearch.connection.Urllib3HttpConnection.perform_request"
+    )
+    es_raw.side_effect = [
+        (200, {}, json.dumps(elastic_resp)),
+        (200, {}, json.dumps(elastic_resp)),
+    ]
+
+    rv = client.get("/feed/rss?q=blood")
+    assert rv.status_code == 200
+    assert rv.content.startswith(b"<rss")
+
+    rv = client.get("/zh/feed/rss?q=blood")
+    assert rv.status_code == 200
+    assert rv.content.startswith(b"<rss")
+
+
 def test_basic_work_landing_page(client: Any, mocker: Any) -> None:
 
     with open("tests/files/elastic_fulltext_get.json") as f:
-- 
cgit v1.2.3