diff options
author | gjoranv <gv@verizonmedia.com> | 2023-02-16 11:08:59 +0100 |
---|---|---|
committer | gjoranv <gv@verizonmedia.com> | 2023-02-16 11:08:59 +0100 |
commit | 7442356da16d303fb31085a4e1ce82d317a1d822 (patch) | |
tree | a286be966baf38de6f0a468156dda79789eb1bd4 /node-admin/src/test | |
parent | ddad8001995f90793bb406660dd0760c7a0fd06e (diff) |
Add unit test for /nodes/v2/wireguard
Diffstat (limited to 'node-admin/src/test')
-rw-r--r-- | node-admin/src/test/java/com/yahoo/vespa/hosted/node/admin/configserver/noderepository/RealNodeRepositoryTest.java | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/node-admin/src/test/java/com/yahoo/vespa/hosted/node/admin/configserver/noderepository/RealNodeRepositoryTest.java b/node-admin/src/test/java/com/yahoo/vespa/hosted/node/admin/configserver/noderepository/RealNodeRepositoryTest.java index 672698c2e37..87275abf872 100644 --- a/node-admin/src/test/java/com/yahoo/vespa/hosted/node/admin/configserver/noderepository/RealNodeRepositoryTest.java +++ b/node-admin/src/test/java/com/yahoo/vespa/hosted/node/admin/configserver/noderepository/RealNodeRepositoryTest.java @@ -11,6 +11,8 @@ import com.yahoo.config.provision.WireguardKey; import com.yahoo.config.provision.host.FlavorOverrides; import com.yahoo.vespa.hosted.node.admin.configserver.ConfigServerApi; import com.yahoo.vespa.hosted.node.admin.configserver.ConfigServerApiImpl; +import com.yahoo.vespa.hosted.node.admin.task.util.network.VersionedIpAddress; +import com.yahoo.vespa.hosted.node.admin.wireguard.ConfigserverPeer; import com.yahoo.vespa.hosted.provision.restapi.NodesV2ApiHandler; import com.yahoo.vespa.hosted.provision.testutils.ContainerConfig; import org.junit.jupiter.api.AfterEach; @@ -196,4 +198,30 @@ public class RealNodeRepositoryTest { assertEquals(NodeType.config, nodeSpec.type()); } + @Test + void wireguard_peer_config_for_configservers_can_be_retrieved() { + List<ConfigserverPeer> cfgPeers = nodeRepositoryApi.getConfigserverPeers(); + assertEquals(2, cfgPeers.size()); + + var cfg1 = cfgPeers.get(0); + assertEquals("cfg1.yahoo.com", cfg1.hostname().value()); + assertEquals(2, cfg1.ipAddresses().size()); + assertIp(cfg1.ipAddresses().get(0), "127.0.201.1", 4); + assertIp(cfg1.ipAddresses().get(1), "::201:1", 6); + assertEquals("lololololololololololololololololololololoo=", cfg1.publicKey().value()); + + var cfg2 = cfgPeers.get(1); + assertEquals("cfg2.yahoo.com", cfg2.hostname().value()); + assertEquals(2, cfg1.ipAddresses().size()); + assertIp(cfg2.ipAddresses().get(0), "127.0.202.1", 4); + assertIp(cfg2.ipAddresses().get(1), "::202:1", 6); + assertEquals("olololololololololololololololololololololo=", cfg2.publicKey().value()); + + } + + private void assertIp(VersionedIpAddress ip, String expectedIp, int expectedVersion) { + assertEquals(expectedIp, ip.asString()); + assertEquals(expectedVersion, ip.version().version()); + } + } |