1 files changed, 63 insertions, 0 deletions
diff --git a/app/src/test/java/com/zeapo/pwdstore/PasswordEntryTest.kt b/app/src/test/java/com/zeapo/pwdstore/PasswordEntryTest.kt
new file mode 100644
index 00000000..2074f40b
--- /dev/null
+++ b/app/src/test/java/com/zeapo/pwdstore/PasswordEntryTest.kt
@@ -0,0 +1,63 @@
+/*
+ * Copyright © 2014-2020 The Android Password Store Authors. All Rights Reserved.
+ * SPDX-License-Identifier: GPL-3.0-only
+ */
+package com.zeapo.pwdstore
+
+import org.junit.Test
+import kotlin.test.assertEquals
+import kotlin.test.assertFalse
+import kotlin.test.assertNull
+import kotlin.test.assertTrue
+
+class PasswordEntryTest {
+    @Test fun testGetPassword() {
+        assertEquals("fooooo", PasswordEntry("fooooo\nbla\n").password)
+        assertEquals("fooooo", PasswordEntry("fooooo\nbla").password)
+        assertEquals("fooooo", PasswordEntry("fooooo\n").password)
+        assertEquals("fooooo", PasswordEntry("fooooo").password)
+        assertEquals("", PasswordEntry("\nblubb\n").password)
+        assertEquals("", PasswordEntry("\nblubb").password)
+        assertEquals("", PasswordEntry("\n").password)
+        assertEquals("", PasswordEntry("").password)
+    }
+
+    @Test fun testGetExtraContent() {
+        assertEquals("bla\n", PasswordEntry("fooooo\nbla\n").extraContent)
+        assertEquals("bla", PasswordEntry("fooooo\nbla").extraContent)
+        assertEquals("", PasswordEntry("fooooo\n").extraContent)
+        assertEquals("", PasswordEntry("fooooo").extraContent)
+        assertEquals("blubb\n", PasswordEntry("\nblubb\n").extraContent)
+        assertEquals("blubb", PasswordEntry("\nblubb").extraContent)
+        assertEquals("", PasswordEntry("\n").extraContent)
+        assertEquals("", PasswordEntry("").extraContent)
+    }
+
+    @Test fun testGetUsername() {
+        for (field in PasswordEntry.USERNAME_FIELDS) {
+            assertEquals("username", PasswordEntry("\n$field username").username)
+            assertEquals("username", PasswordEntry("\n${field.toUpperCase()} username").username)
+        }
+        assertEquals(
+            "username",
+            PasswordEntry("secret\nextra\nlogin: username\ncontent\n").username)
+        assertEquals(
+            "username",
+            PasswordEntry("\nextra\nusername: username\ncontent\n").username)
+        assertEquals(
+            "username", PasswordEntry("\nUSERNaMe:  username\ncontent\n").username)
+        assertEquals("username", PasswordEntry("\nlogin:    username").username)
+        assertEquals("foo@example.com", PasswordEntry("\nemail: foo@example.com").username)
+        assertEquals("username", PasswordEntry("\nidentity: username\nlogin: another_username").username)
+        assertEquals("username", PasswordEntry("\nLOGiN:username").username)
+        assertNull(PasswordEntry("secret\nextra\ncontent\n").username)
+    }
+
+    @Test fun testHasUsername() {
+        assertTrue(PasswordEntry("secret\nextra\nlogin: username\ncontent\n").hasUsername())
+        assertFalse(PasswordEntry("secret\nextra\ncontent\n").hasUsername())
+        assertFalse(PasswordEntry("secret\nlogin failed\n").hasUsername())
+        assertFalse(PasswordEntry("\n").hasUsername())
+        assertFalse(PasswordEntry("").hasUsername())
+    }
+}