1 files changed, 51 insertions, 0 deletions
diff --git a/config-provisioning/src/test/java/com/yahoo/config/provision/AllocatedHostsTest.java b/config-provisioning/src/test/java/com/yahoo/config/provision/AllocatedHostsTest.java
new file mode 100644
index 00000000000..54ce90d76fc
--- /dev/null
+++ b/config-provisioning/src/test/java/com/yahoo/config/provision/AllocatedHostsTest.java
@@ -0,0 +1,51 @@
+// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.config.provision;
+
+import org.junit.Test;
+
+import java.io.IOException;
+import java.util.LinkedHashSet;
+import java.util.Optional;
+import java.util.Set;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
+
+/**
+ * @author Ulf Lilleengen
+ */
+public class AllocatedHostsTest {
+
+    private final HostSpec h1 = new HostSpec("host1", Optional.empty());
+    private final HostSpec h2 = new HostSpec("host2", Optional.empty());
+    private final HostSpec h3 = new HostSpec("host3", Optional.of(ClusterMembership.from("container/test/0", com.yahoo.component.Version.fromString("6.73.1"))));
+
+    @Test
+    public void testProvisionInfoSerialization() throws IOException {
+        Set<HostSpec> hosts = new LinkedHashSet<>();
+        hosts.add(h1);
+        hosts.add(h2);
+        hosts.add(h3);
+        AllocatedHosts info = AllocatedHosts.withHosts(hosts);
+        assertProvisionInfo(info);
+    }
+
+    private void assertProvisionInfo(AllocatedHosts info) throws IOException {
+        AllocatedHosts serializedInfo = AllocatedHosts.fromJson(info.toJson(), Optional.empty());
+        assertEquals(info.getHosts().size(), serializedInfo.getHosts().size());
+        assertTrue(serializedInfo.getHosts().contains(h1));
+        assertTrue(serializedInfo.getHosts().contains(h2));
+        assertTrue(serializedInfo.getHosts().contains(h3));
+        assertTrue(!getHost(h1.hostname(), serializedInfo.getHosts()).membership().isPresent());
+        assertEquals("container/test/0", getHost(h3.hostname(), serializedInfo.getHosts()).membership().get().stringValue());
+        assertEquals(h3.membership().get().cluster().vespaVersion(), getHost(h3.hostname(), serializedInfo.getHosts()).membership().get().cluster().vespaVersion());
+    }
+
+    private HostSpec getHost(String hostname, Set<HostSpec> hosts) {
+        for (HostSpec host : hosts)
+            if (host.hostname().equals(hostname))
+                return host;
+        throw new IllegalArgumentException("No host " + hostname + " is present");
+    }
+
+}