node-admin/src/test/java/com/yahoo/vespa/hosted/node/admin/wireguard/WireguardPeerTest.java


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

// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.node.admin.wireguard;

import com.yahoo.config.provision.HostName;
import com.yahoo.config.provision.WireguardKey;
import com.yahoo.config.provision.WireguardKeyWithTimestamp;
import com.yahoo.vespa.hosted.node.admin.task.util.network.VersionedIpAddress;
import org.junit.jupiter.api.Test;

import java.time.Instant;
import java.util.List;
import java.util.stream.Stream;

import static org.junit.jupiter.api.Assertions.assertEquals;

/**
 * @author gjoranv
 */
public class WireguardPeerTest {

    @Test
    void peers_are_sorted_by_hostname_ascending() {
        List<WireguardPeer> peers = Stream.of(
                peer("b"),
                peer("a"),
                peer("c")
        ).sorted().toList();

        assertEquals("a", peers.get(0).hostname().value());
        assertEquals("b", peers.get(1).hostname().value());
        assertEquals("c", peers.get(2).hostname().value());
    }

    private static WireguardPeer peer(String hostname) {
        return new WireguardPeer(HostName.of(hostname), List.of(VersionedIpAddress.from("::1:1")),
                                 new WireguardKeyWithTimestamp(WireguardKey.generateRandomForTesting(), Instant.EPOCH));
    }

}