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
113
114
115
116
117
118
119
120
121
|
package ai.vespa.hosted.cd;
import ai.vespa.hosted.api.ControllerHttpClient;
import com.yahoo.config.provision.ApplicationId;
import com.yahoo.config.provision.SystemName;
import com.yahoo.slime.ArrayTraverser;
import com.yahoo.slime.Inspector;
import com.yahoo.slime.JsonDecoder;
import com.yahoo.slime.ObjectTraverser;
import com.yahoo.config.provision.zone.ZoneId;
import com.yahoo.slime.Slime;
import java.net.URI;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;
/**
* The place to obtain environment-dependent configuration for the current test run.
*
* If the system property 'vespa.test.config' is set, this class attempts to parse config
* from a JSON file at that location -- otherwise, attempts to access the config will return null.
*
* @author jvenstad
*/
public class TestConfig {
private static TestConfig theConfig;
private final ApplicationId application;
private final ZoneId zone;
private final SystemName system;
private final Map<ZoneId, Deployment> deployments;
private TestConfig(ApplicationId application, ZoneId zone, SystemName system, Map<ZoneId, Deployment> deployments) {
this.application = application;
this.zone = zone;
this.system = system;
this.deployments = Map.copyOf(deployments);
}
/** Returns the config for this test, or null if it has not been provided. */
public static synchronized TestConfig get() {
if (theConfig == null) {
String configPath = System.getProperty("vespa.test.config");
theConfig = configPath != null ? fromFile(configPath) : fromController();
}
return theConfig;
}
/** Returns the full id of the application to be tested. */
public ApplicationId application() { return application; }
/** Returns the zone of the deployment to test. */
public ZoneId zone() { return zone; }
/** Returns an immutable view of all configured endpoints for each zone of the application to test. */
public Map<ZoneId, Deployment> allDeployments() { return deployments; }
/** Returns the deployment to test in this test runtime. */
public Deployment deploymentToTest() { return deployments.get(zone); }
/** Returns the system this is run against. */
public SystemName system() { return system; }
static TestConfig fromFile(String path) {
if (path == null)
return null;
try {
return fromJson(Files.readAllBytes(Paths.get(path)));
}
catch (Exception e) {
throw new IllegalArgumentException("Failed reading config from '" + path + "'!", e);
}
}
static TestConfig fromController() {
ApplicationId id = ApplicationId.from(requireNonBlankProperty("tenant"),
requireNonBlankProperty("application"),
getNonBlankProperty("instance").orElse("default"));
URI endpoint = URI.create(requireNonBlankProperty("endpoint"));
Path privateKeyFile = Paths.get(requireNonBlankProperty("privateKeyFile"));
Optional<Path> certificateFile = getNonBlankProperty("certificateFile").map(Paths::get);
ControllerHttpClient controller = certificateFile.isPresent()
? ControllerHttpClient.withKeyAndCertificate(endpoint, privateKeyFile, certificateFile.get())
: ControllerHttpClient.withSignatureKey(endpoint, privateKeyFile, id);
return null;
}
static TestConfig fromJson(byte[] jsonBytes) {
Inspector config = new JsonDecoder().decode(new Slime(), jsonBytes).get();
ApplicationId application = ApplicationId.fromSerializedForm(config.field("application").asString());
ZoneId zone = ZoneId.from(config.field("zone").asString());
SystemName system = SystemName.from(config.field("system").asString());
Map<ZoneId, Deployment> endpoints = new HashMap<>();
config.field("endpoints").traverse((ObjectTraverser) (zoneId, endpointArray) -> {
List<URI> uris = new ArrayList<>();
endpointArray.traverse((ArrayTraverser) (__, uri) -> uris.add(URI.create(uri.asString())));
endpoints.put(ZoneId.from(zoneId), null); // TODO jvenstad
});
return new TestConfig(application, zone, system, endpoints);
}
static Optional<String> getNonBlankProperty(String name) {
return Optional.ofNullable(System.getProperty(name)).filter(value -> ! value.isBlank());
}
static String requireNonBlankProperty(String name) {
return getNonBlankProperty(name).orElseThrow(() -> new IllegalStateException("Missing required property '" + name + "'"));
}
}
|