diff --git a/src/main/java/Vault.java b/src/main/java/Vault.java index 7f29b23..7086edb 100644 --- a/src/main/java/Vault.java +++ b/src/main/java/Vault.java @@ -34,7 +34,6 @@ public class Vault implements VaultInterface { if (input.equals("e")) { config = false; - } } diff --git a/src/test/java/VaultTest.java b/src/test/java/VaultTest.java index f506dcc..b39508b 100644 --- a/src/test/java/VaultTest.java +++ b/src/test/java/VaultTest.java @@ -55,6 +55,20 @@ public class VaultTest { assertFalse(vlt.config); } + @Test + void doNotExitConfigAfterWrongInput() { + vlt.inputS = new ByteArrayInputStream("__\n".getBytes(StandardCharsets.UTF_8)); + vlt.configure(); + assertTrue(vlt.config); + + ByteArrayOutputStream outputStream = new ByteArrayOutputStream(); + vlt.inputS = getEmptyStringInputStream(); + vlt.outputS = outputStream; + + assertThrowsExactly(NoSuchElementException.class, () -> vlt.configure()); + assertTrue(outputStream.toString().startsWith("Configure:")); + } + private ByteArrayInputStream getEmptyStringInputStream() {