Merge 89a5d58 into 20db8ad

beekeeper-api/src/main/java/com/expediagroup/beekeeper/api/response/MetadataResponseConverter.java

@@ -37,13 +37,18 @@ public final class MetadataResponseConverter {
   public static Page<HousekeepingMetadataResponse> convertToHousekeepingMetadataResponsePage(
       Page<HousekeepingMetadata> housekeepingMetadataPage) {
     List<HousekeepingMetadata> housekeepingMetadataList = housekeepingMetadataPage.getContent();
+    //Create a new ArrayList to store the HousekeepingMetadataResponse objects.
     List<HousekeepingMetadataResponse> housekeepingMetadataResponseList = new ArrayList<>();
+    //Iterate over the list of HousekeepingMetadata objects and convert each one to a HousekeepingMetadataResponse object.
     for (HousekeepingMetadata housekeepingMetadata : housekeepingMetadataList) {
-      HousekeepingMetadataResponse housekeepingMetadataResponse = convertToHousekeepingMetadataResponse(
-          housekeepingMetadata);
+      HousekeepingMetadataResponse housekeepingMetadataResponse = convertToHousekeepingMetadataResponse(housekeepingMetadata);
       housekeepingMetadataResponseList.add(housekeepingMetadataResponse);
     }
-    return new PageImpl<>(housekeepingMetadataResponseList);
+    //Create a new PageImpl<HousekeepingMetadataResponse> object using the HousekeepingMetadataResponse list, the Pageable
+    // object from the HousekeepingMetadataPage object, and the total number of elements from the HousekeepingMetadataPage object.
+    PageImpl<HousekeepingMetadataResponse> housekeepingMetadataResponses = new PageImpl<>(
+        housekeepingMetadataResponseList, housekeepingMetadataPage.getPageable(), housekeepingMetadataPage.getTotalElements());
+    return housekeepingMetadataResponses;
   }
 
   private static HousekeepingMetadataResponse convertToHousekeepingMetadataResponse(

beekeeper-api/src/main/java/com/expediagroup/beekeeper/api/response/PathResponseConverter.java

@@ -51,12 +51,16 @@ private static HousekeepingPathResponse convertToHousekeepingPathResponse(Housek
   public static Page<HousekeepingPathResponse> convertToHousekeepingPathResponsePage(
       Page<HousekeepingPath> housekeepingPathPage) {
     List<HousekeepingPath> housekeepingPathList = housekeepingPathPage.getContent();
+    //Create a new ArrayList to store the HousekeepingPathResponse objects.
     List<HousekeepingPathResponse> housekeepingPathResponseList = new ArrayList<>();
+    //Iterate over the list of HousekeepingPath objects and convert each one to a HousekeepingPathResponse object.
     for (HousekeepingPath housekeepingPath : housekeepingPathList) {
       HousekeepingPathResponse housekeepingPathResponse = convertToHousekeepingPathResponse(housekeepingPath);
       housekeepingPathResponseList.add(housekeepingPathResponse);
     }
-    return new PageImpl<>(housekeepingPathResponseList);
+    PageImpl<HousekeepingPathResponse> HousekeepingPathResponses = new PageImpl<>(
+        housekeepingPathResponseList, housekeepingPathPage.getPageable(), housekeepingPathPage.getTotalElements());
+    return HousekeepingPathResponses;
   }
 
 }

beekeeper-api/src/test/java/com/expediagroup/beekeeper/api/response/HousekeepingMetadataResponseTest.java

@@ -1,5 +1,5 @@
 /**
- * Copyright (C) 2019-2021 Expedia, Inc.
+ * Copyright (C) 2019-2023 Expedia, Inc.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -21,11 +21,13 @@
 import static com.expediagroup.beekeeper.api.util.DummyHousekeepingEntityGenerator.generateDummyHousekeepingMetadata;
 import static com.expediagroup.beekeeper.api.util.DummyHousekeepingEntityGenerator.generateDummyHousekeepingMetadataResponse;
 
+import java.util.ArrayList;
 import java.util.List;
 
 import org.junit.jupiter.api.Test;
 import org.springframework.data.domain.Page;
 import org.springframework.data.domain.PageImpl;
+import org.springframework.data.domain.PageRequest;
 
 import com.expediagroup.beekeeper.core.model.HousekeepingMetadata;
 
@@ -46,7 +48,27 @@ public void testConvertToHouseKeepingMetadataResponsePage() {
 
     assertThat(metadataResponsePageList.get(0)).isEqualTo(metadataResponse1);
     assertThat(metadataResponsePageList.get(1)).isEqualTo(metadataResponse2);
+    assertThat(metadataResponsePage.getTotalElements()).isEqualTo(2L);
+    assertThat(metadataResponsePage.getTotalPages()).isEqualTo(1L);
     assertThat(metadataResponsePage.getPageable()).isEqualTo((new PageImpl<>(housekeepingMetadataList).getPageable()));
   }
 
+  @Test
+  public void testConvertToHouseKeepingMetadataResponsePageWithMultiplePages() {
+    // Create a list of housekeeping metadata objects that is larger than the page size
+    List<HousekeepingMetadata> housekeepingMetadataList = new ArrayList<>();
+    for (int i = 0; i < 50; i++) {
+      housekeepingMetadataList.add(generateDummyHousekeepingMetadata("some_database" + i, "some_table" + i));
+    }
+
+    // Create a page of housekeeping metadata objects
+    Page<HousekeepingMetadata> metadataPage = new PageImpl<>(housekeepingMetadataList, PageRequest.of(0, 10), 50);
+
+    // Convert the page of housekeeping metadata objects to a page of housekeeping metadata response objects
+    Page<HousekeepingMetadataResponse> metadataResponsePage = convertToHousekeepingMetadataResponsePage(metadataPage);
+
+    // Assert that the housekeeping metadata response page has the correct total elements and total pages
+    assertThat(metadataResponsePage.getTotalElements()).isEqualTo(50L);
+    assertThat(metadataResponsePage.getTotalPages()).isEqualTo(5L);
+  }
 }

beekeeper-api/src/test/java/com/expediagroup/beekeeper/api/response/PathResponseConverterTest.java

@@ -0,0 +1,53 @@
+/**
+ * Copyright (C) 2019-2023 Expedia, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.expediagroup.beekeeper.api.response;
+
+import static org.assertj.core.api.AssertionsForClassTypes.assertThat;
+
+import static com.expediagroup.beekeeper.api.response.PathResponseConverter.convertToHousekeepingPathResponsePage;
+import static com.expediagroup.beekeeper.api.util.DummyHousekeepingEntityGenerator.generateDummyHousekeepingPath;
+import static com.expediagroup.beekeeper.api.util.DummyHousekeepingEntityGenerator.generateDummyHousekeepingPathResponse;
+
+import java.util.List;
+
+import org.junit.jupiter.api.Test;
+import org.springframework.data.domain.Page;
+import org.springframework.data.domain.PageImpl;
+
+import com.expediagroup.beekeeper.core.model.HousekeepingPath;
+
+public class PathResponseConverterTest {
+
+  @Test
+  public void testConvertToHousekeepingPathResponsePage() {
+    HousekeepingPath housekeepingPath1 = generateDummyHousekeepingPath("some_database1", "some_table1");
+    HousekeepingPath housekeepingPath2 = generateDummyHousekeepingPath("some_database2", "some_table2");
+    HousekeepingPathResponse housekeepingPathResponse1 = generateDummyHousekeepingPathResponse(housekeepingPath1);
+    HousekeepingPathResponse housekeepingPathResponse2 = generateDummyHousekeepingPathResponse(housekeepingPath2);
+
+    List<HousekeepingPath> housekeepingPathList = List.of(housekeepingPath1, housekeepingPath2);
+    Page<HousekeepingPathResponse> housekeepingPathResponsePage = convertToHousekeepingPathResponsePage(
+        new PageImpl<>(housekeepingPathList));
+
+    List<HousekeepingPathResponse> housekeepingPathResponsePageList = housekeepingPathResponsePage.getContent();
+
+    assertThat(housekeepingPathResponsePageList.get(0)).isEqualTo(housekeepingPathResponse1);
+    assertThat(housekeepingPathResponsePageList.get(1)).isEqualTo(housekeepingPathResponse2);
+    assertThat(housekeepingPathResponsePage.getTotalElements()).isEqualTo(2L);
+    assertThat(housekeepingPathResponsePage.getTotalPages()).isEqualTo(1L);
+    assertThat(housekeepingPathResponsePage.getPageable()).isEqualTo((new PageImpl<>(housekeepingPathList).getPageable()));
+  }
+}

beekeeper-api/src/test/java/com/expediagroup/beekeeper/api/util/DummyHousekeepingEntityGenerator.java

@@ -23,6 +23,7 @@
 import java.time.ZoneId;
 
 import com.expediagroup.beekeeper.api.response.HousekeepingMetadataResponse;
+import com.expediagroup.beekeeper.api.response.HousekeepingPathResponse;
 import com.expediagroup.beekeeper.core.model.HousekeepingMetadata;
 import com.expediagroup.beekeeper.core.model.HousekeepingPath;
 import com.expediagroup.beekeeper.core.model.PeriodDuration;
@@ -88,4 +89,19 @@ public static HousekeepingPath generateDummyHousekeepingPath(String databaseName
         .build();
   }
 
+  public static HousekeepingPathResponse generateDummyHousekeepingPathResponse(HousekeepingPath housekeepingPath) {
+    return HousekeepingPathResponse
+        .builder()
+        .path("s3://some/path/")
+        .databaseName(housekeepingPath.getDatabaseName())
+        .tableName(housekeepingPath.getTableName())
+        .housekeepingStatus(SCHEDULED)
+        .creationTimestamp(CREATION_TIMESTAMP)
+        .modifiedTimestamp(CREATION_TIMESTAMP)
+        .cleanupTimestamp(housekeepingPath.getCleanupTimestamp())
+        .cleanupDelay(CLEANUP_DELAY_STRING)
+        .cleanupAttempts(0)
+        .lifecycleType(EXPIRED.toString())
+        .build();
+  }
 }