com.yahoo.vespa.hosted.controller.persistence.MockCuratorDb Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of controller-server Show documentation

There is a newer version: 8.253.3

// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.controller.persistence;

import com.google.inject.Inject;
import com.yahoo.vespa.curator.mock.MockCurator;

import java.time.Duration;

/**
 * A curator db backed by a mock curator.
 * 
 * @author bratseth
 */
@SuppressWarnings("unused") // injected
public class MockCuratorDb extends CuratorDb {

    private final MockCurator curator;

    @Inject
    public MockCuratorDb() {
        this("test-controller:2222");
    }

    public MockCuratorDb(String zooKeeperEnsembleConnectionSpec) {
        this(new MockCurator() {
            @Override
            public String zooKeeperEnsembleConnectionSpec() {
                return zooKeeperEnsembleConnectionSpec;
            }
        });
    }

    public MockCuratorDb(MockCurator curator) {
        super(curator, Duration.ofMillis(100));
        this.curator = curator;
    }

    public MockCurator curator() { return curator; }

}