summaryrefslogtreecommitdiffstats
path: root/config-proxy/src/test/java/com/yahoo/vespa/config/proxy/ConfigTester.java
blob: 87a8764310aa1fd765b1ef9ed7958db80ea5364e (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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
// 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.proxy;

import com.yahoo.jrt.Request;
import com.yahoo.slime.Slime;
import com.yahoo.vespa.config.ConfigKey;
import com.yahoo.vespa.config.ConfigPayload;
import com.yahoo.vespa.config.RawConfig;
import com.yahoo.vespa.config.protocol.CompressionType;
import com.yahoo.vespa.config.protocol.DefContent;
import com.yahoo.vespa.config.protocol.JRTClientConfigRequestV3;
import com.yahoo.vespa.config.protocol.JRTServerConfigRequest;
import com.yahoo.vespa.config.protocol.JRTServerConfigRequestV3;
import com.yahoo.vespa.config.protocol.Payload;
import com.yahoo.vespa.config.protocol.Trace;
import com.yahoo.vespa.config.util.ConfigUtils;

import java.util.Collections;
import java.util.List;
import java.util.Optional;

/**
 * @author bratseth
 */
public class ConfigTester {

    private static final long defaultTimeout = 10000;

    static RawConfig fooConfig;
    static RawConfig barConfig;
    static Payload fooPayload;

    static {
        String defName = "foo";
        String configId = "id";
        String namespace = "bar";
        ConfigKey<?> configKey = new ConfigKey<>(defName, configId, namespace);

        ConfigPayload fooConfigPayload = createConfigPayload("bar", "value");
        fooPayload = Payload.from(fooConfigPayload);

        List<String> defContent = Collections.singletonList("bar string");
        long generation = 1;
        String defMd5 = ConfigUtils.getDefMd5(defContent);
        String configMd5 = ConfigUtils.getMd5(fooConfigPayload);
        fooConfig = new RawConfig(configKey, defMd5, fooPayload, configMd5,
                                  generation, false, false, defContent, Optional.empty());

        String defName2 = "bar";
        barConfig = new RawConfig(new ConfigKey<>(defName2, configId, namespace), defMd5, fooPayload, configMd5,
                                  generation, false, false, defContent, Optional.empty());
    }

    JRTServerConfigRequest createRequest(RawConfig config) {
        return createRequest(config, defaultTimeout);
    }

    JRTServerConfigRequest createRequest(RawConfig config, long timeout) {
        return createRequest(config.getName(), config.getConfigId(), config.getNamespace(),
                             config.getConfigMd5(), config.getDefMd5(), config.getGeneration(), timeout);
    }

    JRTServerConfigRequest createRequest(String configName, String configId, String namespace, long timeout) {
        return createRequest(configName, configId, namespace, "", null, 0, timeout);
    }

    private JRTServerConfigRequest createRequest(String configName, String configId, String namespace, String md5, String defMd5, long generation, long timeout) {
        Request request = JRTClientConfigRequestV3.
                createWithParams(new ConfigKey<>(configName, configId, namespace, defMd5, null), DefContent.fromList(Collections.emptyList()),
                                 "fromHost", md5, generation, timeout, Trace.createDummy(), CompressionType.UNCOMPRESSED,
                                 Optional.empty()).getRequest();
        return JRTServerConfigRequestV3.createFromRequest(request);
    }

    static ConfigPayload createConfigPayload(String key, String value) {
        Slime slime = new Slime();
        slime.setObject().setString(key, value);
        return new ConfigPayload(slime);
    }

}