From 439f6319169988dbc84704c9cdd59cf8b13e491c Mon Sep 17 00:00:00 2001
From: Martin Polden <mpolden@mpolden.no>
Date: Tue, 5 Apr 2022 13:39:51 +0200
Subject: Limit number of ZIP entries read by controller

---
 .../test/java/com/yahoo/compress/ArchiveStreamReaderTest.java    | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'vespajlib/src/test')

diff --git a/vespajlib/src/test/java/com/yahoo/compress/ArchiveStreamReaderTest.java b/vespajlib/src/test/java/com/yahoo/compress/ArchiveStreamReaderTest.java
index a288b9f0c55..b7f019282b7 100644
--- a/vespajlib/src/test/java/com/yahoo/compress/ArchiveStreamReaderTest.java
+++ b/vespajlib/src/test/java/com/yahoo/compress/ArchiveStreamReaderTest.java
@@ -61,6 +61,15 @@ class ArchiveStreamReaderTest {
         } catch (IllegalArgumentException ignored) {}
     }
 
+    @Test
+    void entry_limit() {
+        Map<String, String> entries = Map.of("foo.xml", "foo", "bar.xml", "bar");
+        try {
+            readAll(zip(entries), Options.standard().maxEntries(1));
+            fail("Expected exception");
+        } catch (IllegalArgumentException ignored) {}
+    }
+
     @Test
     void paths() {
         Map<String, Boolean> tests = Map.of(
-- 
cgit v1.2.3