diff options
author | gjoranv <gv@verizonmedia.com> | 2023-02-13 17:46:05 +0100 |
---|---|---|
committer | gjoranv <gv@verizonmedia.com> | 2023-02-13 17:57:21 +0100 |
commit | 5fac07d3d2fbe80b3c7938123924c55ccaeb647b (patch) | |
tree | d93a8bcca47c2c04d2c4de31d3f1eec0dfbf5a9e /node-admin/src/test | |
parent | 3da5e19509fcc10d4cdbcc49747cacc7cac2ae2e (diff) |
Allow getting and patching wg pubkey from/to node repo.
+ Add missing 'trustStore' to NodeAttributes.hashCode
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 | 19 |
1 files changed, 16 insertions, 3 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 661d3e6073e..672698c2e37 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 @@ -7,6 +7,7 @@ import com.yahoo.config.provision.CloudAccount; import com.yahoo.config.provision.DockerImage; import com.yahoo.config.provision.NodeResources; import com.yahoo.config.provision.NodeType; +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; @@ -24,7 +25,10 @@ import java.util.List; import java.util.Optional; import java.util.Set; -import static org.junit.jupiter.api.Assertions.*; +import static org.junit.jupiter.api.Assertions.assertEquals; +import static org.junit.jupiter.api.Assertions.assertFalse; +import static org.junit.jupiter.api.Assertions.assertTrue; +import static org.junit.jupiter.api.Assertions.fail; /** * Tests the NodeRepository class used for talking to the node repository. It uses a mock from the node repository @@ -128,12 +132,21 @@ public class RealNodeRepositoryTest { @Test void testUpdateNodeAttributes() { - String hostname = "host4.yahoo.com"; + var hostname = "host4.yahoo.com"; + var dockerImage = "registry.example.com/repo/image-1:6.2.3"; + var wireguardKey = WireguardKey.from("111122223333444455556666777788889999000042c="); + nodeRepositoryApi.updateNodeAttributes( hostname, new NodeAttributes() .withRestartGeneration(1) - .withDockerImage(DockerImage.fromString("registry.example.com/repo/image-1:6.2.3"))); + .withDockerImage(DockerImage.fromString(dockerImage)) + .withWireguardPubkey(wireguardKey)); + + NodeSpec hostSpec = nodeRepositoryApi.getOptionalNode(hostname).orElseThrow(); + assertEquals(1, hostSpec.currentRestartGeneration().orElseThrow()); + assertEquals(dockerImage, hostSpec.currentDockerImage().orElseThrow().asString()); + assertEquals(wireguardKey.value(), hostSpec.wireguardPubkey().orElseThrow().value()); } @Test |