test(it): do not stop static containers

Author: bevzzz

src/it/java/io/weaviate/containers/Container.java

@@ -1,7 +1,9 @@
 package io.weaviate.containers;
 
 import java.util.Arrays;
+import java.util.Collection;
 import java.util.List;
+import java.util.Set;
 
 import org.junit.rules.TestRule;
 import org.junit.runner.Description;
@@ -19,6 +21,7 @@ public class Container {
   public static final MinIo MINIO = MinIo.createDefault();
 
   public static ContainerGroup compose(Weaviate weaviate, GenericContainer<?>... containers) {
+    assert weaviate != WEAVIATE : "cannot compose with static WEAVIATE contaier";
     return new ContainerGroup(weaviate, containers);
   }
 
@@ -50,7 +53,9 @@ public void start() {
     @Override
     public void stop() {
       weaviate.stop();
-      containers.forEach(GenericContainer::stop);
+      containers.stream()
+          .dropWhile(c -> c == MODEL2VEC || c == IMG2VEC_NEURAL || c == MINIO)
+          .forEach(GenericContainer::stop);
     }
 
     private void setSharedNetwork() {

src/it/java/io/weaviate/containers/Weaviate.java

@@ -28,9 +28,11 @@ public class Weaviate extends WeaviateContainer {
   public static final String DOCKER_IMAGE = "semitechnologies/weaviate";
   public static final String LATEST_VERSION = Version.latest().semver.toString();
   public static final String VERSION;
+  private static final boolean DEBUG;
 
   static {
     VERSION = System.getenv().getOrDefault("WEAVIATE_VERSION", LATEST_VERSION);
+    DEBUG = System.getenv("DEBUG") != null;
   }
   public static String OIDC_ISSUER = "https://auth.wcs.api.weaviate.io/auth/realms/SeMI";
 
@@ -93,6 +95,9 @@ public String getContainerName() {
   public WeaviateClient getClient() {
     if (!isRunning()) {
       start();
+      if (DEBUG) {
+        followOutput(frame -> System.out.println(frame.getUtf8String()));
+      }
     }
     if (clientInstance != null) {
       return clientInstance;

src/it/java/io/weaviate/integration/ClusterITest.java

@@ -4,9 +4,7 @@
 
 import org.assertj.core.api.Assertions;
 import org.junit.AfterClass;
-import org.junit.ClassRule;
 import org.junit.Test;
-import org.junit.rules.TestRule;
 
 import io.weaviate.ConcurrentTest;
 import io.weaviate.client6.v1.api.WeaviateClient;
@@ -16,7 +14,6 @@
 import io.weaviate.client6.v1.api.cluster.replication.Replication;
 import io.weaviate.client6.v1.api.cluster.replication.ReplicationState;
 import io.weaviate.client6.v1.api.cluster.replication.ReplicationType;
-import io.weaviate.containers.Container;
 import io.weaviate.containers.Weaviate;
 import io.weaviate.containers.Weaviate.Version;
 

src/it/java/io/weaviate/integration/SearchITest.java

@@ -32,6 +32,7 @@
 import io.weaviate.client6.v1.api.collections.generate.GenerativeObject;
 import io.weaviate.client6.v1.api.collections.generate.TaskOutput;
 import io.weaviate.client6.v1.api.collections.generative.DummyGenerative;
+import io.weaviate.client6.v1.api.collections.query.FetchObjectById;
 import io.weaviate.client6.v1.api.collections.query.Filter;
 import io.weaviate.client6.v1.api.collections.query.GroupBy;
 import io.weaviate.client6.v1.api.collections.query.Metadata;
@@ -790,4 +791,52 @@ public void test_rerankQueries() throws IOException {
     // Assert: ranking not important really, just that the request was valid.
     Assertions.assertThat(got.objects()).hasSize(2);
   }
+
+  @Test
+  public void testVectorizerModel2VecPropeties() throws IOException {
+    var collectionName = ns("Books");
+    client.collections.create(collectionName,
+        col -> col
+            .properties(Property.text("title"), Property.text("author"))
+            .vectorConfig(
+                VectorConfig.text2vecModel2Vec("title_vec", v -> v.sourceProperties("title")),
+                VectorConfig.text2vecModel2Vec("author_vec", v -> v.sourceProperties("author"))));
+
+    var books = client.collections.use(collectionName);
+    Assertions.assertThat(books).isNotNull();
+
+    WeaviateObject<Map<String, Object>> dune = WeaviateObject.of(
+        o -> o.properties(Map.of(
+            "title", "Dune",
+            "author", "Frank Herbert")));
+    WeaviateObject<Map<String, Object>> abc = WeaviateObject.of(
+        o -> o.properties(Map.of(
+            "title", "ABC",
+            "author", "ABC")));
+
+    var resp = books.data.insertMany(dune, abc);
+    Assertions.assertThat(resp).isNotNull().satisfies(s -> {
+      Assertions.assertThat(s.errors()).isEmpty();
+    });
+
+    // Assert that for Dune we have generated 2 different vectors
+    var gotDune = books.query.fetchObjectById(dune.uuid(), FetchObjectById.Builder::includeVector);
+    Assertions.assertThat(gotDune).get()
+        .extracting(WeaviateObject::vectors)
+        .satisfies(v -> {
+          Assertions.assertThat(v.getSingle("title_vec")).isNotEmpty();
+          Assertions.assertThat(v.getSingle("author_vec")).isNotEmpty();
+          Assertions.assertThat(v.getSingle("title_vec")).isNotEqualTo(v.getSingle("author_vec"));
+        });
+
+    // Assert that for ABC we have generated same vectors
+    var gotAbc = books.query.fetchObjectById(abc.uuid(), FetchObjectById.Builder::includeVector);
+    Assertions.assertThat(gotAbc).get()
+        .extracting(WeaviateObject::vectors)
+        .satisfies(v -> {
+          Assertions.assertThat(v.getSingle("title_vec")).isNotEmpty();
+          Assertions.assertThat(v.getSingle("author_vec")).isNotEmpty();
+          Assertions.assertThat(v.getSingle("title_vec")).isEqualTo(v.getSingle("author_vec"));
+        });
+  }
 }