1 files changed, 52 insertions, 0 deletions
diff --git a/clustercontroller-standalone/src/test/java/com/yahoo/vespa/clustercontroller/standalone/ClusterControllerConfigFetcherTest.java b/clustercontroller-standalone/src/test/java/com/yahoo/vespa/clustercontroller/standalone/ClusterControllerConfigFetcherTest.java
new file mode 100644
index 00000000000..32096953b76
--- /dev/null
+++ b/clustercontroller-standalone/src/test/java/com/yahoo/vespa/clustercontroller/standalone/ClusterControllerConfigFetcherTest.java
@@ -0,0 +1,52 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.vespa.clustercontroller.standalone;
+
+public class ClusterControllerConfigFetcherTest extends ClusterControllerTest {
+    public void testSimple() throws Exception {
+        setFleetControllerConfigProperty();
+        setSlobrokConfigProperty();
+        addFleetControllerConfig(2, 1);
+        addSlobrokConfig();
+        addDistributionConfig();
+        addZookeepersConfig();
+        ClusterControllerConfigFetcher configFetcher = new ClusterControllerConfigFetcher();
+        configFetcher.getOptions();
+        configFetcher.updated(100);
+        assertEquals(1, configFetcher.getGeneration());
+        configFetcher.close();
+    }
+
+    public void testInitialConfigFailure() throws Exception {
+        setFleetControllerConfigProperty();
+        setSlobrokConfigProperty();
+        addFleetControllerConfig(2, 1);
+        addSlobrokConfig();
+        addDistributionConfig();
+        addZookeepersConfig();
+        try{
+            ClusterControllerConfigFetcher configFetcher = new ClusterControllerConfigFetcher() {
+                boolean configReady() {
+                    return false;
+                }
+            };
+            fail("Control should not reach here");
+        } catch (IllegalStateException e) {
+            assertEquals("Initial configuration failed.", e.getMessage());
+        }
+    }
+
+    public void testConfigUpdate() throws Exception {
+        setFleetControllerConfigProperty();
+        setSlobrokConfigProperty();
+        addFleetControllerConfig(2, 1);
+        addSlobrokConfig();
+        addDistributionConfig();
+        addZookeepersConfig();
+        ClusterControllerConfigFetcher configFetcher = new ClusterControllerConfigFetcher() {
+            boolean configUpdated(long millis) {
+                return true;
+            }
+        };
+        configFetcher.updated(1000);
+    }
+}