blob: 81f5d303d563adad26c5acd89e29bb75a538141a (
plain) (
blame)
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 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.config.model.test;
import com.google.common.annotations.Beta;
import com.yahoo.config.ConfigInstance;
import com.yahoo.config.model.ConfigModel;
import com.yahoo.vespa.model.HostResource;
import com.yahoo.vespa.model.VespaModel;
import java.util.List;
/**
* Test utility class that provides many methods for inspecting the state of a completely built model
*
* @author Ulf Lilleengen
* @since 5.1
*/
@Beta
public class TestRoot {
private final VespaModel model;
TestRoot(VespaModel model) {
this.model = model;
}
/**
* Get a list of all config models of a particular type.
*
* @param clazz The class of the models to find.
* @return A list of models of given type.
*/
public <MODEL extends ConfigModel> List<MODEL> getConfigModels(Class<MODEL> clazz) {
return model.configModelRepo().getModels(clazz);
}
/**
* Ask model to populate builder with config for a given config id. This method gives the same config as the
* configserver would return to its clients.
* @param builder The builder to populate
* @param configId The config id of the producer to ask for config.
* @return the same builder.
*/
public <BUILDER extends ConfigInstance.Builder> BUILDER getConfig(BUILDER builder, String configId) {
return (BUILDER) model.getConfig(builder, configId);
}
/**
* Request config of a given type and id. This method gives the same config as the configserver would return to
* its clients.
*
* @param clazz Type of config to request.
* @param configId The config id of the producer to ask for config.
* @return A config object of the appropriate type with config values set.
*/
public <CONFIGTYPE extends ConfigInstance> CONFIGTYPE getConfig(Class<CONFIGTYPE> clazz, String configId) {
return model.getConfig(clazz, configId);
}
/**
* Retrieve the hosts available in this model. Useful to verify that hostnames are set correctly etc.
*
* @return A list of hosts.
*/
public List<HostResource> getHosts() {
return model.hostSystem().getHosts();
}
}
|