1 files changed, 69 insertions, 0 deletions
diff --git a/vespaclient-container-plugin/src/test/java/com/yahoo/vespaxmlparser/MockReader.java b/vespaclient-container-plugin/src/test/java/com/yahoo/vespaxmlparser/MockReader.java
new file mode 100644
index 00000000000..c751849b84e
--- /dev/null
+++ b/vespaclient-container-plugin/src/test/java/com/yahoo/vespaxmlparser/MockReader.java
@@ -0,0 +1,69 @@
+// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.vespaxmlparser;
+
+import com.yahoo.document.Document;
+import com.yahoo.document.DocumentId;
+import com.yahoo.document.DocumentType;
+import com.yahoo.document.DocumentUpdate;
+import com.yahoo.vespa.http.server.MetaStream;
+import com.yahoo.vespa.http.server.util.ByteLimitedInputStream;
+
+import java.io.InputStream;
+import java.lang.reflect.Field;
+
+/**
+ * Mock for ExternalFeedTestCase which had to override package private methods.
+ *
+ * @author <a href="mailto:steinar@yahoo-inc.com">Steinar Knutsen</a>
+ */
+public class MockReader implements FeedReader {
+
+    MetaStream stream;
+    boolean finished = false;
+
+    public MockReader(InputStream stream) throws Exception {
+        this.stream = getMetaStream(stream);
+    }
+
+    private static MetaStream getMetaStream(InputStream stream) {
+        if (stream instanceof MetaStream) {
+            return (MetaStream) stream;
+        }
+        if (!(stream instanceof ByteLimitedInputStream)) {
+            throw new IllegalStateException("Given unknown stream type.");
+        }
+        //Ooooooo this is so ugly
+        try {
+            ByteLimitedInputStream byteLimitedInputStream = (ByteLimitedInputStream) stream;
+            Field f = byteLimitedInputStream.getClass().getDeclaredField("wrappedStream"); //NoSuchFieldException
+            f.setAccessible(true);
+            return (MetaStream) f.get(byteLimitedInputStream);
+        } catch (Exception e) {
+            throw new IllegalStateException("Implementation of ByteLimitedInputStream has changed.", e);
+        }
+    }
+
+    @Override
+    public FeedOperation read() throws Exception {
+        if (finished) {
+            return FeedOperation.INVALID;
+        }
+
+        byte whatToDo = stream.getNextOperation();
+        DocumentId id = new DocumentId("id:banana:banana::doc1");
+        DocumentType docType = new DocumentType("banana");
+        switch (whatToDo) {
+        case 0:
+            return FeedOperation.INVALID;
+        case 1:
+            return new DocumentFeedOperation(new Document(docType, id));
+        case 2:
+            return new RemoveFeedOperation(id);
+        case 3:
+            return new DocumentUpdateFeedOperation(new DocumentUpdate(docType, id));
+        default:
+            throw new RuntimeException("boom");
+        }
+    }
+
+}