diff --git a/src/main/java/Vault.java b/src/main/java/Vault.java index 7ff3112..c0a7a43 100644 --- a/src/main/java/Vault.java +++ b/src/main/java/Vault.java @@ -14,7 +14,6 @@ public class Vault implements VaultInterface { public boolean credentialM; public String userName; public String password; - public boolean isInt = false; InputStream inputS = System.in; OutputStream outputS = System.out; private CredentialRepository credentialRepository = new CredentialRepository(); @@ -88,9 +87,11 @@ public class Vault implements VaultInterface { public void showCredential(String str){ if(str.matches("[0-9]+")) { - isInt = true; + println("Getting credential via ID"); credentialRepository.getCredentialsViaId(Integer.parseInt(str)); } else { + + println("Getting credential via name"); credentialRepository.getCredentialsViaName(str); } } diff --git a/src/test/java/VaultTest.java b/src/test/java/VaultTest.java index f789a42..793a9d3 100644 --- a/src/test/java/VaultTest.java +++ b/src/test/java/VaultTest.java @@ -86,26 +86,22 @@ public class VaultTest { assertEquals("peter", vlt.userName); assertTrue(outputStream.toString().contains("password")); assertEquals("password",vlt.password); - } + }*/ - @Test + /*@Test void showCredentialTest() { - ByteArrayOutputStream outputStream = new ByteArrayOutputStream(); - vlt.outputS = outputStream; - vlt.inputS = new ByteArrayInputStream("1".getBytes(StandardCharsets.UTF_8)); - vlt.showCredential(); + vlt.inputS = new ByteArrayInputStream("l\n1".getBytes(StandardCharsets.UTF_8)); + vlt.credentialMenu(); - assertTrue(outputStream.toString().contains("Type")); assertTrue(vlt.isInt); - vlt.outputS = outputStream; - vlt.inputS = new ByteArrayInputStream("peter".getBytes(StandardCharsets.UTF_8)); - vlt.showCredential(); + vlt.inputS = new ByteArrayInputStream("l\npeter".getBytes(StandardCharsets.UTF_8)); + vlt.credentialMenu(); - assertTrue(outputStream.toString().contains("Type")); assertFalse(vlt.isInt); }*/ + @Test void getAsJson() {vlt.getAsJson();}