From 60ab5dd698ea8bbec637bfd6ec6cf10603769368 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andre=CC=81=20Bispo?= <abispo@bitwarden.com>
Date: Fri, 20 Dec 2024 17:39:59 +0000
Subject: [PATCH] [PM-8217] PR fixes

---
 .../newdevicenotice/NewDeviceNoticeTwoFactorViewModel.kt    | 1 -
 .../newdevicenotice/NewDeviceNoticeTwoFactorScreenTest.kt   | 6 ++++++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/app/src/main/java/com/x8bit/bitwarden/ui/auth/feature/newdevicenotice/NewDeviceNoticeTwoFactorViewModel.kt b/app/src/main/java/com/x8bit/bitwarden/ui/auth/feature/newdevicenotice/NewDeviceNoticeTwoFactorViewModel.kt
index 451525c8a57..b342f8c8faf 100644
--- a/app/src/main/java/com/x8bit/bitwarden/ui/auth/feature/newdevicenotice/NewDeviceNoticeTwoFactorViewModel.kt
+++ b/app/src/main/java/com/x8bit/bitwarden/ui/auth/feature/newdevicenotice/NewDeviceNoticeTwoFactorViewModel.kt
@@ -12,7 +12,6 @@ import javax.inject.Inject
 /**
  * Manages application state for the new device notice two factor screen.
  */
-@Suppress("TooManyFunctions")
 @HiltViewModel
 class NewDeviceNoticeTwoFactorViewModel @Inject constructor(
     val environmentRepository: EnvironmentRepository,
diff --git a/app/src/test/java/com/x8bit/bitwarden/ui/auth/feature/newdevicenotice/NewDeviceNoticeTwoFactorScreenTest.kt b/app/src/test/java/com/x8bit/bitwarden/ui/auth/feature/newdevicenotice/NewDeviceNoticeTwoFactorScreenTest.kt
index c062ba74113..37f7055b141 100644
--- a/app/src/test/java/com/x8bit/bitwarden/ui/auth/feature/newdevicenotice/NewDeviceNoticeTwoFactorScreenTest.kt
+++ b/app/src/test/java/com/x8bit/bitwarden/ui/auth/feature/newdevicenotice/NewDeviceNoticeTwoFactorScreenTest.kt
@@ -13,6 +13,7 @@ import io.mockk.mockk
 import io.mockk.runs
 import io.mockk.verify
 import junit.framework.TestCase.assertTrue
+import org.junit.After
 import org.junit.Before
 import org.junit.Test
 
@@ -37,6 +38,11 @@ class NewDeviceNoticeTwoFactorScreenTest : BaseComposeTest() {
         }
     }
 
+    @After
+    fun tearDown() {
+        onNavigateBackCalled = false
+    }
+
     @Test
     fun `Turn on two-step verification click should send TurnOnTwoFactorClick action`() {
         composeTestRule