configserver/src/test/java/com/yahoo/vespa/config/server/MockProvisioner.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66

// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.config.server;

import com.yahoo.config.model.api.HostProvisioner;
import com.yahoo.config.provision.ActivationContext;
import com.yahoo.config.provision.ApplicationId;
import com.yahoo.config.provision.ApplicationTransaction;
import com.yahoo.config.provision.Capacity;
import com.yahoo.config.provision.ClusterSpec;
import com.yahoo.config.provision.HostFilter;
import com.yahoo.config.provision.HostSpec;
import com.yahoo.config.provision.ApplicationMutex;
import com.yahoo.config.provision.ProvisionLogger;
import com.yahoo.config.provision.Provisioner;
import com.yahoo.config.provision.exception.LoadBalancerServiceException;

import java.util.Collection;
import java.util.List;

/**
 * @author mpolden
 */
public class MockProvisioner implements Provisioner {

    private boolean transientFailureOnPrepare = false;
    private HostProvisioner hostProvisioner = null;

    public MockProvisioner hostProvisioner(HostProvisioner hostProvisioner) {
        this.hostProvisioner = hostProvisioner;
        return this;
    }

    public MockProvisioner transientFailureOnPrepare() {
        transientFailureOnPrepare = true;
        return this;
    }

    @Override
    public List<HostSpec> prepare(ApplicationId applicationId, ClusterSpec cluster, Capacity capacity, ProvisionLogger logger) {
        if (hostProvisioner != null) {
            return hostProvisioner.prepare(cluster, capacity, logger);
        }
        if (transientFailureOnPrepare) {
            throw new LoadBalancerServiceException("Unable to create load balancer", new Exception("some internal exception"));
        }
        throw new UnsupportedOperationException("This mock does not support prepare");
    }

    @Override
    public void activate(Collection<HostSpec> hosts, ActivationContext context, ApplicationTransaction transaction) {
    }

    @Override
    public void remove(ApplicationTransaction transaction) {
    }

    @Override
    public void restart(ApplicationId application, HostFilter filter) {
    }

    @Override
    public ApplicationMutex lock(ApplicationId application) {
        return new ApplicationMutex(application, () -> {});
    }

}