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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.node.verification.commons.noderepo;
import org.junit.Before;
import org.junit.Test;
import java.net.InetAddress;
import java.net.UnknownHostException;
import java.util.Arrays;
import static org.junit.Assert.assertArrayEquals;
import static org.junit.Assert.assertEquals;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.doThrow;
import static org.mockito.Mockito.spy;
/**
* @author sgrostad
* @author olaaun
*/
public class IPAddressVerifierTest {
private final String ipv4Address = "10.2.4.8";
private final String ipv6Address = "fdab:0:0:0:0:0:0:1234";
private final NodeSpec nodeSpec = new NodeSpec(1920, 256, 48, true, 10_000, new String[]{ipv4Address, ipv6Address});
private final String hostname = "test123.region.domain.tld";
private IPAddressVerifier ipAddressVerifier = spy(new IPAddressVerifier(hostname, false));
private String ipv4LookupFormat;
private String ipv6LookupFormat;
@Before
public void setup() {
ipv4LookupFormat = "8.4.2.10.in-addr.arpa";
ipv6LookupFormat = "4.3.2.1.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.b.a.d.f.ip6.arpa";
}
@Test
public void getFaultyIpAddresses_with_hostname_resolving_to_other_ips() throws Exception {
doReturn(Arrays.asList(InetAddress.getByName("1.2.3.4"), InetAddress.getByName("fd00::1"))).when(ipAddressVerifier).mockableGetAllByName(hostname);
String[] faultyIpAddresses = ipAddressVerifier.getFaultyIpAddresses(nodeSpec);
String[] expectedFaultyIpAddresses = new String[]{ipv4Address, ipv6Address};
assertArrayEquals(expectedFaultyIpAddresses, faultyIpAddresses);
}
@Test
public void getFaultyIpAddresses_with_hostname_not_resolving_to_ipv4_address() throws Exception {
doReturn(Arrays.asList(InetAddress.getByName(ipv6Address))).when(ipAddressVerifier).mockableGetAllByName(hostname);
doReturn(hostname).when(ipAddressVerifier).reverseLookUp(ipv6LookupFormat);
String[] faultyIpAddresses = ipAddressVerifier.getFaultyIpAddresses(nodeSpec);
String[] expectedFaultyIpAddresses = new String[]{ipv4Address};
assertArrayEquals(expectedFaultyIpAddresses, faultyIpAddresses);
}
@Test
public void getFaultyIpAddresses_with_failing_hostname_resolution() throws Exception {
doThrow(new UnknownHostException("bad hostname")).when(ipAddressVerifier).mockableGetAllByName(hostname);
String[] faultyIpAddresses = ipAddressVerifier.getFaultyIpAddresses(nodeSpec);
String[] expectedFaultyIpAddresses = new String[]{ipv4Address, ipv6Address};
assertArrayEquals(expectedFaultyIpAddresses, faultyIpAddresses);
}
@Test
public void getFaultyIpAddresses_with_hostname_not_resolving_to_ipv6_address() throws Exception {
doReturn(Arrays.asList(InetAddress.getByName(ipv4Address))).when(ipAddressVerifier).mockableGetAllByName(hostname);
doReturn(hostname).when(ipAddressVerifier).reverseLookUp(ipv4LookupFormat);
String[] faultyIpAddresses = ipAddressVerifier.getFaultyIpAddresses(nodeSpec);
String[] expectedFaultyIpAddresses = new String[]{ipv6Address};
assertArrayEquals(expectedFaultyIpAddresses, faultyIpAddresses);
}
@Test
public void getFaultyIpAddresses_should_return_IP_address_when_different_hostname() throws Exception {
String wrongHostName = "www.yahoo.com";
doReturn(Arrays.asList(InetAddress.getByName(ipv4Address), InetAddress.getByName(ipv6Address))).when(ipAddressVerifier).mockableGetAllByName(hostname);
doReturn(hostname).when(ipAddressVerifier).reverseLookUp(ipv4LookupFormat);
doReturn(wrongHostName).when(ipAddressVerifier).reverseLookUp(ipv6LookupFormat);
String[] faultyIpAddresses = ipAddressVerifier.getFaultyIpAddresses(nodeSpec);
String[] expectedFaultyIpAddresses = new String[]{ipv6Address};
assertArrayEquals(expectedFaultyIpAddresses, faultyIpAddresses);
}
@Test
public void getFaultyIpAddresses_should_return_empty_array_when_all_addresses_point_to_correct_hostname() throws Exception {
doReturn(Arrays.asList(InetAddress.getByName(ipv4Address), InetAddress.getByName(ipv6Address))).when(ipAddressVerifier).mockableGetAllByName(hostname);
doReturn(hostname).when(ipAddressVerifier).reverseLookUp(ipv4LookupFormat);
doReturn(hostname).when(ipAddressVerifier).reverseLookUp(ipv6LookupFormat);
String[] faultyIpAddresses = ipAddressVerifier.getFaultyIpAddresses(nodeSpec);
String[] expectedFaultyIpAddresses = new String[]{};
assertArrayEquals(expectedFaultyIpAddresses, faultyIpAddresses);
}
@Test
public void getFaultyIpAddresses_should_return_empty_array_when_reverse_lookup_is_skipped() throws Exception {
ipAddressVerifier = spy(new IPAddressVerifier(hostname, true));
doReturn(Arrays.asList(InetAddress.getByName(ipv4Address), InetAddress.getByName(ipv6Address))).when(ipAddressVerifier).mockableGetAllByName(hostname);
String[] faultyIpAddresses = ipAddressVerifier.getFaultyIpAddresses(nodeSpec);
String[] expectedFaultyIpAddresses = new String[]{};
assertArrayEquals(expectedFaultyIpAddresses, faultyIpAddresses);
}
@Test
public void convertIpv6ToLookupFormat_should_return_properly_converted_ipv6_address() {
String actualConvertedAddress = ipAddressVerifier.convertIpv6ToLookupFormat(ipv6Address);
assertEquals(ipv6LookupFormat, actualConvertedAddress);
}
@Test
public void convertIpv4ToLookupFormat_should_return_properly_converted_ipv6_address() {
String actualConvertedAddress = ipAddressVerifier.convertIpv4ToLookupFormat(ipv4Address);
assertEquals(ipv4LookupFormat, actualConvertedAddress);
}
@Test
public void getFaultyIpAddresses_should_return_empty_array_when_parameters_are_invalid() {
final NodeSpec nodeWithNoIP = new NodeSpec(1920, 256, 48, true, 10_000, new String[0]);
assertEquals(0, ipAddressVerifier.getFaultyIpAddresses(nodeWithNoIP).length);
}
}
|
