77
88import io .weaviate .client6 .v1 .api .collections .query .ConsistencyLevel ;
99import io .weaviate .client6 .v1 .internal .ObjectBuilder ;
10+ import io .weaviate .client6 .v1 .internal .grpc .GrpcTransport ;
1011import io .weaviate .client6 .v1 .internal .orm .CollectionDescriptor ;
12+ import io .weaviate .client6 .v1 .internal .rest .RestTransport ;
13+ import io .weaviate .testutil .transport .MockGrpcTransport ;
14+ import io .weaviate .testutil .transport .MockRestTransport ;
1115
1216public class CollectionHandleDefaultsTest {
1317 private static final CollectionDescriptor <Map <String , Object >> DESCRIPTOR = CollectionDescriptor .ofMap ("Things" );
1418 private static final CollectionHandleDefaults NONE_DEFAULTS = CollectionHandleDefaults .of (ObjectBuilder .identity ());
1519
20+ private static final RestTransport REST = new MockRestTransport ();
21+ private static final GrpcTransport GRPC = new MockGrpcTransport ();
22+
1623 /** CollectionHandle with no defaults. */
1724 private static final CollectionHandle <Map <String , Object >> HANDLE_NONE = new CollectionHandle <>(
18- null , null , DESCRIPTOR , NONE_DEFAULTS );
25+ REST , GRPC , DESCRIPTOR , NONE_DEFAULTS );
1926
2027 /** CollectionHandleAsync with no defaults. */
2128 private static final CollectionHandleAsync <Map <String , Object >> HANDLE_NONE_ASYNC = new CollectionHandleAsync <>(
22- null , null , DESCRIPTOR , NONE_DEFAULTS );
29+ REST , GRPC , DESCRIPTOR , NONE_DEFAULTS );
2330
2431 /** All defaults are {@code null} if none were set. */
2532 @ Test
2633 public void test_defaults () {
2734 Assertions .assertThat (HANDLE_NONE .consistencyLevel ()).as ("default ConsistencyLevel" ).isEmpty ();
28- Assertions .assertThat (HANDLE_NONE .tenant ()).as ("default tenant" ).isNull ();
35+ Assertions .assertThat (HANDLE_NONE .tenant ()).as ("default tenant" ).isEmpty ();
2936 }
3037
3138 /**
@@ -57,7 +64,7 @@ public void test_withConsistencyLevel_async() {
5764 @ Test
5865 public void test_withTenant () {
5966 var handle = HANDLE_NONE .withTenant ("john_doe" );
60- Assertions .assertThat (handle .tenant ()).isEqualTo ("john_doe" );
67+ Assertions .assertThat (handle .tenant ()).get (). isEqualTo ("john_doe" );
6168 Assertions .assertThat (HANDLE_NONE .consistencyLevel ()).isEmpty ();
6269 }
6370
@@ -68,7 +75,7 @@ public void test_withTenant() {
6875 @ Test
6976 public void test_withTenant_async () {
7077 var handle = HANDLE_NONE_ASYNC .withTenant ("john_doe" );
71- Assertions .assertThat (handle .tenant ()).isEqualTo ("john_doe" );
78+ Assertions .assertThat (handle .tenant ()).get (). isEqualTo ("john_doe" );
7279 Assertions .assertThat (HANDLE_NONE_ASYNC .consistencyLevel ()).isEmpty ();
7380 }
7481}
0 commit comments