[PM-30664] Unlock with PIN doesn’t appear as enabled after enabling ‘Require master password on app restart’ (#6344)

app/src/main/kotlin/com/x8bit/bitwarden/data/auth/datasource/disk/AuthDiskSourceImpl.kt

@@ -373,7 +373,10 @@ class AuthDiskSourceImpl(
         inMemoryOnly: Boolean,
     ) {
         inMemoryPinProtectedUserKeyEnvelopes[userId] = pinProtectedUserKeyEnvelope
-        if (inMemoryOnly) return
+        if (inMemoryOnly) {
+            getMutablePinProtectedUserKeyEnvelopeFlow(userId).tryEmit(pinProtectedUserKeyEnvelope)
+            return
+        }
         putString(
             key = PIN_PROTECTED_USER_KEY_KEY_ENVELOPE.appendIdentifier(userId),
             value = pinProtectedUserKeyEnvelope,

app/src/test/kotlin/com/x8bit/bitwarden/data/auth/datasource/disk/AuthDiskSourceTest.kt

@@ -865,6 +865,31 @@ class AuthDiskSourceTest {
             }
         }
 
+    @Test
+    @Suppress("MaxLineLength")
+    fun `storePinProtectedUserKeyEnvelope with inMemoryOnly true emits flow and stores only in memory`() =
+        runTest {
+            val userId = "mockUserId"
+            val envelope = "topSecretEnvelope"
+
+            authDiskSource.getPinProtectedUserKeyEnvelopeFlow(userId).test {
+                assertNull(awaitItem())
+                authDiskSource.storePinProtectedUserKeyEnvelope(
+                    userId = userId,
+                    pinProtectedUserKeyEnvelope = envelope,
+                    inMemoryOnly = true,
+                )
+                assertEquals(envelope, awaitItem())
+                assertEquals(envelope, authDiskSource.getPinProtectedUserKeyEnvelope(userId))
+                assertNull(
+                    fakeSharedPreferences.getString(
+                        "pinKeyEncryptedUserKeyEnvelope_$userId",
+                        null,
+                    ),
+                )
+            }
+        }
+
     @Test
     fun `getPinProtectedUserKeyEnvelope should pull from SharedPreferences`() {
         val pinProtectedUserKeyEnvelopeBaseKey =