[NOID] format changes

neo4j-contrib · Jan 20, 2025 · 4fd9699 · 4fd9699
1 parent ba722c6
commit 4fd9699
Show file tree

Hide file tree

Showing 9 changed files with 1,036 additions and 809 deletions.
diff --git a/full-it/src/test/java/apoc/full/it/vectordb/ChromaDbTest.java b/full-it/src/test/java/apoc/full/it/vectordb/ChromaDbTest.java
diff --git a/full-it/src/test/java/apoc/full/it/vectordb/MilvusTest.java b/full-it/src/test/java/apoc/full/it/vectordb/MilvusTest.java
diff --git a/full-it/src/test/java/apoc/full/it/vectordb/QdrantTest.java b/full-it/src/test/java/apoc/full/it/vectordb/QdrantTest.java
diff --git a/full-it/src/test/java/apoc/full/it/vectordb/WeaviateEnterpriseTest.java b/full-it/src/test/java/apoc/full/it/vectordb/WeaviateEnterpriseTest.java
@@ -1,19 +1,5 @@
 package apoc.full.it.vectordb;
 
-import apoc.util.MapUtil;
-import apoc.util.Neo4jContainerExtension;
-import apoc.util.TestContainerUtil;
-import org.junit.AfterClass;
-import org.junit.BeforeClass;
-import org.junit.ClassRule;
-import org.junit.Test;
-import org.junit.rules.TemporaryFolder;
-import org.neo4j.driver.Session;
-import org.testcontainers.containers.Network;
-
-import java.util.List;
-import java.util.Map;
-
 import static apoc.full.it.vectordb.WeaviateTestUtil.*;
 import static apoc.ml.RestAPIConfig.HEADERS_KEY;
 import static apoc.util.TestContainerUtil.createEnterpriseDB;
@@ -26,18 +12,31 @@
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertFalse;
 
+import apoc.util.MapUtil;
+import apoc.util.Neo4jContainerExtension;
+import apoc.util.TestContainerUtil;
+import java.util.List;
+import java.util.Map;
+import org.junit.AfterClass;
+import org.junit.BeforeClass;
+import org.junit.ClassRule;
+import org.junit.Test;
+import org.junit.rules.TemporaryFolder;
+import org.neo4j.driver.Session;
+import org.testcontainers.containers.Network;
 
 public class WeaviateEnterpriseTest {
 
     @ClassRule
     public static TemporaryFolder storeDir = new TemporaryFolder();
+
     private static Neo4jContainerExtension neo4jContainer;
     private static Session session;
 
     @BeforeClass
     public static void setUp() throws Exception {
         Network network = Network.newNetwork();
-        
+
         // We build the project, the artifact will be placed into ./build/libs
         neo4jContainer = createEnterpriseDB(List.of(TestContainerUtil.ApocPackage.FULL), true)
                 .withNetwork(network)
@@ -46,21 +45,22 @@ public static void setUp() throws Exception {
         session = neo4jContainer.getSession();
 
         String weaviateAlias = "weaviate";
-        WEAVIATE_CONTAINER.withNetwork(network)
+        WEAVIATE_CONTAINER
+                .withNetwork(network)
                 .withNetworkAliases(weaviateAlias)
                 .withExposedPorts(WEAVIATE_PORT);
         WEAVIATE_CONTAINER.start();
-        
+
         HOST = weaviateAlias + ":" + WEAVIATE_PORT;
 
-        testCall(session, WEAVIATE_CREATE_COLLECTION_APOC,
-                map("host", HOST, "conf", ADMIN_HEADER_CONF),
-                r -> {
-                    Map value = (Map) r.get("value");
-                    assertEquals("TestCollection", value.get("class"));
-                });
+        testCall(session, WEAVIATE_CREATE_COLLECTION_APOC, map("host", HOST, "conf", ADMIN_HEADER_CONF), r -> {
+            Map value = (Map) r.get("value");
+            assertEquals("TestCollection", value.get("class"));
+        });
 
-        testResult(session, WEAVIATE_UPSERT_QUERY,
+        testResult(
+                session,
+                WEAVIATE_UPSERT_QUERY,
                 map("host", HOST, "id1", ID_1, "id2", ID_2, "conf", ADMIN_HEADER_CONF),
                 r -> {
                     Map<String, Object> row = r.next();
@@ -77,28 +77,34 @@ public static void setUp() throws Exception {
                 });
 
         // -- delete vector
-        testCall(session, WEAVIATE_DELETE_VECTOR_APOC,
-                map("host", HOST, "conf", ADMIN_HEADER_CONF),
-                r -> {
-                    List value = (List) r.get("value");
-                    assertEquals(List.of("7ef2b3a7-1e56-4ddd-b8c3-2ca8901ce308", "7ef2b3a7-1e56-4ddd-b8c3-2ca8901ce309"), value);
-                });
+        testCall(session, WEAVIATE_DELETE_VECTOR_APOC, map("host", HOST, "conf", ADMIN_HEADER_CONF), r -> {
+            List value = (List) r.get("value");
+            assertEquals(
+                    List.of("7ef2b3a7-1e56-4ddd-b8c3-2ca8901ce308", "7ef2b3a7-1e56-4ddd-b8c3-2ca8901ce309"), value);
+        });
     }
 
     @AfterClass
     public static void tearDown() throws Exception {
-        testCallEmpty(session, WEAVIATE_DELETE_COLLECTION_APOC,
-                MapUtil.map("host", HOST, "collectionName", COLLECTION_NAME, "conf", ADMIN_HEADER_CONF)
-        );
+        testCallEmpty(
+                session,
+                WEAVIATE_DELETE_COLLECTION_APOC,
+                MapUtil.map("host", HOST, "collectionName", COLLECTION_NAME, "conf", ADMIN_HEADER_CONF));
         session.close();
         neo4jContainer.close();
         WEAVIATE_CONTAINER.stop();
     }
 
     @Test
     public void queryVectors() {
-        testResult(session,  WEAVIATE_QUERY_APOC,
-                map("host", HOST, "conf", map(ALL_RESULTS_KEY, true, FIELDS_KEY, FIELDS, HEADERS_KEY, ADMIN_AUTHORIZATION)),
+        testResult(
+                session,
+                WEAVIATE_QUERY_APOC,
+                map(
+                        "host",
+                        HOST,
+                        "conf",
+                        map(ALL_RESULTS_KEY, true, FIELDS_KEY, FIELDS, HEADERS_KEY, ADMIN_AUTHORIZATION)),
                 WeaviateTestUtil::queryVectorsAssertions);
     }
 }