Adapt stream Java tests to client 0.12.0 snapshot

acogoluegnes · mergify[bot] · commit 422be98e7312 · 2023-07-17T09:38:48.000Z
After changes for rabbitmq/rabbitmq-stream-java-client#333. (cherry picked from commit c594c77) # Conflicts: # deps/rabbitmq_stream/test/rabbit_stream_SUITE_data/pom.xml # deps/rabbitmq_stream_management/test/http_SUITE_data/pom.xml (cherry picked from commit f57c6db)
diff --git a/deps/rabbitmq_stream/test/rabbit_stream_SUITE_data/pom.xml b/deps/rabbitmq_stream/test/rabbit_stream_SUITE_data/pom.xml
@@ -26,13 +26,23 @@
     </developers>
 
     <properties>
+<<<<<<< HEAD
         <stream-client.version>[0.7.0-SNAPSHOT,)</stream-client.version>
         <junit.jupiter.version>5.9.0</junit.jupiter.version>
         <assertj.version>3.23.1</assertj.version>
         <logback.version>1.2.11</logback.version>
         <maven.compiler.plugin.version>3.10.1</maven.compiler.plugin.version>
         <maven-surefire-plugin.version>2.22.2</maven-surefire-plugin.version>
         <spotless.version>2.24.0</spotless.version>
+=======
+        <stream-client.version>[0.12.0-SNAPSHOT,)</stream-client.version>
+        <junit.jupiter.version>5.9.3</junit.jupiter.version>
+        <assertj.version>3.24.2</assertj.version>
+        <logback.version>1.2.12</logback.version>
+        <maven.compiler.plugin.version>3.11.0</maven.compiler.plugin.version>
+        <maven-surefire-plugin.version>3.1.2</maven-surefire-plugin.version>
+        <spotless.version>2.37.0</spotless.version>
+>>>>>>> c594c77049 (Adapt stream Java tests to client 0.12.0 snapshot)
         <google-java-format.version>1.17.0</google-java-format.version>
         <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
     </properties>
diff --git a/deps/rabbitmq_stream/test/rabbit_stream_SUITE_data/src/test/java/com/rabbitmq/stream/FailureTest.java b/deps/rabbitmq_stream/test/rabbit_stream_SUITE_data/src/test/java/com/rabbitmq/stream/FailureTest.java
@@ -16,9 +16,8 @@
 
 package com.rabbitmq.stream;
 
+import static com.rabbitmq.stream.TestUtils.*;
 import static com.rabbitmq.stream.TestUtils.ResponseConditions.ok;
-import static com.rabbitmq.stream.TestUtils.waitAtMost;
-import static com.rabbitmq.stream.TestUtils.waitUntil;
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.assertj.core.api.Assertions.fail;
 
@@ -164,7 +163,7 @@ void leaderFailureWhenPublisherConnectedToReplica() throws Exception {
             new Client.ClientParameters()
                 .port(TestUtils.streamPortNode1())
                 .messageListener(
-                    (subscriptionId, offset, chunkTimestamp, committedChunkId, msg) -> {
+                    (subscriptionId, offset, chunkTimestamp, committedChunkId, context, msg) -> {
                       bodies.add(new String(msg.getBodyAsBinary(), StandardCharsets.UTF_8));
                       consumeLatch.countDown();
                     }));
@@ -341,11 +340,14 @@ void noLostConfirmedMessagesWhenLeaderGoesAway() throws Exception {
         cf.get(
             new Client.ClientParameters()
                 .port(m.getReplicas().get(0).getPort())
-                .chunkListener(
-                    (client1, subscriptionId, offset, messageCount, dataSize) ->
-                        client1.credit(subscriptionId, 1))
+                .chunkListener(credit())
                 .messageListener(
-                    (subscriptionId, offset, chunkTimestamp, committedChunkId, message) -> {
+                    (subscriptionId,
+                        offset,
+                        chunkTimestamp,
+                        committedChunkId,
+                        context,
+                        message) -> {
                       consumed.add(message);
                       generations.add((Long) message.getApplicationProperties().get("generation"));
                       if (consumed.size() == confirmed.size()) {
@@ -447,7 +449,7 @@ void consumerReattachesToOtherReplicaWhenReplicaGoesAway() throws Exception {
     Set<Long> generations = ConcurrentHashMap.newKeySet();
     Set<Long> consumedIds = ConcurrentHashMap.newKeySet();
     Client.MessageListener messageListener =
-        (subscriptionId, offset, chunkTimestamp, committedChunkId, message) -> {
+        (subscriptionId, offset, chunkTimestamp, committedChunkId, context, message) -> {
           consumed.add(message);
           generations.add((Long) message.getApplicationProperties().get("generation"));
           consumedIds.add(message.getProperties().getMessageIdAsLong());
@@ -471,9 +473,7 @@ void consumerReattachesToOtherReplicaWhenReplicaGoesAway() throws Exception {
                           new Client.ClientParameters()
                               .port(newReplicaPort)
                               .shutdownListener(shutdownListenerReference.get())
-                              .chunkListener(
-                                  (client1, subscriptionId, offset, messageCount, dataSize) ->
-                                      client1.credit(subscriptionId, 1))
+                              .chunkListener(credit())
                               .messageListener(messageListener));
 
                   newConsumer.subscribe(
@@ -494,9 +494,7 @@ void consumerReattachesToOtherReplicaWhenReplicaGoesAway() throws Exception {
             new Client.ClientParameters()
                 .port(replica.getPort())
                 .shutdownListener(shutdownListener)
-                .chunkListener(
-                    (client1, subscriptionId, offset, messageCount, dataSize) ->
-                        client1.credit(subscriptionId, 1))
+                .chunkListener(credit())
                 .messageListener(messageListener));
 
     Client.Response response =
diff --git a/deps/rabbitmq_stream/test/rabbit_stream_SUITE_data/src/test/java/com/rabbitmq/stream/StreamTest.java b/deps/rabbitmq_stream/test/rabbit_stream_SUITE_data/src/test/java/com/rabbitmq/stream/StreamTest.java
@@ -16,6 +16,7 @@
 
 package com.rabbitmq.stream;
 
+import static com.rabbitmq.stream.TestUtils.credit;
 import static org.assertj.core.api.Assertions.assertThat;
 
 import com.rabbitmq.stream.impl.Client;
@@ -115,11 +116,14 @@ void shouldBePossibleToPublishFromAnyNodeAndConsumeFromAnyMember(
         cf.get(
             new Client.ClientParameters()
                 .port(consumerBroker.apply(streamMetadata).getPort())
-                .chunkListener(
-                    (client1, subscriptionId, offset, messageCount1, dataSize) ->
-                        client1.credit(subscriptionId, 10))
+                .chunkListener(credit())
                 .messageListener(
-                    (subscriptionId, offset, chunkTimestamp, committedChunkId, message) -> {
+                    (subscriptionId,
+                        offset,
+                        chunkTimestamp,
+                        committedChunkId,
+                        context,
+                        message) -> {
                       bodies.add(new String(message.getBodyAsBinary(), StandardCharsets.UTF_8));
                       consumingLatch.countDown();
                     }));
@@ -128,7 +132,8 @@ void shouldBePossibleToPublishFromAnyNodeAndConsumeFromAnyMember(
 
     assertThat(consumingLatch.await(10, TimeUnit.SECONDS)).isTrue();
     assertThat(bodies).hasSize(messageCount);
-    IntStream.range(0, messageCount).forEach(i -> assertThat(bodies.contains("hello " + i)));
+    IntStream.range(0, messageCount)
+        .forEach(i -> assertThat(bodies.contains("hello " + i)).isTrue());
   }
 
   @Test
diff --git a/deps/rabbitmq_stream/test/rabbit_stream_SUITE_data/src/test/java/com/rabbitmq/stream/TestUtils.java b/deps/rabbitmq_stream/test/rabbit_stream_SUITE_data/src/test/java/com/rabbitmq/stream/TestUtils.java
@@ -218,4 +218,11 @@ static Condition<Response> responseCode(short expectedResponse) {
           expectedResponse);
     }
   }
+
+  static Client.ChunkListener credit() {
+    return (client, subscriptionId, offset, messageCount, dataSize) -> {
+      client.credit(subscriptionId, 1);
+      return null;
+    };
+  }
 }
diff --git a/deps/rabbitmq_stream_management/test/http_SUITE_data/pom.xml b/deps/rabbitmq_stream_management/test/http_SUITE_data/pom.xml
@@ -26,13 +26,23 @@
     </developers>
 
     <properties>
+<<<<<<< HEAD
         <stream-client.version>[0.7.0-SNAPSHOT,)</stream-client.version>
         <junit.jupiter.version>5.9.0</junit.jupiter.version>
         <assertj.version>3.23.1</assertj.version>
         <logback.version>1.2.11</logback.version>
         <maven.compiler.plugin.version>3.10.1</maven.compiler.plugin.version>
         <maven-surefire-plugin.version>2.22.2</maven-surefire-plugin.version>
         <spotless.version>2.24.0</spotless.version>
+=======
+        <stream-client.version>[0.12.0-SNAPSHOT,)</stream-client.version>
+        <junit.jupiter.version>5.9.3</junit.jupiter.version>
+        <assertj.version>3.24.2</assertj.version>
+        <logback.version>1.2.12</logback.version>
+        <maven.compiler.plugin.version>3.11.0</maven.compiler.plugin.version>
+        <maven-surefire-plugin.version>3.1.2</maven-surefire-plugin.version>
+        <spotless.version>2.37.0</spotless.version>
+>>>>>>> c594c77049 (Adapt stream Java tests to client 0.12.0 snapshot)
         <google-java-format.version>1.17.0</google-java-format.version>
         <okhttp.version>4.9.3</okhttp.version>
         <gson.version>2.8.9</gson.version>
diff --git a/deps/rabbitmq_stream_management/test/http_SUITE_data/src/test/java/com/rabbitmq/stream/HttpTest.java b/deps/rabbitmq_stream_management/test/http_SUITE_data/src/test/java/com/rabbitmq/stream/HttpTest.java
@@ -140,11 +140,6 @@ static List<Map<String, Object>> entities(List<Map<String, Object>> entities, Cl
         .collect(Collectors.toList());
   }
 
-  static List<Map<String, Object>> entities(
-      List<Map<String, Object>> entities, Predicate<Map<String, Object>> filter) {
-    return entities.stream().filter(filter).collect(Collectors.toList());
-  }
-
   static Map<String, Object> entity(
       List<Map<String, Object>> entities, Predicate<Map<String, Object>> filter) {
     return entities.stream().filter(filter).findFirst().orElse(Collections.emptyMap());
@@ -558,9 +553,7 @@ void consumers(Map<String, String> subscriptionProperties) throws Exception {
         cf.get(
             new ClientParameters()
                 .clientProperty("connection_name", connectionProvidedName)
-                .chunkListener(
-                    (client1, subscriptionId, offset, messageCount, dataSize) ->
-                        client1.credit(subscriptionId, 1))
+                .chunkListener(TestUtils.credit())
                 .shutdownListener(shutdownContext -> closed.set(true)));
 
     client.subscribe((byte) 0, stream, OffsetSpecification.first(), 10, subscriptionProperties);
diff --git a/deps/rabbitmq_stream_management/test/http_SUITE_data/src/test/java/com/rabbitmq/stream/TestUtils.java b/deps/rabbitmq_stream_management/test/http_SUITE_data/src/test/java/com/rabbitmq/stream/TestUtils.java
@@ -257,4 +257,11 @@ static Condition<Object> notNull() {
   static Condition<Object> isNull() {
     return new Condition<>(Objects::isNull, "null");
   }
+
+  static Client.ChunkListener credit() {
+    return (client, subscriptionId, offset, messageCount, dataSize) -> {
+      client.credit(subscriptionId, 1);
+      return null;
+    };
+  }
 }

Original file line number	Diff line number	Diff line change
`@@ -218,4 +218,11 @@ static Condition<Response> responseCode(short expectedResponse) {`
`218`	`218`	`expectedResponse);`
`219`	`219`	`}`
`220`	`220`	`}`
	`221`	`+`
	`222`	`+ static Client.ChunkListener credit() {`
	`223`	`+ return (client, subscriptionId, offset, messageCount, dataSize) -> {`
	`224`	`+ client.credit(subscriptionId, 1);`
	`225`	`+ return null;`
	`226`	`+ };`
	`227`	`+ }`
`221`	`228`	`}`
Original file line number	Diff line number	Diff line change
`@@ -257,4 +257,11 @@ static Condition<Object> notNull() {`
`257`	`257`	`static Condition<Object> isNull() {`
`258`	`258`	`return new Condition<>(Objects::isNull, "null");`
`259`	`259`	`}`
	`260`	`+`
	`261`	`+ static Client.ChunkListener credit() {`
	`262`	`+ return (client, subscriptionId, offset, messageCount, dataSize) -> {`
	`263`	`+ client.credit(subscriptionId, 1);`
	`264`	`+ return null;`
	`265`	`+ };`
	`266`	`+ }`
`260`	`267`	`}`