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
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.config;
import com.yahoo.config.FileReference;
import com.yahoo.config.ModelReference;
import com.yahoo.config.ResolvedTypesConfig;
import com.yahoo.config.UrlReference;
import org.junit.Test;
import java.io.File;
import java.nio.file.Path;
import java.time.Duration;
import java.util.Optional;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.fail;
/**
* @author bratseth
*/
public class ConfigPayloadApplierTest {
@Test
public void testAllConfigApplierReferenceTypes() {
var configBuilder = new ResolvedTypesConfig.Builder();
var applier = new ConfigPayloadApplier<>(configBuilder, new MockAcquirer(), new MockDownloader());
var inputConfig = new ResolvedTypesConfig.Builder();
inputConfig.myPath(new FileReference("myPath.txt"));
inputConfig.myUrl(new UrlReference("myUrl.txt"));
inputConfig.myModel(ModelReference.unresolved(Optional.empty(),
Optional.of(new UrlReference("myUrl.txt")),
Optional.of(new FileReference("myPath.txt"))));
applier.applyPayload(ConfigPayload.fromInstance(inputConfig.build()));
var config = configBuilder.build();
assertEndsWith("resolvedPath/myPath.txt", config.myPath().toString());
assertEndsWith("resolvedUrl/myUrl.txt", config.myUrl().toString());
assertEndsWith("resolvedUrl/myUrl.txt", config.myModel().toString());
}
@Test
public void testModelWithUrlOnly() {
var configBuilder = new ResolvedTypesConfig.Builder();
var applier = new ConfigPayloadApplier<>(configBuilder, new MockAcquirer(), new MockDownloader());
var inputConfig = new ResolvedTypesConfig.Builder();
inputConfig.myPath(new FileReference("myPath.txt"));
inputConfig.myUrl(new UrlReference("myUrl.txt"));
inputConfig.myModel(ModelReference.valueOf("my-id myUrl.txt \"\""));
applier.applyPayload(ConfigPayload.fromInstance(inputConfig.build()));
var config = configBuilder.build();
assertEndsWith("resolvedUrl/myUrl.txt", config.myModel().toString());
}
@Test
public void testModelWithPathOnly() {
var configBuilder = new ResolvedTypesConfig.Builder();
var applier = new ConfigPayloadApplier<>(configBuilder, new MockAcquirer(), new MockDownloader());
var inputConfig = new ResolvedTypesConfig.Builder();
inputConfig.myPath(new FileReference("myPath.txt"));
inputConfig.myUrl(new UrlReference("myUrl.txt"));
inputConfig.myModel(ModelReference.valueOf("my-id \"\" myPath.txt"));
applier.applyPayload(ConfigPayload.fromInstance(inputConfig.build()));
var config = configBuilder.build();
assertEndsWith("resolvedPath/myPath.txt", config.myModel().toString());
}
private void assertEndsWith(String ending, String string) {
String assertingThat = "'" + string + "' ends with '" + ending + "'";
try {
assertEquals(assertingThat, ending, string.substring(string.length() - ending.length()));
}
catch (StringIndexOutOfBoundsException e) {
fail(assertingThat);
}
}
private static class MockAcquirer implements ConfigTransformer.PathAcquirer {
@Override
public Path getPath(FileReference fileReference) {
return Path.of("resolvedPath", fileReference.value());
}
}
private static class MockDownloader extends UrlDownloader {
@Override
public File waitFor(UrlReference urlReference, Duration timeout) {
return new File("resolvedUrl", urlReference.value());
}
}
}
|