1 files changed, 21 insertions, 0 deletions
diff --git a/app/src/androidTest/java/com/zeapo/pwdstore/PasswordEntryTest.java b/app/src/androidTest/java/com/zeapo/pwdstore/PasswordEntryTest.java
index e8ddc04c..3df296fe 100644
--- a/app/src/androidTest/java/com/zeapo/pwdstore/PasswordEntryTest.java
+++ b/app/src/androidTest/java/com/zeapo/pwdstore/PasswordEntryTest.java
@@ -59,4 +59,25 @@ public class PasswordEntryTest extends TestCase {
         assertTrue(entry.hasTotp());
         assertEquals("JBSWY3DPEHPK3PXP", entry.getTotpSecret());
     }
+
+    public void testNoHotpUriPresent() {
+        PasswordEntry entry = new PasswordEntry("secret\nextra\nlogin: username\ncontent");
+        assertFalse(entry.hasHotp());
+        assertNull(entry.getHotpSecret());
+        assertNull(entry.getHotpCounter());
+    }
+
+    public void testHotpUriInPassword() {
+        PasswordEntry entry = new PasswordEntry("otpauth://hotp/test?secret=JBSWY3DPEHPK3PXP&counter=25");
+        assertTrue(entry.hasHotp());
+        assertEquals("JBSWY3DPEHPK3PXP", entry.getHotpSecret());
+        assertEquals(new Long(25 ), entry.getHotpCounter());
+    }
+
+    public void testHotpUriInContent() {
+        PasswordEntry entry = new PasswordEntry("secret\nusername: test\notpauth://hotp/test?secret=JBSWY3DPEHPK3PXP&counter=25");
+        assertTrue(entry.hasHotp());
+        assertEquals("JBSWY3DPEHPK3PXP", entry.getHotpSecret());
+        assertEquals(new Long(25), entry.getHotpCounter());
+    }
 }
 \ No newline at end of file