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
|
// Copyright Yahoo. 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.application.api.ApplicationPackage;
import com.yahoo.config.application.api.DeployLogger;
import com.yahoo.config.model.ConfigModelContext;
import com.yahoo.config.model.ConfigModelRepo;
import com.yahoo.config.model.api.HostProvisioner;
import com.yahoo.config.model.builder.xml.XmlHelper;
import com.yahoo.config.model.deploy.DeployState;
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.FileReferencesRepository;
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.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 final HostSystem hostSystem;
private final DeployState deployState;
private final FileReferencesRepository fileReferencesRepository;
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, ApplicationPackage applicationPackage, HostProvisioner provisioner) {
this(rootConfigId, new DeployState.Builder().applicationPackage(applicationPackage)
.modelHostProvisioner(provisioner).build());
}
public MockRoot(String rootConfigId, DeployState deployState) {
super(rootConfigId);
hostSystem = new HostSystem(this, "hostsystem", deployState.getProvisioner(), deployState.getDeployLogger());
this.deployState = deployState;
fileReferencesRepository = new FileReferencesRepository();
}
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).getDeclaredConstructor().newInstance(), configId);
return configClass.getConstructor(builder.getClass()).newInstance(builder);
} catch (ReflectiveOperationException 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");
}
public DeployState getDeployState() {
return deployState;
}
public FileReferencesRepository fileReferencesRepository() { return fileReferencesRepository; }
public HostSystem hostSystem() { 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, false, new ArrayList<>()).
build(deployState, 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;
}
public DeployLogger deployLogger() {
return deployState.getDeployLogger();
}
}
|
