Rwa 4067 refactor integration tests

src/integrationTest/java/uk/gov/hmcts/reform/wataskmanagementapi/cft/query/ia/CftQueryServiceITTest.java

@@ -40,15 +40,20 @@
 import uk.gov.hmcts.reform.wataskmanagementapi.domain.search.SortingParameter;
 import uk.gov.hmcts.reform.wataskmanagementapi.domain.search.parameter.SearchParameterList;
 import uk.gov.hmcts.reform.wataskmanagementapi.domain.task.Task;
+import uk.gov.hmcts.reform.wataskmanagementapi.entity.TaskResource;
+import uk.gov.hmcts.reform.wataskmanagementapi.repository.TaskResourceRepository;
+import uk.gov.hmcts.reform.wataskmanagementapi.services.CFTTaskDatabaseService;
 import uk.gov.hmcts.reform.wataskmanagementapi.services.CFTTaskMapper;
 import uk.gov.hmcts.reform.wataskmanagementapi.services.CamundaService;
 
 import java.time.OffsetDateTime;
 import java.time.ZonedDateTime;
 import java.time.format.DateTimeFormatter;
 import java.util.ArrayList;
+import java.util.Collections;
 import java.util.List;
 import java.util.Map;
+import java.util.Optional;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
 
@@ -76,14 +81,21 @@ public class CftQueryServiceITTest extends RoleAssignmentHelper {
     @Autowired
     private AllowedJurisdictionConfiguration allowedJurisdictionConfiguration;
 
+    private CFTTaskDatabaseService cftTaskDatabaseService;
+
     private static final UserInfo userInfo = UserInfo.builder().email("[email protected]").uid("user").build();
 
+    @Autowired
+    TaskResourceRepository taskResourceRepository;
+
     @BeforeEach
     void setUp() {
         CFTTaskMapper cftTaskMapper = new CFTTaskMapper(new ObjectMapper());
         cftQueryService = new CftQueryService(camundaService, cftTaskMapper, new TaskResourceDao(entityManager),
             allowedJurisdictionConfiguration
         );
+
+        cftTaskDatabaseService = new CFTTaskDatabaseService(taskResourceRepository, cftTaskMapper);
     }
 
     @Disabled
@@ -111,9 +123,9 @@ void shouldRetrieveTasks(TaskQueryScenario scenario) {
         AccessControlResponse accessControlResponse = new AccessControlResponse(scenario.userInfo,
             scenario.roleAssignments);
         SearchRequest searchRequest = SearchTaskRequestMapper.map(scenario.searchTaskRequest);
-
+        indexRecord();
         //when
-        final GetTasksResponse<Task> allTasks = cftQueryService.searchForTasks(
+        final GetTasksResponse<Task> allTasks = cftTaskDatabaseService.searchForTasks(
             scenario.firstResult,
             scenario.maxResults,
             searchRequest,
@@ -818,6 +830,7 @@ private static List<RoleAssignment> roleAssignmentWithAllGrantTypes(Classificati
         List<RoleAssignment> roleAssignments = new ArrayList<>();
         RoleAssignment roleAssignment = RoleAssignment.builder().roleName("hmcts-judiciary")
             .classification(classification)
+            .attributes(Collections.emptyMap())
             .grantType(GrantType.SPECIFIC)
             .roleType(RoleType.ORGANISATION)
             .beginTime(OffsetDateTime.now().minusYears(1))
@@ -869,12 +882,9 @@ private static List<RoleAssignment> roleAssignmentWithAllGrantTypes(Classificati
             .build();
         roleAssignments.add(roleAssignment);
 
-        final Map<String, String> excludeddAttributes = Map.of(
-            RoleAttributeDefinition.CASE_ID.value(), "1623278362431003"
-        );
         roleAssignment = RoleAssignment.builder().roleName("senior-tribunal-caseworker")
             .classification(classification)
-            .attributes(excludeddAttributes)
+            .attributes(Collections.emptyMap())
             .grantType(GrantType.EXCLUDED)
             .roleType(RoleType.CASE)
             .beginTime(OffsetDateTime.now().minusYears(1))
@@ -891,6 +901,7 @@ private static List<RoleAssignment> defaultSort(Classification classification) {
             .classification(classification)
             .roleType(RoleType.ORGANISATION)
             .grantType(GrantType.SPECIFIC)
+            .attributes(Collections.emptyMap())
             .beginTime(OffsetDateTime.now().minusYears(1))
             .endTime(OffsetDateTime.now().plusYears(1))
             .build();
@@ -919,4 +930,15 @@ public String toString() {
         }
 
     }
+
+    private void indexRecord() {
+        List<String> ids = new ArrayList<>();
+        taskResourceRepository.findAll().forEach(taskResource -> ids.add(taskResource.getTaskId()));
+        ids.forEach(id -> {
+            Optional<TaskResource> taskResource = taskResourceRepository.findById(id);
+            TaskResource task = taskResource.get();
+            task.setIndexed(true);
+            taskResourceRepository.save(task);
+        });
+    }
 }