diff --git a/tests/manager/api/controller/test_loan.py b/tests/manager/api/controller/test_loan.py index e9c69a821..50c04ef61 100644 --- 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 index b4ee11410..cf358e75b 100644 --- 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"]