Implement drawer integration cleanup API (RedHatInsights#3205)

Author: g-duval

backend/src/main/java/com/redhat/cloud/notifications/db/repositories/DrawerNotificationRepository.java

@@ -147,4 +147,12 @@ private String getOrderBy(Query.Sort sort) {
         }
     }
 
+    @Transactional
+    public void cleanupIntegrations(int limit) {
+        String deleteQuery = "delete from endpoints where id in (select id from endpoints where endpoint_type_v2 = 'DRAWER' " +
+            "and org_id is not null and not exists (select 1 from endpoint_event_type where endpoint_id = id) limit :limit)";
+        entityManager.createNativeQuery(deleteQuery)
+            .setParameter("limit", limit)
+            .executeUpdate();
+    }
 }

backend/src/main/java/com/redhat/cloud/notifications/routers/internal/cleanup/DrawerIntegrationCleanUpResource.java

@@ -0,0 +1,36 @@
+package com.redhat.cloud.notifications.routers.internal.cleanup;
+
+import com.redhat.cloud.notifications.auth.ConsoleIdentityProvider;
+import com.redhat.cloud.notifications.db.repositories.DrawerNotificationRepository;
+import com.redhat.cloud.notifications.models.Environment;
+import jakarta.annotation.security.RolesAllowed;
+import jakarta.inject.Inject;
+import jakarta.ws.rs.Consumes;
+import jakarta.ws.rs.POST;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
+
+import static com.redhat.cloud.notifications.Constants.API_INTERNAL;
+
+@RolesAllowed(ConsoleIdentityProvider.RBAC_INTERNAL_ADMIN)
+@Path(API_INTERNAL + "/drawer_cleanup")
+public class DrawerIntegrationCleanUpResource {
+
+    @Inject
+    DrawerNotificationRepository drawerNotificationRepository;
+
+    @Inject
+    Environment environment;
+
+    @POST
+    @Path("/")
+    @Consumes(MediaType.APPLICATION_JSON)
+    public Response cleanUp(int limit) {
+        if (environment.isStage() || environment.isLocal()) {
+            drawerNotificationRepository.cleanupIntegrations(limit);
+            return Response.ok().build();
+        }
+        return Response.status(Response.Status.FORBIDDEN).build();
+    }
+}

backend/src/test/java/com/redhat/cloud/notifications/routers/internal/cleanup/DrawerIntegrationCleanUpResourceTest.java

@@ -0,0 +1,78 @@
+package com.redhat.cloud.notifications.routers.internal.cleanup;
+
+import com.redhat.cloud.notifications.Constants;
+import com.redhat.cloud.notifications.TestHelpers;
+import com.redhat.cloud.notifications.TestLifecycleManager;
+import com.redhat.cloud.notifications.db.DbIsolatedTest;
+import com.redhat.cloud.notifications.db.ResourceHelpers;
+import com.redhat.cloud.notifications.models.Endpoint;
+import com.redhat.cloud.notifications.models.EndpointType;
+import io.quarkus.test.common.QuarkusTestResource;
+import io.quarkus.test.junit.QuarkusTest;
+import jakarta.inject.Inject;
+import jakarta.persistence.EntityManager;
+import org.eclipse.microprofile.config.inject.ConfigProperty;
+import org.junit.jupiter.api.Test;
+import java.util.List;
+
+import static io.restassured.RestAssured.given;
+import static io.restassured.http.ContentType.JSON;
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+@QuarkusTest
+@QuarkusTestResource(TestLifecycleManager.class)
+class DrawerIntegrationCleanUpResourceTest extends DbIsolatedTest {
+
+    @ConfigProperty(name = "internal.admin-role")
+    String adminRole;
+
+    @Inject
+    ResourceHelpers resourceHelpers;
+
+    @Inject
+    EntityManager entityManager;
+
+    @Test
+    void testDrawerIntegrationDelete() {
+
+        for (int i = 0; i < 15; i++) {
+            resourceHelpers.createEndpoint("123456", "123456", EndpointType.DRAWER);
+        }
+        String query = "SELECT e FROM Endpoint e WHERE e.orgId = :orgId";
+        List<Endpoint> endpoints = entityManager.createQuery(query, Endpoint.class)
+            .setParameter("orgId", 123456)
+            .getResultList();
+
+        assertEquals(15, endpoints.size());
+
+        given()
+                .basePath(Constants.API_INTERNAL)
+                .header(TestHelpers.createTurnpikeIdentityHeader("user", this.adminRole))
+                .when()
+                .contentType(JSON)
+                .body(10)
+                .post("/drawer_cleanup")
+                .then()
+                .statusCode(200);
+
+        endpoints = entityManager.createQuery(query, Endpoint.class)
+            .setParameter("orgId", 123456)
+            .getResultList();
+        assertEquals(5, endpoints.size());
+
+        given()
+            .basePath(Constants.API_INTERNAL)
+            .header(TestHelpers.createTurnpikeIdentityHeader("user", this.adminRole))
+            .when()
+            .contentType(JSON)
+            .body(10)
+            .post("/drawer_cleanup")
+            .then()
+            .statusCode(200);
+
+        endpoints = entityManager.createQuery(query, Endpoint.class)
+            .setParameter("orgId", 123456)
+            .getResultList();
+        assertEquals(0, endpoints.size());
+    }
+}