diff --git a/src/main/java/Vault.java b/src/main/java/Vault.java index a15638d..ab867b5 100644 --- a/src/main/java/Vault.java +++ b/src/main/java/Vault.java @@ -118,14 +118,19 @@ public class Vault implements VaultInterface { println("Type in ID or username"); String str = scan.nextLine(); + Credential credential = null; if(str.matches("[0-9]+")) { println("Getting credential via ID"); - credentialRepository.getCredentialsViaId(Integer.parseInt(str)); + credential = credentialRepository.getCredentialsViaId(Integer.parseInt(str)); } else { println("Getting credential via name"); - credentialRepository.getCredentialsViaName(str); + credential = credentialRepository.getCredentialsViaName(str); + } + + if (credential != null) { + println(credential.getName() + ": " + credential.getPassword()); } } diff --git a/src/test/java/VaultTest.java b/src/test/java/VaultTest.java index 5fa68e1..2a96584 100644 --- a/src/test/java/VaultTest.java +++ b/src/test/java/VaultTest.java @@ -84,6 +84,8 @@ public class VaultTest { @Test void showCredentialTest() { + addCredentialTest(); + ByteArrayOutputStream outputStream = new ByteArrayOutputStream(); vlt.outputS = outputStream; vlt.inputS = new ByteArrayInputStream("1".getBytes(StandardCharsets.UTF_8)); @@ -94,7 +96,8 @@ public class VaultTest { vlt.inputS = new ByteArrayInputStream("peter".getBytes(StandardCharsets.UTF_8)); vlt.credentialMenu(); - assertTrue(outputStream.toString().contains("name")); + assertTrue(outputStream.toString().contains("name")); + assertTrue(outputStream.toString().contains("peter: " + vlt.password)); } @Test