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
|
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.config.server.rpc;
import com.yahoo.config.provision.ApplicationId;
import com.yahoo.config.provision.Version;
import com.yahoo.vespa.config.ConfigKey;
import com.yahoo.vespa.config.GetConfigRequest;
import com.yahoo.vespa.config.protocol.ConfigResponse;
import com.yahoo.vespa.config.server.application.ApplicationSet;
import com.yahoo.vespa.config.server.ReloadHandler;
import com.yahoo.vespa.config.server.RequestHandler;
import com.yahoo.vespa.config.server.tenant.TenantHandlerProvider;
import java.util.*;
/**
* Test utility class
* @author lulf
* @since 5.25
*/
public class MockRequestHandler implements RequestHandler, ReloadHandler, TenantHandlerProvider {
volatile String serverStats = "";
volatile boolean reloadResponse = false;
volatile boolean throwException = false;
public long appGeneration = 0;
private Set<ConfigKey<?>> allConfigs = new HashSet<>();
public volatile ConfigResponse responseConfig = null; // for some v1 mocking
public Map<ApplicationId, ConfigResponse> responses = new LinkedHashMap<>(); // for v2 mocking
private final boolean pretendToHaveLoadedAnyApplication;
public MockRequestHandler() {
this(false);
}
public MockRequestHandler(boolean pretendToHaveLoadedAnyApplication) {
this.pretendToHaveLoadedAnyApplication = pretendToHaveLoadedAnyApplication;
}
@Override
public ConfigResponse resolveConfig(ApplicationId appId, GetConfigRequest req, Optional<Version> vespaVersion) {
if (appId==null) {
checkThrow();
return responseConfig;
}
return responses.get(appId);
}
@Override
public Set<ConfigKey<?>> listConfigs(ApplicationId appId, Optional<Version> vespaVersion, boolean recursive) {
return Collections.emptySet();
}
@Override
public void removeApplication(ApplicationId applicationId) {
}
@Override
public void reloadConfig(ApplicationSet application) {
checkThrow();
}
private void checkThrow() {
if (throwException) {
throw new RuntimeException("foo");
}
}
@Override
public Set<ConfigKey<?>> listNamedConfigs(ApplicationId appId, Optional<Version> vespaVersion, ConfigKey<?> key, boolean recursive) {
return Collections.emptySet();
}
@Override
public Set<String> allConfigIds(ApplicationId appId, Optional<Version> vespaVersion) {
Set<String> ret = new HashSet<>();
for (ConfigKey<?> k : allConfigs) {
ret.add(k.getConfigId());
}
return ret;
}
@Override
public Set<ConfigKey<?>> allConfigsProduced(ApplicationId appId, Optional<Version> vespaVersion) {
return allConfigs;
}
public void setAllConfigs(Set<ConfigKey<?>> allConfigs) {
this.allConfigs = allConfigs;
}
@Override
public boolean hasApplication(ApplicationId appId, Optional<Version> vespaVersion) {
if (pretendToHaveLoadedAnyApplication) return true;
return responses.containsKey(appId);
}
@Override
public ApplicationId resolveApplicationId(String hostName) {
return ApplicationId.defaultId();
}
@Override
public RequestHandler getRequestHandler() {
return this;
}
@Override
public ReloadHandler getReloadHandler() {
return this;
}
}
|