Fix broken test.

ThePalaceProject · Nov 4, 2024 · 9600bc1 · 9600bc1
1 parent a9a99a3
commit 9600bc1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tests/manager/api/controller/test_loan.py b/tests/manager/api/controller/test_loan.py
@@ -154,7 +154,7 @@ def verify_and_get_single_entry_feed_links(self, response):
         if self.expected_content_type == OPDSFeed.ENTRY_TYPE:
             feed = feedparser.parse(response.get_data())
             [entry] = feed["entries"]
-        elif self.expected_content_type == OPDS2Version1Serializer.CONTENT_TYPE:
+        elif self.expected_content_type == OPDS2Serializer.CONTENT_TYPE:
             entry = response.get_json()
         else:
             assert (

diff --git a/tests/manager/api/controller/test_work.py b/tests/manager/api/controller/test_work.py
@@ -673,7 +673,9 @@ def test_related_books(self, work_fixture: WorkFixture):
                 novelist_api=mock_api,
             )
         assert 200 == response.status_code
-        assert OPDSFeed.ACQUISITION_FEED_TYPE == response.headers["content-type"]
+        assert OPDSFeed.ACQUISITION_FEED_TYPE.startswith(
+            response.headers["content-type"]
+        )
         feed = feedparser.parse(response.data)
         assert "Related Books" == feed["feed"]["title"]