X-Git-Url: http://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fde%2Fjuplo%2Fkafka%2FApplicationTests.java;fp=src%2Ftest%2Fjava%2Fde%2Fjuplo%2Fkafka%2FApplicationTests.java;h=7246e173edf8738fcb3766851e7b2825030625b7;hb=1c6d263c619010d23bf502c14dda45db11a2baf6;hp=740c09ce57f348c3aa2b29f65f96a68be5b41a00;hpb=600b0b10f8e98bebef80f75e391a78c459ffb45c;p=demos%2Fkafka%2Ftraining

diff --git a/src/test/java/de/juplo/kafka/ApplicationTests.java b/src/test/java/de/juplo/kafka/ApplicationTests.java
index 740c09c..7246e17 100644
--- a/src/test/java/de/juplo/kafka/ApplicationTests.java
+++ b/src/test/java/de/juplo/kafka/ApplicationTests.java
@@ -4,6 +4,7 @@ import lombok.extern.slf4j.Slf4j;
 import org.apache.kafka.clients.producer.ProducerRecord;
 import org.apache.kafka.common.serialization.StringSerializer;
 import org.apache.kafka.common.utils.Bytes;
+import org.assertj.core.api.Assertions;
 import org.springframework.beans.factory.annotation.Autowired;
 
 import java.util.*;
@@ -124,15 +125,29 @@ public class ApplicationTests extends GenericApplicationTests<String, String>
             tests.stateRepository.findById(Integer.toString(i)).get();
 
         stateDocument
-            .results.entrySet().stream()
+            .results
+            .entrySet()
+            .stream()
             .forEach(entry ->
             {
               String user = entry.getKey();
               List<AdderResult> resultsForUser = entry.getValue();
 
+              for (int j=0; j < resultsForUser.size(); j++)
+              {
+                if (!(j < state.get(user).size()))
+                {
+                  break;
+                }
+
+                assertThat(resultsForUser.get(j))
+                    .as("Unexpected results calculation %i of user %s", j, user)
+                    .isEqualTo(state.get(user).get(j));
+              }
+
               assertThat(state.get(user))
-                  .as("Unexpected results for user %s", user)
-                  .containsExactlyElementsOf(resultsForUser);
+                  .as("More results calculated for user %s as expected", user)
+                  .containsAll(resultsForUser);
             });
       }
     }