From f6ad16cde9d2be2a53329cf3063750e93ed7a794 Mon Sep 17 00:00:00 2001
From: Prateek Keerthi <prateekkeerthi7@gmail.com>
Date: Thu, 7 Nov 2024 11:31:48 -0500
Subject: [PATCH 1/2] MAT-7830 remoed feature flag redundant code

---
 .../madiefhirservice/services/MeasureTranslatorService.java | 6 ------
 1 file changed, 6 deletions(-)

diff --git a/src/main/java/gov/cms/madie/madiefhirservice/services/MeasureTranslatorService.java b/src/main/java/gov/cms/madie/madiefhirservice/services/MeasureTranslatorService.java
index 4e5e6f5e..f2eaf1be 100644
--- a/src/main/java/gov/cms/madie/madiefhirservice/services/MeasureTranslatorService.java
+++ b/src/main/java/gov/cms/madie/madiefhirservice/services/MeasureTranslatorService.java
@@ -388,12 +388,6 @@ private List<MeasureGroupStratifierComponent> buildStratifications(Group madieGr
               .map(
                   strat -> {
                     List<PopulationType> associations = strat.getAssociations();
-                    if (!appConfigService.isFlagEnabled(MadieFeatureFlag.QiCore_STU4_UPDATES)) {
-                      associations = new ArrayList<>();
-                      if (strat.getAssociation() != null) {
-                        associations.add(strat.getAssociation());
-                      }
-                    }
                     MeasureGroupStratifierComponent stratComponent = null;
                     if (CollectionUtils.isNotEmpty(associations)) {
                       List<Extension> extensionList =

From ae1b1c97dc3970e70abda102b9b2ab9329c0a7d6 Mon Sep 17 00:00:00 2001
From: Prateek Keerthi <prateekkeerthi7@gmail.com>
Date: Thu, 7 Nov 2024 12:29:35 -0500
Subject: [PATCH 2/2] MAT-7830 fixed test cases

---
 .../services/MeasureTranslatorService.java    |  1 -
 .../MeasureTranslatorServiceTest.java         | 19 +++++++------------
 2 files changed, 7 insertions(+), 13 deletions(-)

diff --git a/src/main/java/gov/cms/madie/madiefhirservice/services/MeasureTranslatorService.java b/src/main/java/gov/cms/madie/madiefhirservice/services/MeasureTranslatorService.java
index f2eaf1be..39d1ae27 100644
--- a/src/main/java/gov/cms/madie/madiefhirservice/services/MeasureTranslatorService.java
+++ b/src/main/java/gov/cms/madie/madiefhirservice/services/MeasureTranslatorService.java
@@ -4,7 +4,6 @@
 import java.util.*;
 import java.util.concurrent.atomic.AtomicReference;
 import java.util.stream.Collectors;
-import gov.cms.madie.madiefhirservice.dto.MadieFeatureFlag;
 import gov.cms.madie.madiefhirservice.constants.IdentifierType;
 import gov.cms.madie.madiefhirservice.utils.FhirResourceHelpers;
 import gov.cms.madie.models.common.Organization;
diff --git a/src/test/java/gov/cms/madie/madiefhirservice/services/MeasureTranslatorServiceTest.java b/src/test/java/gov/cms/madie/madiefhirservice/services/MeasureTranslatorServiceTest.java
index 6ee9737e..13b80f07 100644
--- a/src/test/java/gov/cms/madie/madiefhirservice/services/MeasureTranslatorServiceTest.java
+++ b/src/test/java/gov/cms/madie/madiefhirservice/services/MeasureTranslatorServiceTest.java
@@ -13,11 +13,9 @@
 import static org.hamcrest.CoreMatchers.notNullValue;
 import static org.hamcrest.MatcherAssert.assertThat;
 import static org.junit.jupiter.api.Assertions.*;
-import static org.mockito.Mockito.when;
 
 import com.fasterxml.jackson.core.JsonProcessingException;
 import gov.cms.madie.madiefhirservice.constants.UriConstants;
-import gov.cms.madie.madiefhirservice.dto.MadieFeatureFlag;
 import gov.cms.madie.madiefhirservice.utils.FhirResourceHelpers;
 import gov.cms.madie.madiefhirservice.utils.MeasureTestHelper;
 import gov.cms.madie.madiefhirservice.utils.ResourceFileUtil;
@@ -771,11 +769,13 @@ public void testBuildFhirPopulationGroupsWithStratifications() {
     Stratification strat1 = new Stratification();
     strat1.setId("testStrat1Id");
     strat1.setDescription("strat-description");
-    strat1.setAssociation(PopulationType.INITIAL_POPULATION);
+    strat1.setAssociations(
+        List.of(PopulationType.INITIAL_POPULATION, PopulationType.MEASURE_POPULATION));
     stratifications.add(strat1);
     Stratification strat2 = new Stratification();
     strat2.setDescription("strat-description");
-    strat2.setAssociation(PopulationType.MEASURE_POPULATION);
+    strat1.setAssociations(
+        List.of(PopulationType.INITIAL_POPULATION, PopulationType.MEASURE_POPULATION));
     stratifications.add(strat2);
     group.setStratifications(stratifications);
     List<Group> groups = new ArrayList<>();
@@ -795,12 +795,9 @@ public void testBuildFhirPopulationGroupsWithStratifications() {
     assertThat(measureGroupStratifierComponent.getDescription(), is(equalTo("strat-description")));
     Expression expression = measureGroupStratifierComponent.getCriteria();
     assertThat(expression, is(notNullValue()));
-    assertThat(
-        measureGroupStratifierComponent.getExtensionByUrl(UriConstants.CqfMeasures.APPLIES_TO_URI),
-        is(notNullValue()));
-    Extension appliesToExt =
-        measureGroupStratifierComponent.getExtensionByUrl(UriConstants.CqfMeasures.APPLIES_TO_URI);
-    Type value = appliesToExt.getValue();
+    List<Extension> appliesToExt = measureGroupStratifierComponent.getExtension();
+    assertThat(appliesToExt.size(), is(2));
+    Type value = appliesToExt.get(0).getValue();
     CodeableConcept codeableConcept = value.castToCodeableConcept(value);
     assertThat(codeableConcept.getCoding(), is(notNullValue()));
     assertThat(codeableConcept.getCoding().size(), is(equalTo(1)));
@@ -815,7 +812,6 @@ public void testBuildFhirPopulationGroupsWithStratifications() {
 
   @Test
   public void testBuildFhirPopulationGroupsWithStratificationsOfMultipleAssociations() {
-    when(appConfigService.isFlagEnabled(MadieFeatureFlag.QiCore_STU4_UPDATES)).thenReturn(true);
 
     Population ip1 = new Population();
     ip1.setName(PopulationType.INITIAL_POPULATION);
@@ -886,7 +882,6 @@ public void testBuildFhirPopulationGroupsWithStratificationsOfMultipleAssociatio
 
   @Test
   public void testBuildFhirPopulationGroupsWithStratificationsOfNoAssociations() {
-    when(appConfigService.isFlagEnabled(MadieFeatureFlag.QiCore_STU4_UPDATES)).thenReturn(false);
 
     Population ip1 = new Population();
     ip1.setName(PopulationType.INITIAL_POPULATION);