Refactor ZarrV2 and ZarrV3 tests to improve readability and structure; add new tests for array and group creation

Author: brokkoli71

src/test/java/dev/zarr/zarrjava/ZarrApiTest.java

@@ -0,0 +1,76 @@
+package dev.zarr.zarrjava;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import dev.zarr.zarrjava.store.FilesystemStore;
+import dev.zarr.zarrjava.store.HttpStore;
+import dev.zarr.zarrjava.store.S3Store;
+import dev.zarr.zarrjava.v3.*;
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.Test;
+import software.amazon.awssdk.auth.credentials.AnonymousCredentialsProvider;
+import software.amazon.awssdk.regions.Region;
+import software.amazon.awssdk.services.s3.S3Client;
+
+import java.io.IOException;
+import java.nio.file.Files;
+
+import static dev.zarr.zarrjava.v3.Node.makeObjectMapper;
+
+public class ZarrStoreTest extends ZarrTest {
+        @Test
+    public void testFileSystemStores() throws IOException, ZarrException {
+        FilesystemStore fsStore = new FilesystemStore(TESTDATA);
+        ObjectMapper objectMapper = makeObjectMapper();
+
+        GroupMetadata groupMetadata = objectMapper.readValue(
+            Files.readAllBytes(TESTDATA.resolve("l4_sample").resolve("zarr.json")),
+            GroupMetadata.class
+        );
+
+        String groupMetadataString = objectMapper.writeValueAsString(groupMetadata);
+        Assertions.assertTrue(groupMetadataString.contains("\"zarr_format\":3"));
+        Assertions.assertTrue(groupMetadataString.contains("\"node_type\":\"group\""));
+
+        ArrayMetadata arrayMetadata = objectMapper.readValue(Files.readAllBytes(TESTDATA.resolve(
+                "l4_sample").resolve("color").resolve("1").resolve("zarr.json")),
+            ArrayMetadata.class);
+
+        String arrayMetadataString = objectMapper.writeValueAsString(arrayMetadata);
+        Assertions.assertTrue(arrayMetadataString.contains("\"zarr_format\":3"));
+        Assertions.assertTrue(arrayMetadataString.contains("\"node_type\":\"array\""));
+        Assertions.assertTrue(arrayMetadataString.contains("\"shape\":[1,4096,4096,2048]"));
+
+        Assertions.assertInstanceOf(Array.class, Array.open(fsStore.resolve("l4_sample", "color", "1")));
+
+        Node[] subNodes = Group.open(fsStore.resolve("l4_sample")).list().toArray(Node[]::new);
+        Assertions.assertEquals(2, subNodes.length);
+        Assertions.assertInstanceOf(Group.class, subNodes[0]);
+
+        Array[] colorSubNodes = ((Group) Group.open(fsStore.resolve("l4_sample")).get("color")).list().toArray(Array[]::new);
+
+        Assertions.assertEquals(5, colorSubNodes.length);
+        Assertions.assertInstanceOf(Array.class, colorSubNodes[0]);
+
+        Array array = (Array) ((Group) Group.open(fsStore.resolve("l4_sample")).get("color")).get("1");
+        Assertions.assertArrayEquals(new long[]{1, 4096, 4096, 2048}, array.metadata.shape);
+    }
+
+    @Test
+    public void testS3Store() throws IOException, ZarrException {
+        S3Store s3Store = new S3Store(S3Client.builder()
+            .region(Region.of("eu-west-1"))
+            .credentialsProvider(AnonymousCredentialsProvider.create())
+            .build(), "static.webknossos.org", "data");
+        Array array = Array.open(s3Store.resolve("zarr_v3", "l4_sample", "color", "1"));
+
+        Assertions.assertArrayEquals(new long[]{1, 4096, 4096, 2048}, array.metadata.shape);
+    }
+
+    @Test
+    public void testHttpStore() throws IOException, ZarrException {
+        HttpStore httpStore = new dev.zarr.zarrjava.store.HttpStore("https://static.webknossos.org/data/zarr_v3/l4_sample");
+        Array array = Array.open(httpStore.resolve("color", "1"));
+
+        Assertions.assertArrayEquals(new long[]{1, 4096, 4096, 2048}, array.metadata.shape);
+    }
+}

src/test/java/dev/zarr/zarrjava/ZarrStoreTest.java

@@ -1,9 +1,6 @@
 package dev.zarr.zarrjava;
 
-import dev.zarr.zarrjava.store.S3Store;
-import dev.zarr.zarrjava.v3.Array;
 import org.junit.jupiter.api.BeforeAll;
-import org.junit.jupiter.api.Test;
 
 import java.io.File;
 import java.io.IOException;
@@ -15,18 +12,22 @@
 
 public class ZarrTest {
 
-    final static Path TESTDATA = Paths.get("testdata");
-    final static Path TESTOUTPUT = Paths.get("testoutput");
+    public static final Path TESTDATA = Paths.get("testdata");
+    public static final Path TESTOUTPUT = Paths.get("testoutput");
 
     @BeforeAll
     public static void clearTestoutputFolder() throws IOException {
         if (Files.exists(TESTOUTPUT)) {
             try (Stream<Path> walk = Files.walk(TESTOUTPUT)) {
-                walk.sorted(Comparator.reverseOrder()).map(Path::toFile).forEach(File::delete);
+                walk.sorted(Comparator.reverseOrder())
+                    .map(Path::toFile)
+                    .forEach(file -> {
+                        if (!file.delete()) {
+                            throw new RuntimeException("Failed to delete file: " + file.getAbsolutePath());
+                        }
+                    });
             }
         }
         Files.createDirectory(TESTOUTPUT);
     }
-
-
 }