From 4b3923aab6dcbdd7d963cc61eabd66d77585f5ef Mon Sep 17 00:00:00 2001 From: Ben Papillon Date: Wed, 14 Oct 2026 17:10:43 -0700 Subject: [PATCH] chore(tests): read quantity_rates from the typed entitlement field --- .../com/schematic/api/datastream/EventQuantitiesTest.java | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/src/test/java/com/schematic/api/datastream/EventQuantitiesTest.java b/src/test/java/com/schematic/api/datastream/EventQuantitiesTest.java index 58d81618..b990f969 100644 --- a/src/test/java/com/schematic/api/datastream/EventQuantitiesTest.java +++ b/src/test/java/com/schematic/api/datastream/EventQuantitiesTest.java @@ -220,10 +220,7 @@ void aCompanyEntitlementsQuantityRatesReachTheResult() throws Exception { engine.checkFlag(inferenceFlag(0.5, null), MAPPER.treeToValue(company, RulesengineCompany.class), null); assertTrue(result.getEntitlement().isPresent()); - assertEquals( - MAPPER.valueToTree(RATES), - MAPPER.valueToTree( - result.getEntitlement().get().getAdditionalProperties().get("quantity_rates"))); + assertEquals(Optional.of(RATES), result.getEntitlement().get().getQuantityRates()); } /**