ACS-6151 - Create ATS liveness/readiness probes tests

engines/aio/src/test/java/org/alfresco/transform/aio/AIOLivenessReadinessProbeTest.java

@@ -0,0 +1,37 @@
+/*
+ * #%L
+ * Alfresco Transform Core
+ * %%
+ * Copyright (C) 2005 - 2023 Alfresco Software Limited
+ * %%
+ * This file is part of the Alfresco software.
+ * -
+ * If the software was purchased under a paid Alfresco license, the terms of
+ * the paid license agreement will prevail.  Otherwise, the software is
+ * provided under the following open source license terms:
+ * -
+ * Alfresco is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ * -
+ * Alfresco is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Lesser General Public License for more details.
+ * -
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with Alfresco. If not, see <http://www.gnu.org/licenses/>.
+ * #L%
+ */
+
+package org.alfresco.transform.aio;
+
+import org.alfresco.transform.base.LivenessReadinessProbeTest;
+
+public class AIOLivenessReadinessProbeTest extends LivenessReadinessProbeTest {
+    @Override
+    protected LivenessReadinessProbeTest.ImagesForTests getImageForTest() {
+        return new ImagesForTests("alfresco-transform-core-aio", "text/plain", "text/plain", "original.txt");
+    }
+}

engines/base/pom.xml

@@ -97,6 +97,19 @@
             <version>4.2.0</version>
             <scope>test</scope>
         </dependency>
+
+        <dependency>
+            <groupId>org.testcontainers</groupId>
+            <artifactId>testcontainers</artifactId>
+            <version>1.19.1</version>
+            <scope>test</scope>
+        </dependency>
+
+        <dependency>
+            <groupId>org.testcontainers</groupId>
+            <artifactId>junit-jupiter</artifactId>
+            <version>1.19.1</version>
+        </dependency>
     </dependencies>
 
     <build>

engines/base/src/test/java/org/alfresco/transform/base/LivenessReadinessProbeTest.java

@@ -0,0 +1,107 @@
+package org.alfresco.transform.base;
+
+import org.junit.jupiter.api.Test;
+
+import org.springframework.core.io.ClassPathResource;
+
+import org.springframework.http.client.MultipartBodyBuilder;
+import org.springframework.test.web.reactive.server.WebTestClient;
+import org.testcontainers.containers.GenericContainer;
+import org.testcontainers.containers.wait.strategy.Wait;
+
+import java.net.URISyntaxException;
+
+import java.util.Objects;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+import static org.springframework.http.HttpStatus.OK;
+import static org.springframework.http.HttpStatus.TOO_MANY_REQUESTS;
+
+import org.springframework.web.reactive.function.client.WebClient;
+
+public abstract class LivenessReadinessProbeTest
+{
+    @Test
+    public void readinessShouldReturnAn429ErrorAfterReachingMaxTransforms() throws URISyntaxException
+    {
+        final ImagesForTests testData = getImageForTest();
+
+        try (final var env = createEnv(testData.image))
+        {
+            env.start();
+            var url = "http://localhost:" + env.getFirstMappedPort();
+
+            int max_transforms = 11;
+            for (int i = 0; i<max_transforms; i++) {
+                sendTransformRequest(url, testData.sourceMimetype, testData.targetMimetype, testData.filename);
+            }
+
+            assertProbeDied(url);
+
+            final String logs = env.getLogs();
+            System.out.println(logs);
+        }
+    }
+
+    protected abstract ImagesForTests getImageForTest();
+
+    private GenericContainer<?> createEnv(String image) throws URISyntaxException
+    {
+        System.out.println(image);
+        final GenericContainer<?> transformCore = new GenericContainer<>("alfresco/"+image+":latest");
+
+        return transformCore.withEnv("livenessTransformEnabled", "true")
+            .withEnv("maxTransforms", "10")
+            .withNetworkAliases(image)
+            .withExposedPorts(8090)
+            .waitingFor(Wait.forListeningPort());
+    }
+
+    protected static class ImagesForTests
+    {
+        private final String image;
+        private final String sourceMimetype;
+        private final String targetMimetype;
+        private final String filename;
+
+        public ImagesForTests(String image, String sourceMimetype, String targetMimetype, String filename)
+        {
+            this.image = Objects.requireNonNull(image);
+            this.sourceMimetype = Objects.requireNonNull(sourceMimetype);
+            this.targetMimetype = Objects.requireNonNull(targetMimetype);
+            this.filename = Objects.requireNonNull(filename);
+        }
+    }
+
+    private void sendTransformRequest(String url, String sourceMimetype, String targetMimetype, String filename)
+    {
+        var builder = createRequestBuilder(sourceMimetype, targetMimetype, filename);
+        WebClient client = WebClient.create();
+        WebClient.ResponseSpec responseSpec = client.post()
+                .uri(url + "/test")
+                .bodyValue(builder.build())
+                .retrieve();
+
+        System.out.println(responseSpec.toBodilessEntity().block());
+        assertEquals(OK, responseSpec.toBodilessEntity().block().getStatusCode());
+    }
+
+    private MultipartBodyBuilder createRequestBuilder(String sourceMimetype, String targetMimetype, String filename)
+    {
+        MultipartBodyBuilder builder = new MultipartBodyBuilder();
+        builder.part("_sourceMimetype", sourceMimetype);
+        builder.part("_targetMimetype", targetMimetype);
+        builder.part("file", new ClassPathResource(filename));
+
+        return builder;
+    }
+
+    private static void assertProbeDied(String url)
+    {
+        WebTestClient client = WebTestClient.bindToServer().baseUrl(url+"/ready").build();
+        client.get()
+                .exchange()
+                .expectStatus().isEqualTo(TOO_MANY_REQUESTS);
+    }
+}

engines/imagemagick/src/test/java/org/alfresco/transform/imagemagick/ImageMagickLivenessReadinessProbeTest.java

@@ -0,0 +1,37 @@
+/*
+ * #%L
+ * Alfresco Transform Core
+ * %%
+ * Copyright (C) 2005 - 2023 Alfresco Software Limited
+ * %%
+ * This file is part of the Alfresco software.
+ * -
+ * If the software was purchased under a paid Alfresco license, the terms of
+ * the paid license agreement will prevail.  Otherwise, the software is
+ * provided under the following open source license terms:
+ * -
+ * Alfresco is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ * -
+ * Alfresco is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Lesser General Public License for more details.
+ * -
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with Alfresco. If not, see <http://www.gnu.org/licenses/>.
+ * #L%
+ */
+
+package org.alfresco.transform.imagemagick;
+
+import org.alfresco.transform.base.LivenessReadinessProbeTest;
+
+public class ImageMagickLivenessReadinessProbeTest extends LivenessReadinessProbeTest {
+    @Override
+    protected ImagesForTests getImageForTest() {
+        return new ImagesForTests("alfresco-imagemagick", "image/jpeg", "image/png", "test.jpeg");
+    }
+}

engines/libreoffice/src/test/java/org/alfresco/transform/libreoffice/LibreOfficeLivenessReadinessProbeTest.java

@@ -0,0 +1,37 @@
+/*
+ * #%L
+ * Alfresco Transform Core
+ * %%
+ * Copyright (C) 2005 - 2023 Alfresco Software Limited
+ * %%
+ * This file is part of the Alfresco software.
+ * -
+ * If the software was purchased under a paid Alfresco license, the terms of
+ * the paid license agreement will prevail.  Otherwise, the software is
+ * provided under the following open source license terms:
+ * -
+ * Alfresco is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ * -
+ * Alfresco is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Lesser General Public License for more details.
+ * -
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with Alfresco. If not, see <http://www.gnu.org/licenses/>.
+ * #L%
+ */
+
+package org.alfresco.transform.libreoffice;
+
+import org.alfresco.transform.base.LivenessReadinessProbeTest;
+
+public class LibreOfficeLivenessReadinessProbeTest extends LivenessReadinessProbeTest {
+    @Override
+    protected ImagesForTests getImageForTest() {
+        return new ImagesForTests("alfresco-libreoffice", "text/plain", "application/pdf", "original.txt");
+    }
+}

engines/misc/src/test/java/org/alfresco/transform/misc/MiscLivenessReadinessProbeTest.java

@@ -0,0 +1,37 @@
+/*
+ * #%L
+ * Alfresco Transform Core
+ * %%
+ * Copyright (C) 2005 - 2023 Alfresco Software Limited
+ * %%
+ * This file is part of the Alfresco software.
+ * -
+ * If the software was purchased under a paid Alfresco license, the terms of
+ * the paid license agreement will prevail.  Otherwise, the software is
+ * provided under the following open source license terms:
+ * -
+ * Alfresco is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ * -
+ * Alfresco is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Lesser General Public License for more details.
+ * -
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with Alfresco. If not, see <http://www.gnu.org/licenses/>.
+ * #L%
+ */
+
+package org.alfresco.transform.misc;
+
+import org.alfresco.transform.base.LivenessReadinessProbeTest;
+
+public class MiscLivenessReadinessProbeTest extends LivenessReadinessProbeTest {
+    @Override
+    protected ImagesForTests getImageForTest() {
+        return new ImagesForTests("alfresco-transform-misc", "text/plain", "text/plain", "original.txt");
+    }
+}

engines/pdfrenderer/src/test/java/org/alfresco/transform/pdfrenderer/PdfRendererLivenessReadinessProbeTest.java

@@ -0,0 +1,37 @@
+/*
+ * #%L
+ * Alfresco Transform Core
+ * %%
+ * Copyright (C) 2005 - 2023 Alfresco Software Limited
+ * %%
+ * This file is part of the Alfresco software.
+ * -
+ * If the software was purchased under a paid Alfresco license, the terms of
+ * the paid license agreement will prevail.  Otherwise, the software is
+ * provided under the following open source license terms:
+ * -
+ * Alfresco is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ * -
+ * Alfresco is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Lesser General Public License for more details.
+ * -
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with Alfresco. If not, see <http://www.gnu.org/licenses/>.
+ * #L%
+ */
+
+package org.alfresco.transform.pdfrenderer;
+
+import org.alfresco.transform.base.LivenessReadinessProbeTest;
+
+public class PdfRendererLivenessReadinessProbeTest extends LivenessReadinessProbeTest {
+    @Override
+    protected ImagesForTests getImageForTest() {
+        return new ImagesForTests("alfresco-pdf-renderer", "application/pdf", "image/png", "test.pdf");
+    }
+}

engines/tika/src/test/java/org/alfresco/transform/tika/TikaLivenessReadinessProbeTest.java

@@ -0,0 +1,38 @@
+/*
+ * #%L
+ * Alfresco Transform Core
+ * %%
+ * Copyright (C) 2005 - 2023 Alfresco Software Limited
+ * %%
+ * This file is part of the Alfresco software.
+ * -
+ * If the software was purchased under a paid Alfresco license, the terms of
+ * the paid license agreement will prevail.  Otherwise, the software is
+ * provided under the following open source license terms:
+ * -
+ * Alfresco is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ * -
+ * Alfresco is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Lesser General Public License for more details.
+ * -
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with Alfresco. If not, see <http://www.gnu.org/licenses/>.
+ * #L%
+ */
+
+package org.alfresco.transform.tika;
+
+import org.alfresco.transform.base.LivenessReadinessProbeTest;
+
+
+public class TikaLivenessReadinessProbeTest extends LivenessReadinessProbeTest {
+    @Override
+    protected ImagesForTests getImageForTest() {
+        return new ImagesForTests("alfresco-tika", "text/plain", "text/plain", "original.txt");
+    }
+}