1 files changed, 11 insertions, 1 deletions
diff --git a/hosted-zone-api/src/test/java/ai/vespa/cloud/SystemInfoTest.java b/hosted-zone-api/src/test/java/ai/vespa/cloud/SystemInfoTest.java
index 9f66d4949bc..6bdb38eb735 100644
--- a/hosted-zone-api/src/test/java/ai/vespa/cloud/SystemInfoTest.java
+++ b/hosted-zone-api/src/test/java/ai/vespa/cloud/SystemInfoTest.java
@@ -14,9 +14,12 @@ public class SystemInfoTest {
     @Test
     public void testSystemInfo() {
         Zone zone = new Zone(Environment.dev, "us-west-1");
+        Cluster cluster = new Cluster(1);
         Node node = new Node(0);
-        SystemInfo info = new SystemInfo(zone, node);
+
+        SystemInfo info = new SystemInfo(zone, cluster, node);
         assertEquals(zone, info.zone());
+        assertEquals(cluster, info.cluster());
         assertEquals(node, info.node());
     }
 
@@ -49,6 +52,13 @@ public class SystemInfoTest {
     }
 
     @Test
+    public void testCluster() {
+        int size = 1;
+        Cluster cluster = new Cluster(size);
+        assertEquals(size, cluster.size());
+    }
+
+    @Test
     public void testNode() {
         int index = 0;
         Node node = new Node(index);