Fix CI test

Author: tadayosi

src/test/java/com/github/tadayosi/torchserve/client/InferenceTest.java

@@ -13,40 +13,40 @@
 import static org.junit.jupiter.api.Assertions.assertThrows;
 
 @Testcontainers
-class InferenceTest extends TorchServeTestSupport {
+public class InferenceTest extends TorchServeTestSupport {
 
     private static final String DEFAULT_MODEL = "squeezenet1_1";
     private static final String DEFAULT_MODEL_VERSION = "1.0";
     private static final String TEST_DATA = "src/test/resources/data/kitten.jpg";
 
     @Test
-    void testApiDescription() throws Exception {
+    public void testApiDescription() throws Exception {
         var response = client.inference().apiDescription();
         assertNotNull(response);
     }
 
     @Test
-    void testPing() throws Exception {
+    public void testPing() throws Exception {
         var response = client.inference().ping();
         assertEquals("Healthy", response.getStatus());
     }
 
     @Test
-    void testPredictions() throws Exception {
+    public void testPredictions() throws Exception {
         var body = Files.readAllBytes(Path.of(TEST_DATA));
         var response = client.inference().predictions(DEFAULT_MODEL, body);
         assertInstanceOf(Map.class, response);
     }
 
     @Test
-    void testPredictions_version() throws Exception {
+    public void testPredictions_version() throws Exception {
         var body = Files.readAllBytes(Path.of(TEST_DATA));
         var response = client.inference().predictions(DEFAULT_MODEL, DEFAULT_MODEL_VERSION, body);
         assertInstanceOf(Map.class, response);
     }
 
     @Test
-    void testExplanations() {
+    public void testExplanations() {
         assertThrows(UnsupportedOperationException.class, () -> client.inference().explanations(DEFAULT_MODEL));
     }
 

src/test/java/com/github/tadayosi/torchserve/client/ManagementTest.java

@@ -22,7 +22,7 @@
 import static org.junit.jupiter.api.Assertions.fail;
 
 @Testcontainers
-class ManagementTest extends TorchServeTestSupport {
+public class ManagementTest extends TorchServeTestSupport {
 
     private static final String DEFAULT_MODEL = "squeezenet1_1";
     private static final String DEFAULT_MODEL_VERSION = "1.0";
@@ -31,7 +31,7 @@ class ManagementTest extends TorchServeTestSupport {
     private static final String TEST_DATA_DIR = "src/test/resources/data";
 
     @Test
-    void testRegisterModel() throws Exception {
+    public void testRegisterModel() throws Exception {
         var url = "https://torchserve.pytorch.org/mar_files/mnist_v2.mar";
         try {
             var response = client.management().registerModel(url, RegisterModelOptions.empty());
@@ -46,7 +46,7 @@ void testRegisterModel() throws Exception {
     class AfterRegisteringModel {
 
         @BeforeEach
-        void registerModel() throws Exception {
+        public void registerModel() throws Exception {
             var url = "https://torchserve.pytorch.org/mar_files/mnist_v2.mar";
             try {
                 client.management().registerModel(url, RegisterModelOptions.empty());
@@ -56,13 +56,13 @@ void registerModel() throws Exception {
         }
 
         @Test
-        void testUnregisterModel() throws Exception {
+        public void testUnregisterModel() throws Exception {
             var response = client.management().unregisterModel(ADDED_MODEL, UnregisterModelOptions.empty());
             assertTrue(response.getStatus().contains("unregistered"));
         }
 
         @Test
-        void testUnregisterModel_version() throws Exception {
+        public void testUnregisterModel_version() throws Exception {
             var response = client.management().unregisterModel(ADDED_MODEL, ADDED_MODEL_VERSION, UnregisterModelOptions.empty());
             assertTrue(response.getStatus().contains("unregistered"));
         }
@@ -71,12 +71,12 @@ void testUnregisterModel_version() throws Exception {
         class BeforeUnregisteringModel {
 
             @AfterEach
-            void unregisterModel() throws Exception {
+            public void unregisterModel() throws Exception {
                 client.management().unregisterModel(ADDED_MODEL, UnregisterModelOptions.empty());
             }
 
             @Test
-            void testSetAutoScale() throws Exception {
+            public void testSetAutoScale() throws Exception {
                 var response1 = client.management().setAutoScale(ADDED_MODEL,
                     SetAutoScaleOptions.builder()
                         .minWorker(1)
@@ -92,7 +92,7 @@ void testSetAutoScale() throws Exception {
             }
 
             @Test
-            void testSetAutoScale_version() throws Exception {
+            public void testSetAutoScale_version() throws Exception {
                 var response1 = client.management().setAutoScale(ADDED_MODEL, ADDED_MODEL_VERSION,
                     SetAutoScaleOptions.builder()
                         .minWorker(1)
@@ -109,22 +109,22 @@ void testSetAutoScale_version() throws Exception {
     }
 
     @Test
-    void testDescribeModel() throws Exception {
+    public void testDescribeModel() throws Exception {
         var response = client.management().describeModel(DEFAULT_MODEL);
         assertEquals(1, response.size());
         assertEquals("squeezenet1_1", response.get(0).getModelName());
     }
 
     @Test
-    void testDescribeModel_version() throws Exception {
+    public void testDescribeModel_version() throws Exception {
         var response = client.management().describeModel(DEFAULT_MODEL, DEFAULT_MODEL_VERSION);
         assertEquals(1, response.size());
         assertEquals("squeezenet1_1", response.get(0).getModelName());
         assertEquals("1.0", response.get(0).getModelVersion());
     }
 
     @Test
-    void testListModels() throws Exception {
+    public void testListModels() throws Exception {
         int limit = 10;
         String nextPageToken = null;
         var response = client.management().listModels(limit, nextPageToken);
@@ -134,19 +134,19 @@ void testListModels() throws Exception {
     }
 
     @Test
-    void testSetDefault() throws Exception {
+    public void testSetDefault() throws Exception {
         var response = client.management().setDefault(DEFAULT_MODEL, DEFAULT_MODEL_VERSION);
         assertTrue(response.getStatus().contains("Default vesion succsesfully updated"));
     }
 
     @Test
-    void testApiDescription() throws Exception {
+    public void testApiDescription() throws Exception {
         var response = client.management().apiDescription();
         assertEquals("TorchServe APIs", response.getInfo().get("title"));
     }
 
     @Test
-    void testToken() throws Exception {
+    public void testToken() throws Exception {
         assertThrows(UnsupportedOperationException.class, () -> client.management().token("management"));
     }
 

src/test/java/com/github/tadayosi/torchserve/client/MetricsTest.java

@@ -6,10 +6,10 @@
 import static org.junit.jupiter.api.Assertions.assertNotNull;
 
 @Testcontainers
-class MetricsTest extends TorchServeTestSupport {
+public class MetricsTest extends TorchServeTestSupport {
 
     @Test
-    void testMetrics() throws Exception {
+    public void testMetrics() throws Exception {
         var response = client.metrics().metrics();
         assertNotNull(response);
     }

src/test/java/com/github/tadayosi/torchserve/client/TorchServeTestSupport.java

@@ -7,12 +7,12 @@
 import org.testcontainers.utility.DockerImageName;
 import org.testcontainers.utility.MountableFile;
 
-class TorchServeTestSupport {
+public class TorchServeTestSupport {
 
     private static final String IMAGE_NAME = "aarch64".equals(System.getProperty("os.arch")) ? "tadayosi/torchserve" : "pytorch/torchserve";
 
     @Container
-    static GenericContainer<?> torchServe = new GenericContainer<>(DockerImageName.parse(IMAGE_NAME))
+    public static GenericContainer<?> torchServe = new GenericContainer<>(DockerImageName.parse(IMAGE_NAME))
         .withExposedPorts(8080, 8081, 8082)
         .withCopyFileToContainer(MountableFile.forClasspathResource("torchserve/config.properties"), "/home/model-server/config.properties")
         .withCopyFileToContainer(MountableFile.forClasspathResource("models/squeezenet1_1.mar"), "/home/model-server/model-store/squeezenet1_1.mar")
@@ -22,7 +22,7 @@ class TorchServeTestSupport {
     protected TorchServeClient client;
 
     @BeforeEach
-    void setUp() {
+    public void setUp() {
         client = TorchServeClient.builder()
             .inferencePort(torchServe.getMappedPort(8080))
             .managementPort(torchServe.getMappedPort(8081))