aboutsummaryrefslogtreecommitdiffstats
path: root/node-admin/src/test/java/com/yahoo/vespa/hosted/node/admin/task/util/network/VersionedIpAddressTest.java
blob: 69d5c6f2c31edb51158871b775e57ca9da43c5d7 (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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
// 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.task.util.network;

import org.junit.jupiter.api.Test;

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

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

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

    @Test
    void ip4_address_can_be_generated_from_string() {
        var ip4 = VersionedIpAddress.from("10.0.0.1");
        assertEquals(IPVersion.IPv4, ip4.version());
        assertEquals("10.0.0.1", ip4.asString());
    }

    @Test
    void ip6_address_can_be_generated_from_string() {
        var ip6 = VersionedIpAddress.from("::1");
        assertEquals(IPVersion.IPv6, ip6.version());
        assertEquals("::1", ip6.asString());
    }

    @Test
    void they_are_sorted_by_version_then_by_address() {
        var ip4 = VersionedIpAddress.from("10.0.0.1");
        var ip4_2 = VersionedIpAddress.from("127.0.0.1");
        var ip6 = VersionedIpAddress.from("::1");
        var ip6_2 = VersionedIpAddress.from("::2");

        var sorted = Stream.of(ip4_2, ip6, ip4, ip6_2)
                .sorted()
                .toList();
        assertEquals(List.of(ip6, ip6_2, ip4, ip4_2), sorted);
    }

    @Test
    void endpoint_with_port_is_generated_correctly_for_both_versions() {
        var ip4 = VersionedIpAddress.from("10.0.0.1");
        var ip6 = VersionedIpAddress.from("::1");

        assertEquals("10.0.0.1:8080", ip4.asEndpoint(8080));
        assertEquals("[::1]:8080", ip6.asEndpoint(8080));
    }

    @Test
    void equals_and_hashCode_are_implemented() {
        var one = VersionedIpAddress.from("::1");
        var two = VersionedIpAddress.from("::2");
        var local = VersionedIpAddress.from("127.0.0.1");
        var ten = VersionedIpAddress.from("10.0.0.1");
        assertEquals(one, VersionedIpAddress.from("::1"));
        assertNotEquals(one, two);
        assertNotEquals(one, local);
        assertNotEquals(one, ten);

        assertEquals(local, VersionedIpAddress.from("127.0.0.1"));
        assertNotEquals(local, two);
        assertNotEquals(local, 10);
    }

}