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
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.model.application.validation;
import com.yahoo.config.application.api.ApplicationPackage;
import com.yahoo.config.model.NullConfigModelRegistry;
import com.yahoo.config.model.deploy.DeployState;
import com.yahoo.config.model.deploy.TestProperties;
import com.yahoo.config.model.test.MockApplicationPackage;
import com.yahoo.config.provision.Environment;
import com.yahoo.config.provision.RegionName;
import com.yahoo.config.provision.SystemName;
import com.yahoo.config.provision.Zone;
import com.yahoo.vespa.model.VespaModel;
import org.junit.jupiter.api.Test;
import org.xml.sax.SAXException;
import java.io.IOException;
import static com.yahoo.config.provision.Environment.prod;
import static org.junit.jupiter.api.Assertions.*;
/**
* @author gjoranv
*/
public class CloudWatchValidatorTest {
@Test
void cloudwatch_in_public_zones_passes_validation() throws IOException, SAXException {
DeployState deployState = deployState(servicesWithCloudwatch(), true, true);
VespaModel model = new VespaModel(new NullConfigModelRegistry(), deployState);
new CloudWatchValidator().validate(model, deployState);
}
@Test
void cloudwatch_passes_validation_for_self_hosted_vespa() throws IOException, SAXException {
DeployState deployState = deployState(servicesWithCloudwatch(), false, false);
VespaModel model = new VespaModel(new NullConfigModelRegistry(), deployState);
new CloudWatchValidator().validate(model, deployState);
}
@Test
void cloudwatch_in_non_public_zones_fails_validation() throws IOException, SAXException {
DeployState deployState = deployState(servicesWithCloudwatch(), true, false);
VespaModel model = new VespaModel(new NullConfigModelRegistry(), deployState);
try {
new CloudWatchValidator().validate(model, deployState);
fail();
} catch (IllegalArgumentException e) {
assertEquals("CloudWatch cannot be set up for non-public hosted Vespa and must be removed for consumers: [cloudwatch-consumer]",
e.getMessage());
}
}
private static DeployState deployState(String servicesXml, boolean isHosted, boolean isPublic) {
ApplicationPackage app = new MockApplicationPackage.Builder()
.withServices(servicesXml)
.build();
DeployState.Builder builder = new DeployState.Builder()
.applicationPackage(app)
.properties(new TestProperties().setHostedVespa(isHosted));
if (isHosted) {
var system = isPublic ? SystemName.Public : SystemName.main;
builder.zone(new Zone(system, Environment.prod, RegionName.from("foo")));
}
final DeployState deployState = builder.build();
if (isHosted) {
assertTrue(deployState.isHosted(), "Test must emulate a hosted deployment.");
assertEquals(prod, deployState.zone().environment(), "Test must emulate a prod environment.");
}
return deployState;
}
private String servicesWithCloudwatch() {
return String.join("\n",
"<services>",
" <admin version='2.0'>",
" <adminserver hostalias='node1'/>",
" <metrics>",
" <consumer id='cloudwatch-consumer'>",
" <metric id='my-metric'/>",
" <cloudwatch region='us-east-1' namespace='my-namespace' >",
" <credentials access-key-name='my-access-key' ",
" secret-key-name='my-secret-key' />",
" </cloudwatch>",
" </consumer>",
" </metrics>",
" </admin>",
"</services>"
);
}
}
|