diff options
author | Jon Bratseth <bratseth@verizonmedia.com> | 2020-03-20 11:37:46 +0100 |
---|---|---|
committer | Jon Bratseth <bratseth@verizonmedia.com> | 2020-03-20 11:37:46 +0100 |
commit | f8100933e566ee1e1309930f62a937fcaa8dcb40 (patch) | |
tree | f07c423968ebaf40c843e1aeb5c066c6f188f587 /node-repository/src/test/java/com/yahoo/vespa/hosted/provision/NodeRepositoryTest.java | |
parent | cc659eb6a33016e412f89b797ea09b10fa4c5f3a (diff) |
Clear wantToDeproivision and wantToRetire when deprovisioning
Diffstat (limited to 'node-repository/src/test/java/com/yahoo/vespa/hosted/provision/NodeRepositoryTest.java')
-rw-r--r-- | node-repository/src/test/java/com/yahoo/vespa/hosted/provision/NodeRepositoryTest.java | 19 |
1 files changed, 15 insertions, 4 deletions
diff --git a/node-repository/src/test/java/com/yahoo/vespa/hosted/provision/NodeRepositoryTest.java b/node-repository/src/test/java/com/yahoo/vespa/hosted/provision/NodeRepositoryTest.java index 8830204547e..56b848a5cf2 100644 --- a/node-repository/src/test/java/com/yahoo/vespa/hosted/provision/NodeRepositoryTest.java +++ b/node-repository/src/test/java/com/yahoo/vespa/hosted/provision/NodeRepositoryTest.java @@ -157,12 +157,23 @@ public class NodeRepositoryTest { tester.addNode("id2", "host2", "default", NodeType.host); assertFalse(tester.nodeRepository().getNode("host1").get().history().hasEventAfter(History.Event.Type.deprovisioned, testStart)); + // Set host 1 properties and remove it + Node host1 = tester.nodeRepository().getNode("host1").get(); + host1 = host1.withWantToRetire(true, Agent.system, tester.nodeRepository().clock().instant()); + host1 = host1.with(host1.status().withWantToDeprovision(true)); + tester.nodeRepository().write(host1, tester.nodeRepository().lock(host1)); tester.nodeRepository().removeRecursively("host1"); - assertEquals(Node.State.deprovisioned, tester.nodeRepository().getNode("host1").get().state()); - assertTrue(tester.nodeRepository().getNode("host1").get().history().hasEventAfter(History.Event.Type.deprovisioned, testStart)); - Node existing = tester.addNode("id1", "host1", "default", NodeType.host); - assertTrue(existing.history().hasEventAfter(History.Event.Type.deprovisioned, testStart)); + // Host 1 is deprovisioned and unwanted properties are cleared + host1 = tester.nodeRepository().getNode("host1").get(); + assertEquals(Node.State.deprovisioned, host1.state()); + assertTrue(host1.history().hasEventAfter(History.Event.Type.deprovisioned, testStart)); + assertFalse(host1.status().wantToRetire()); + assertFalse(host1.status().wantToDeprovision()); + + // Adding it again moves it from deprovisioned + host1 = tester.addNode("id1", "host1", "default", NodeType.host); + assertTrue(host1.history().hasEventAfter(History.Event.Type.deprovisioned, testStart)); } @Test |