pm-29777 Add tests for new extension

Author: LRNcardozoWDF

app/src/main/kotlin/com/x8bit/bitwarden/data/vault/repository/util/WrappedAccountCryptographicStateExtensions.kt

@@ -19,6 +19,7 @@ fun createWrappedAccountCryptographicState(
     signingKey: String?,
     signedPublicKey: String?,
 ): WrappedAccountCryptographicState {
+    // TODO PM-29925: Check if signedPublicKey is not null
     return if (signingKey != null && securityState != null) {
         WrappedAccountCryptographicState.V2(
             privateKey = privateKey,

app/src/test/kotlin/com/x8bit/bitwarden/data/vault/repository/util/WrappedAccountCryptographicStateExtensionsTest.kt

@@ -0,0 +1,86 @@
+package com.x8bit.bitwarden.data.vault.repository.util
+
+import com.bitwarden.core.WrappedAccountCryptographicState
+import org.junit.jupiter.api.Assertions.assertEquals
+import org.junit.jupiter.api.Assertions.assertNull
+import org.junit.jupiter.api.Test
+
+private const val PRIVATE_KEY = "test-private-key"
+private const val SECURITY_STATE = "test-security-state"
+private const val SIGNING_KEY = "test-signing-key"
+private const val SIGNED_PUBLIC_KEY = "test-signed-public-key"
+
+class WrappedAccountCryptographicStateExtensionsTest {
+    // TODO PM-29925: Add tests for signedPublicKey
+
+    @Test
+    fun `createWrappedAccountCryptographicState returns V2 when both signingKey and securityState are non-null`() {
+        val result = createWrappedAccountCryptographicState(
+            privateKey = PRIVATE_KEY,
+            securityState = SECURITY_STATE,
+            signingKey = SIGNING_KEY,
+            signedPublicKey = SIGNED_PUBLIC_KEY,
+        )
+
+        val v2State = result as WrappedAccountCryptographicState.V2
+        assertEquals(PRIVATE_KEY, v2State.privateKey)
+        assertEquals(SECURITY_STATE, v2State.securityState)
+        assertEquals(SIGNING_KEY, v2State.signingKey)
+        assertEquals(SIGNED_PUBLIC_KEY, v2State.signedPublicKey)
+    }
+
+    @Test
+    fun `createWrappedAccountCryptographicState returns V1 when signingKey is null`() {
+        val result = createWrappedAccountCryptographicState(
+            privateKey = PRIVATE_KEY,
+            securityState = SECURITY_STATE,
+            signingKey = null,
+            signedPublicKey = SIGNED_PUBLIC_KEY,
+        )
+
+        val v1State = result as WrappedAccountCryptographicState.V1
+        assertEquals(PRIVATE_KEY, v1State.privateKey)
+    }
+
+    @Test
+    fun `createWrappedAccountCryptographicState returns V1 when securityState is null`() {
+        val result = createWrappedAccountCryptographicState(
+            privateKey = PRIVATE_KEY,
+            securityState = null,
+            signingKey = SIGNING_KEY,
+            signedPublicKey = SIGNED_PUBLIC_KEY,
+        )
+
+        val v1State = result as WrappedAccountCryptographicState.V1
+        assertEquals(PRIVATE_KEY, v1State.privateKey)
+    }
+
+    @Test
+    fun `createWrappedAccountCryptographicState returns V1 when both signingKey and securityState are null`() {
+        val result = createWrappedAccountCryptographicState(
+            privateKey = PRIVATE_KEY,
+            securityState = null,
+            signingKey = null,
+            signedPublicKey = SIGNED_PUBLIC_KEY,
+        )
+
+        val v1State = result as WrappedAccountCryptographicState.V1
+        assertEquals(PRIVATE_KEY, v1State.privateKey)
+    }
+
+    @Test
+    fun `createWrappedAccountCryptographicState returns V2 with null signedPublicKey when signingKey and securityState are present`() {
+        val result = createWrappedAccountCryptographicState(
+            privateKey = PRIVATE_KEY,
+            securityState = SECURITY_STATE,
+            signingKey = SIGNING_KEY,
+            signedPublicKey = null,
+        )
+
+        val v2State = result as WrappedAccountCryptographicState.V2
+        assertEquals(PRIVATE_KEY, v2State.privateKey)
+        assertEquals(SECURITY_STATE, v2State.securityState)
+        assertEquals(SIGNING_KEY, v2State.signingKey)
+        assertNull(v2State.signedPublicKey)
+    }
+}