Add tests for StatBucketMessage

1 files changed, 22 insertions, 0 deletions

diff --git a/documentapi/src/test/java/com/yahoo/documentapi/messagebus/protocol/test/Messages60TestCase.java b/documentapi/src/test/java/com/yahoo/documentapi/messagebus/protocol/test/Messages60TestCase.java
index 7655d170223..50e5ba0fe19 100644
--- a/documentapi/src/test/java/com/yahoo/documentapi/messagebus/protocol/test/Messages60TestCase.java
+++ b/documentapi/src/test/java/com/yahoo/documentapi/messagebus/protocol/test/Messages60TestCase.java
@@ -6,6 +6,7 @@ import com.yahoo.document.BucketId;
 import com.yahoo.document.select.OrderingSpecification;
 import com.yahoo.documentapi.messagebus.protocol.CreateVisitorMessage;
 import com.yahoo.documentapi.messagebus.protocol.DocumentProtocol;
+import com.yahoo.documentapi.messagebus.protocol.StatBucketMessage;
 import com.yahoo.text.Utf8;
 
 import java.util.Map;
@@ -77,6 +78,27 @@ public class Messages60TestCase extends Messages52TestCase {
         }
     }
 
+    public class testStatBucketMessage implements RunnableTest {
+
+        private static final String BUCKET_SPACE = "andrei";
+
+        @Override
+        public void run() {
+            StatBucketMessage msg = new StatBucketMessage(new BucketId(16, 123), "id.user=123");
+            msg.setLoadType(null);
+            msg.setBucketSpace(BUCKET_SPACE);
+            assertEquals(BASE_MESSAGE_LENGTH + 27 + serializedLength(BUCKET_SPACE), serialize("StatBucketMessage", msg));
+
+            for (Language lang : LANGUAGES) {
+                msg = (StatBucketMessage)deserialize("StatBucketMessage", DocumentProtocol.MESSAGE_STATBUCKET, lang);
+                assertEquals(new BucketId(16, 123), msg.getBucketId());
+                assertEquals("id.user=123", msg.getDocumentSelection());
+                assertEquals("default", msg.getLoadType().getName());
+                assertEquals(BUCKET_SPACE, msg.getBucketSpace());
+            }
+        }
+    }
+
     // TODO want to test that non-default bucket space fails to encode with old version
 
 }