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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.provision.testutils;
import com.google.inject.Inject;
import com.yahoo.config.provision.ApplicationId;
import com.yahoo.config.provision.Capacity;
import com.yahoo.config.provision.ClusterSpec;
import com.yahoo.config.provision.Deployer;
import com.yahoo.config.provision.Deployment;
import com.yahoo.config.provision.HostFilter;
import com.yahoo.config.provision.HostSpec;
import com.yahoo.transaction.NestedTransaction;
import com.yahoo.vespa.hosted.provision.provisioning.NodeRepositoryProvisioner;
import java.time.Duration;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Optional;
/**
* @author bratseth
*/
public class MockDeployer implements Deployer {
private final NodeRepositoryProvisioner provisioner;
private final Map<ApplicationId, ApplicationContext> applications;
/** The number of redeployments done to this */
public int redeployments = 0;
@Inject
@SuppressWarnings("unused")
public MockDeployer() {
this(null, Collections.emptyMap());
}
/**
* Create a mock deployer which contains a substitute for an application repository, fullfilled to
* be able to call provision with the right parameters.
*/
public MockDeployer(NodeRepositoryProvisioner provisioner, Map<ApplicationId, ApplicationContext> applications) {
this.provisioner = provisioner;
this.applications = applications;
}
@Override
public Optional<Deployment> deployFromLocalActive(ApplicationId id, Duration timeout) {
return Optional.of(new MockDeployment(provisioner, applications.get(id)));
}
public class MockDeployment implements Deployment {
private final NodeRepositoryProvisioner provisioner;
private final ApplicationContext application;
/** The list of hosts prepared in this. Only set after prepare is called (and a provisioner is assigned) */
private List<HostSpec> preparedHosts = null;
private MockDeployment(NodeRepositoryProvisioner provisioner, ApplicationContext application) {
this.provisioner = provisioner;
this.application = application;
}
@Override
public void prepare() {
preparedHosts = application.prepare(provisioner);
}
@Override
public void activate() {
if (preparedHosts == null)
prepare();
redeployments++;
try (NestedTransaction t = new NestedTransaction()) {
provisioner.activate(t, application.id(), preparedHosts);
t.commit();
}
}
@Override
public void restart(HostFilter filter) {}
}
/** An application context which substitutes for an application repository */
public static class ApplicationContext {
private final ApplicationId id;
private final ClusterSpec cluster;
private final Capacity capacity;
private final int groups;
public ApplicationContext(ApplicationId id, ClusterSpec cluster, Capacity capacity, int groups) {
this.id = id;
this.cluster = cluster;
this.capacity = capacity;
this.groups = groups;
}
public ApplicationId id() { return id; }
/** Returns the spec of the cluster of this application. Only a single cluster per application is supported */
public ClusterSpec cluster() { return cluster; }
private List<HostSpec> prepare(NodeRepositoryProvisioner provisioner) {
return provisioner.prepare(id, cluster, capacity, groups, null);
}
}
}
|