From 8d5d266f1a429385f8692fef198955b137850a2b Mon Sep 17 00:00:00 2001 From: Jonathan Green Date: Thu, 14 Sep 2023 16:01:59 -0300 Subject: [PATCH] Use the constant --- tests/core/test_lane.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/tests/core/test_lane.py b/tests/core/test_lane.py index a5d21f929b..4293142983 100644 --- a/tests/core/test_lane.py +++ b/tests/core/test_lane.py @@ -50,7 +50,7 @@ tuple_to_numericrange, ) from core.model.collection import Collection -from core.model.configuration import ExternalIntegration +from core.model.configuration import ConfigurationAttributeValue, ExternalIntegration from core.problem_details import INVALID_INPUT from core.util.datetime_helpers import utc_now from core.util.opds_writer import OPDSFeed @@ -2645,7 +2645,11 @@ def test_worklist_for_resultset_no_holds_allowed( w1.license_pools[0].licenses_available = 0 collection1: Collection = w1.license_pools[0].collection integration1 = collection1.integration_configuration - integration1.settings_update({ExternalIntegration.DISPLAY_RESERVES: "no"}) + integration1.settings_update( + { + ExternalIntegration.DISPLAY_RESERVES: ConfigurationAttributeValue.NOVALUE.value + } + ) class MockHit: def __init__(self, work_id, has_last_update=False): @@ -2693,7 +2697,9 @@ def __contains__(self, k): # Now both collections are restricted and have no availability alternate_collection.integration_configuration.settings_update( - {ExternalIntegration.DISPLAY_RESERVES: "no"} + { + ExternalIntegration.DISPLAY_RESERVES: ConfigurationAttributeValue.NOVALUE.value + } ) assert [[w2], []] == m(db.session, [[hit2], [hit1]])