Add end2end test

LangStream · Dec 13, 2023 · 072a2ad · 072a2ad
1 parent c05e0a6
commit 072a2ad
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/langstream-e2e-tests/src/test/java/ai/langstream/tests/PythonAgentsIT.java b/langstream-e2e-tests/src/test/java/ai/langstream/tests/PythonAgentsIT.java
@@ -24,6 +24,7 @@
 import org.junit.jupiter.api.Tag;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.extension.ExtendWith;
+import org.testcontainers.shaded.com.google.common.base.Strings;
 
 @Slf4j
 @ExtendWith(BaseEndToEndTest.class)
@@ -115,8 +116,12 @@ public void testSource() {
                                 .formatted(applicationId)
                                 .split(" "));
         log.info("Output: {}", output);
+        String bigPayload = Strings.repeat("test", 10000);
         Assertions.assertTrue(
-                output.contains("{\"record\":{\"key\":null,\"value\":\"test\",\"headers\":{}}"));
+                output.contains(
+                        "{\"record\":{\"key\":null,\"value\":\""
+                                + bigPayload
+                                + "\",\"headers\":{}}"));
 
         deleteAppAndAwaitCleanup(tenant, applicationId);
     }

diff --git a/langstream-e2e-tests/src/test/resources/apps/python-source/python/example.py b/langstream-e2e-tests/src/test/resources/apps/python-source/python/example.py
@@ -26,7 +26,7 @@ def read(self):
         if not self.sent:
             logging.info("Sending the record")
             self.sent = True
-            return [SimpleRecord("test")]
+            return [SimpleRecord("test" * 10000)]
         return []
 
     def commit(self, records):