From 8993b9df81bf0acbbe42db002a202f46f019e7e6 Mon Sep 17 00:00:00 2001
From: Henning Baldersheim <balder@yahoo-inc.com>
Date: Thu, 27 Apr 2023 14:30:27 +0200
Subject: Unify passing of all feed operations through the various feed apis.

---
 .../vespaxmlparser/PositionParserTestCase.java     |  2 +-
 .../yahoo/vespaxmlparser/UriParserTestCase.java    |  2 +-
 .../vespaxmlparser/VespaXMLReaderTestCase.java     | 22 +++++++++++-----------
 3 files changed, 13 insertions(+), 13 deletions(-)

(limited to 'document/src/test/java/com/yahoo')

diff --git a/document/src/test/java/com/yahoo/vespaxmlparser/PositionParserTestCase.java b/document/src/test/java/com/yahoo/vespaxmlparser/PositionParserTestCase.java
index e2070edd397..a8ee7685790 100644
--- a/document/src/test/java/com/yahoo/vespaxmlparser/PositionParserTestCase.java
+++ b/document/src/test/java/com/yahoo/vespaxmlparser/PositionParserTestCase.java
@@ -41,7 +41,7 @@ public class PositionParserTestCase {
     private static void assertDocument(Struct expected, FeedOperation operation) {
         assertNotNull(operation);
         assertEquals(FeedOperation.Type.DOCUMENT, operation.getType());
-        Document doc = operation.getDocument();
+        Document doc = operation.getDocumentPut().getDocument();
         assertNotNull(doc);
         assertEquals(expected, doc.getFieldValue("my_pos"));
     }
diff --git a/document/src/test/java/com/yahoo/vespaxmlparser/UriParserTestCase.java b/document/src/test/java/com/yahoo/vespaxmlparser/UriParserTestCase.java
index 3eb05e6871b..87797e3f4bd 100644
--- a/document/src/test/java/com/yahoo/vespaxmlparser/UriParserTestCase.java
+++ b/document/src/test/java/com/yahoo/vespaxmlparser/UriParserTestCase.java
@@ -64,7 +64,7 @@ public class UriParserTestCase {
         FeedOperation op = it.next();
         assertNotNull(op);
         assertEquals(FeedOperation.Type.DOCUMENT, op.getType());
-        Document doc = op.getDocument();
+        Document doc = op.getDocumentPut().getDocument();
         assertNotNull(doc);
         return doc;
     }
diff --git a/document/src/test/java/com/yahoo/vespaxmlparser/VespaXMLReaderTestCase.java b/document/src/test/java/com/yahoo/vespaxmlparser/VespaXMLReaderTestCase.java
index b2fa7014480..20be6109076 100644
--- a/document/src/test/java/com/yahoo/vespaxmlparser/VespaXMLReaderTestCase.java
+++ b/document/src/test/java/com/yahoo/vespaxmlparser/VespaXMLReaderTestCase.java
@@ -68,7 +68,7 @@ public class VespaXMLReaderTestCase {
         FeedOperation op = parser.read();
 
         assertTrue(FeedOperation.Type.INVALID != op.getType());
-        Document doc = op.getDocument();
+        Document doc = op.getDocumentPut().getDocument();
         assertEquals(new StringFieldValue("testUrl"), doc.getFieldValue("url"));
         assertEquals(new StringFieldValue("testTitle"), doc.getFieldValue("title"));
         assertEquals(new IntegerFieldValue(1), doc.getFieldValue("last_downloaded"));
@@ -339,7 +339,7 @@ public class VespaXMLReaderTestCase {
 
         // empty string
         FeedOperation op = parser.read();
-        assertEquals("id:ns:news::http://news6b", op.getDocument().getId().toString());
+        assertEquals("id:ns:news::http://news6b", op.getDocumentPut().getDocument().getId().toString());
 
         // int array with text
         try {
@@ -398,7 +398,7 @@ public class VespaXMLReaderTestCase {
         }
 
         op = parser.read();
-        assertEquals("id:ns:news::http://news6j", op.getDocument().getId().toString());
+        assertEquals("id:ns:news::http://news6j", op.getDocumentPut().getDocument().getId().toString());
 
         op = parser.read();
         assertEquals(FeedOperation.Type.INVALID, op.getType());
@@ -515,13 +515,13 @@ public class VespaXMLReaderTestCase {
             FeedOperation op = parser.read();
 
             assertEquals(FeedOperation.Type.REMOVE, op.getType());
-            assertEquals("id:ns:news::http://news9a", op.getRemove().toString());
+            assertEquals("id:ns:news::http://news9a", op.getDocumentRemove().getId().toString());
         }
         {
             FeedOperation op = parser.read();
 
             assertEquals(FeedOperation.Type.REMOVE, op.getType());
-            assertEquals("id:ns:news::http://news9b", op.getRemove().toString());
+            assertEquals("id:ns:news::http://news9b", op.getDocumentRemove().getId().toString());
         }
         {
             // Remove without documentid. Not supported.
@@ -539,7 +539,7 @@ public class VespaXMLReaderTestCase {
         VespaXMLFeedReader parser = new VespaXMLFeedReader("src/test/vespaxmlparser/test10.xml", manager);
         {
             FeedOperation op = parser.read();
-            Document doc = op.getDocument();
+            Document doc = op.getDocumentPut().getDocument();
 
             assertEquals(new StringFieldValue("testUrl"), doc.getFieldValue("url"));
             assertEquals(new StringFieldValue("testTitle"), doc.getFieldValue("title"));
@@ -576,7 +576,7 @@ public class VespaXMLReaderTestCase {
         }
         {
             FeedOperation op = parser.read();
-            Document doc = op.getDocument();
+            Document doc = op.getDocumentPut().getDocument();
             assertNotNull(doc);
             assertEquals(new StringFieldValue("testUrl2"), doc.getFieldValue("url"));
         }
@@ -649,7 +649,7 @@ public class VespaXMLReaderTestCase {
         }
         {
             FeedOperation op = parser.read();
-            assertEquals("id:ns:news::http://news10e", op.getRemove().toString());
+            assertEquals("id:ns:news::http://news10e", op.getDocumentRemove().getId().toString());
         }
         {
             // Illegal remove without documentid attribute
@@ -792,11 +792,11 @@ public class VespaXMLReaderTestCase {
         System.err.println(op);
 
         assertEquals(FeedOperation.Type.DOCUMENT, op.getType());
-        assertNull(op.getRemove());
+        assertNull(op.getDocumentRemove());
         assertNull(op.getDocumentUpdate());
-        assertNotNull(op.getDocument());
+        assertNotNull(op.getDocumentPut().getDocument());
 
-        Document doc = op.getDocument();
+        Document doc = op.getDocumentPut().getDocument();
 
         assertEquals("outerdoc", doc.getDataType().getName());
         assertEquals("id:outer:outerdoc::this:is:outer:doc", doc.getId().toString());
-- 
cgit v1.2.3