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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
|
// 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.yahoo.config.ConfigInstance;
import com.yahoo.config.model.ConfigModelContext;
import com.yahoo.config.model.ConfigModelRepo;
import com.yahoo.config.application.api.ApplicationPackage;
import com.yahoo.config.model.application.provider.BaseDeployLogger;
import com.yahoo.config.application.api.DeployLogger;
import com.yahoo.config.model.deploy.DeployState;
import com.yahoo.config.model.builder.xml.XmlHelper;
import com.yahoo.config.model.producer.AbstractConfigProducer;
import com.yahoo.config.model.producer.AbstractConfigProducerRoot;
import com.yahoo.text.XML;
import com.yahoo.vespa.model.ConfigProducer;
import com.yahoo.vespa.model.HostSystem;
import com.yahoo.vespa.model.admin.Admin;
import com.yahoo.vespa.model.builder.xml.dom.DomAdminV2Builder;
import com.yahoo.vespa.model.filedistribution.FileDistributionConfigProducer;
import com.yahoo.vespa.model.filedistribution.FileDistributor;
import org.w3c.dom.Document;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;
import java.io.IOException;
import java.io.StringReader;
import java.lang.reflect.InvocationTargetException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Set;
/**
* Use for testing. Use as parent for the config producer(s) you want to test, to test
* only a subtree of the producers.
*
* @author gjoranv
*/
// TODO: mockRoot instances can probably be replaced by VespaModel.createIncomplete
public class MockRoot extends AbstractConfigProducerRoot {
private static final long serialVersionUID = 1L;
private HostSystem hostSystem;
private final DeployState deployState;
private FileDistributor fileDistributor;
private Admin admin;
public MockRoot() {
this("");
}
public MockRoot(String rootConfigId) {
this(rootConfigId, new MockApplicationPackage.Builder().build());
}
public MockRoot(String rootConfigId, ApplicationPackage applicationPackage) {
this(rootConfigId, new DeployState.Builder().applicationPackage(applicationPackage).build());
}
public MockRoot(String rootConfigId, DeployState deployState) {
super(rootConfigId);
hostSystem = new HostSystem(this, "hostsystem", deployState.getProvisioner());
this.deployState = deployState;
fileDistributor = new FileDistributor(deployState.getFileRegistry());
}
public FileDistributionConfigProducer getFileDistributionConfigProducer() {
return null;
}
@Override
public ConfigModelRepo configModelRepo() {
return new ConfigModelRepo();
}
public Set<String> getConfigIds() {
return Collections.unmodifiableSet(id2producer.keySet());
}
@Override
public ConfigInstance.Builder getConfig(ConfigInstance.Builder builder, String configId) {
ConfigProducer cp = id2producer.get(configId);
if (cp == null) return null;
cp.cascadeConfig(builder);
cp.addUserConfig(builder);
return builder;
}
@SuppressWarnings("unchecked")
public <T extends ConfigInstance> T getConfig(Class<T> configClass, String configId) {
try {
ConfigInstance.Builder builder = getConfig(getBuilder(configClass).newInstance(), configId);
return configClass.getConstructor(builder.getClass()).newInstance(builder);
} catch (InstantiationException | NoSuchMethodException | InvocationTargetException | IllegalAccessException e) {
throw new RuntimeException(e);
}
}
public ConfigProducer getProducer(String configId) {
return id2producer.get(configId);
}
@SuppressWarnings("unchecked")
public static <T extends ConfigInstance> Class<? extends ConfigInstance.Builder> getBuilder(Class<T> configClass) {
for (Class<?> memberClass : configClass.getClasses()) {
if (memberClass.getSimpleName().equals("Builder"))
return (Class<? extends ConfigInstance.Builder>) memberClass;
}
throw new RuntimeException("Missing builder");
}
@Override
public DeployState getDeployState() {
return deployState;
}
public FileDistributor getFileDistributor() {
return fileDistributor;
}
public HostSystem getHostSystem() {
return hostSystem;
}
public void addDescendant(String configId, AbstractConfigProducer descendant) {
if (id2producer.containsKey(configId)) {
throw new RuntimeException
("Config ID '" + configId + "' cannot be reserved by an instance of class '" +
descendant.getClass().getName() + "' since it is already used by an instance of class '" +
id2producer.get(configId).getClass().getName() + "'");
}
id2producer.put(configId, descendant);
}
@Override
public void addChild(AbstractConfigProducer abstractConfigProducer) {
super.addChild(abstractConfigProducer);
}
public final void setAdmin(String xml) {
String servicesXml =
"<?xml version='1.0' encoding='utf-8' ?>" +
"<services>" + xml + "</services>";
try {
Document doc = XmlHelper.getDocumentBuilder().parse(new InputSource(new StringReader(servicesXml)));
setAdmin(new DomAdminV2Builder(ConfigModelContext.ApplicationType.DEFAULT, deployState.getFileRegistry(), false, new ArrayList<>()).
build(this, XML.getChildren(doc.getDocumentElement(), "admin").get(0)));
} catch (SAXException | IOException e) {
throw new RuntimeException(e);
}
}
public final void setAdmin(Admin admin) {
this.admin = admin;
}
@Override
public final Admin getAdmin() {
return admin;
}
@Override
public DeployLogger deployLogger() {
return new BaseDeployLogger();
}
}
|
