aboutsummaryrefslogtreecommitdiffstats
path: root/node-repository/src/test/java/com/yahoo/vespa/hosted/provision/lb/SharedLoadBalancerServiceTest.java
blob: 98effdca182ae8f8fdb32ddb6d6033db491c8d09 (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
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.provision.lb;

import com.yahoo.config.provision.ApplicationId;
import com.yahoo.config.provision.CloudAccount;
import com.yahoo.config.provision.ClusterSpec;
import com.yahoo.config.provision.HostName;
import com.yahoo.config.provision.ZoneEndpoint;
import org.junit.Test;

import java.util.Optional;
import java.util.Set;

import static org.junit.Assert.assertEquals;

/**
 * @author ogronnesby
 */
public class SharedLoadBalancerServiceTest {

    private final SharedLoadBalancerService loadBalancerService = new SharedLoadBalancerService("vip.example.com");
    private final ApplicationId applicationId = ApplicationId.from("tenant1", "application1", "default");
    private final ClusterSpec.Id clusterId = ClusterSpec.Id.from("qrs1");
    private final Set<Real> reals = Set.of(
            new Real(HostName.of("some.nice.host"), "10.23.56.102"),
            new Real(HostName.of("some.awful.host"), "10.23.56.103")
    );

    @Test
    public void test_create_lb() {
        LoadBalancerSpec spec = new LoadBalancerSpec(applicationId, clusterId, reals,
                                                     ZoneEndpoint.defaultEndpoint, CloudAccount.empty);

        var lb = loadBalancerService.configure(loadBalancerService.provision(spec), spec, false);

        assertEquals(Optional.of(HostName.of("vip.example.com")), lb.hostname());
        assertEquals(Optional.empty(), lb.dnsZone());
        assertEquals(Set.of(), lb.networks());
        assertEquals(Set.of(4443), lb.ports());
    }

    @Test
    public void test_protocol() {
        assertEquals(LoadBalancerService.Protocol.dualstack, loadBalancerService.protocol(false));
    }

}