configuration-as-code-plugin/integrations/src/test/java/io/jenkins/plugins/casc/EC2CloudTest.java at 91ef09b4e6cd7eb2518a25c99e5f4606e9b6b814 · jenkinsci/configuration-as-code-plugin · GitHub

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
package io.jenkins.plugins.casc;

import static org.hamcrest.CoreMatchers.containsString;
import static org.hamcrest.CoreMatchers.equalTo;
import static org.hamcrest.CoreMatchers.is;
import static org.hamcrest.CoreMatchers.notNullValue;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.collection.IsCollectionWithSize.hasSize;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;

import hudson.model.labels.LabelAtom;
import hudson.plugins.ec2.AMITypeData;
import hudson.plugins.ec2.EC2Cloud;
import hudson.plugins.ec2.SlaveTemplate;
import hudson.plugins.ec2.UnixData;
import io.jenkins.plugins.casc.misc.ConfiguredWithReadme;
import io.jenkins.plugins.casc.misc.JenkinsConfiguredWithReadmeRule;
import java.util.List;
import jenkins.model.Jenkins;
import org.junit.Rule;
import org.junit.Test;
import software.amazon.awssdk.services.ec2.model.InstanceType;

public class EC2CloudTest {

    @Rule
    public JenkinsConfiguredWithReadmeRule j = new JenkinsConfiguredWithReadmeRule();

    @Test
    @ConfiguredWithReadme("ec2/README.md")
    public void configure_ec2_cloud() {
        final EC2Cloud ec2Cloud = (EC2Cloud) Jenkins.get().getCloud("ec2");
        assertNotNull(ec2Cloud);

        assertTrue(ec2Cloud.isUseInstanceProfileForCredentials());
        assertThat(ec2Cloud.getSshKeysCredentialsId(), is("ssh-key-credential-id"));
        final List<SlaveTemplate> templates = ec2Cloud.getTemplates();
        assertThat(templates, hasSize(2));

        SlaveTemplate slaveTemplate = templates.get(0);
        assertThat(slaveTemplate.getDisplayName(), containsString("Auto configured EC2 Agent Small"));
        assertFalse(slaveTemplate.getAssociatePublicIp());
        assertFalse(slaveTemplate.isConnectBySSHProcess());
        assertFalse(slaveTemplate.deleteRootOnTermination);
        assertFalse(slaveTemplate.ebsOptimized);
        assertFalse(slaveTemplate.monitoring);
        assertFalse(slaveTemplate.stopOnTerminate);
        assertFalse(slaveTemplate.useEphemeralDevices);
        assertThat(slaveTemplate.type, is(InstanceType.T2_SMALL.toString()));
        assertThat(slaveTemplate.getAmi(), equalTo("ami-0c6bb742864ffa3f3"));
        assertThat(slaveTemplate.getLabelString(), containsString("Small"));
        assertThat(slaveTemplate.getLabelSet(), is(notNullValue()));
        assertThat(slaveTemplate.remoteFS, equalTo("/home/ec2-user"));
        assertThat(slaveTemplate.getRemoteAdmin(), equalTo("ec2-user"));
        assertThat(slaveTemplate.zone, equalTo("us-east-1"));
        assertThat(slaveTemplate.getSecurityGroupString(), equalTo("some-group"));

        // fails here without mode specified
        assertTrue(ec2Cloud.canProvision(new LabelAtom("Small")));

        // Checks that the AMI type is Unix and configured
        AMITypeData amiType = slaveTemplate.getAmiType();
        assertTrue(amiType.isUnix());
        assertTrue(amiType instanceof UnixData);
        UnixData unixData = (UnixData) amiType;
        assertThat(unixData.getRootCommandPrefix(), equalTo("sudo"));
        assertThat(unixData.getSlaveCommandPrefix(), equalTo("sudo -u jenkins"));
        assertThat(unixData.getSshPort(), equalTo("61120"));

        slaveTemplate = templates.get(1);
        assertThat(slaveTemplate.getDisplayName(), containsString("Auto configured EC2 Agent Large"));
        assertFalse(slaveTemplate.getAssociatePublicIp());
        assertFalse(slaveTemplate.isConnectBySSHProcess());
        assertFalse(slaveTemplate.deleteRootOnTermination);
        assertFalse(slaveTemplate.ebsOptimized);
        assertFalse(slaveTemplate.monitoring);
        assertFalse(slaveTemplate.stopOnTerminate);
        assertFalse(slaveTemplate.useEphemeralDevices);
        assertThat(slaveTemplate.type, is(InstanceType.T2_XLARGE.toString()));
        assertThat(slaveTemplate.getAmi(), equalTo("ami-0c6bb742864ffa3f3"));
        assertThat(slaveTemplate.getLabelString(), containsString("Large"));
        assertThat(slaveTemplate.getLabelSet(), is(notNullValue()));
        assertThat(slaveTemplate.remoteFS, equalTo("/home/ec2-user"));
        assertThat(slaveTemplate.getRemoteAdmin(), equalTo("ec2-user"));
        assertThat(slaveTemplate.zone, equalTo("us-east-1"));
        assertThat(slaveTemplate.getSecurityGroupString(), equalTo("some-group"));

        // fails here without mode specified
        assertTrue(ec2Cloud.canProvision(new LabelAtom("Large")));

        // Checks that the AMI type is Unix and configured
        amiType = slaveTemplate.getAmiType();
        assertTrue(amiType.isUnix());
        assertTrue(amiType instanceof UnixData);
        unixData = (UnixData) amiType;
        assertThat(unixData.getRootCommandPrefix(), equalTo("sudo"));
        assertThat(unixData.getSlaveCommandPrefix(), equalTo("sudo -u jenkins"));
        assertThat(unixData.getSshPort(), equalTo("61120"));
    }
}