Add ConnectScreen instrumental tests

1 files changed, 581 insertions, 0 deletions

diff --git a/android/app/src/androidTest/kotlin/net/mullvad/mullvadvpn/compose/screen/ConnectScreenTest.kt b/android/app/src/androidTest/kotlin/net/mullvad/mullvadvpn/compose/screen/ConnectScreenTest.kt
new file mode 100644
index 0000000000..48671136e0
--- /dev/null
+++ b/android/app/src/androidTest/kotlin/net/mullvad/mullvadvpn/compose/screen/ConnectScreenTest.kt
@@ -0,0 +1,581 @@
+package net.mullvad.mullvadvpn.compose.screen
+
+import androidx.compose.ui.test.junit4.createComposeRule
+import androidx.compose.ui.test.onNodeWithTag
+import androidx.compose.ui.test.onNodeWithText
+import androidx.compose.ui.test.performClick
+import io.mockk.MockKAnnotations
+import io.mockk.every
+import io.mockk.mockk
+import io.mockk.unmockkAll
+import io.mockk.verify
+import net.mullvad.mullvadvpn.compose.state.ConnectUiState
+import net.mullvad.mullvadvpn.compose.test.CIRCULAR_PROGRESS_INDICATOR
+import net.mullvad.mullvadvpn.compose.test.CONNECT_BUTTON_TEST_TAG
+import net.mullvad.mullvadvpn.compose.test.LOCATION_INFO_TEST_TAG
+import net.mullvad.mullvadvpn.compose.test.RECONNECT_BUTTON_TEST_TAG
+import net.mullvad.mullvadvpn.compose.test.SELECT_LOCATION_BUTTON_TEST_TAG
+import net.mullvad.mullvadvpn.model.GeoIpLocation
+import net.mullvad.mullvadvpn.model.TunnelState
+import net.mullvad.mullvadvpn.relaylist.RelayItem
+import net.mullvad.talpid.net.TransportProtocol
+import net.mullvad.talpid.net.TunnelEndpoint
+import net.mullvad.talpid.tunnel.ActionAfterDisconnect
+import net.mullvad.talpid.tunnel.ErrorState
+import net.mullvad.talpid.tunnel.ErrorStateCause
+import org.junit.After
+import org.junit.Before
+import org.junit.Rule
+import org.junit.Test
+
+class ConnectScreenTest {
+    @get:Rule val composeTestRule = createComposeRule()
+
+    @Before
+    fun setup() {
+        MockKAnnotations.init(this)
+    }
+
+    @After
+    fun teardown() {
+        unmockkAll()
+    }
+
+    @Test
+    fun testDefaultState() {
+        // Arrange
+        composeTestRule.setContent { ConnectScreen(uiState = ConnectUiState.INITIAL) }
+
+        // Assert
+        composeTestRule.apply {
+            onNodeWithText("UNSECURED CONNECTION").assertExists()
+            onNodeWithText("Secure my connection").assertExists()
+        }
+    }
+
+    @Test
+    fun testConnectingState() {
+        // Arrange
+        composeTestRule.setContent {
+            ConnectScreen(
+                uiState =
+                    ConnectUiState(
+                        location = null,
+                        relayLocation = null,
+                        versionInfo = null,
+                        tunnelUiState = TunnelState.Connecting(null, null),
+                        tunnelRealState = TunnelState.Connecting(null, null),
+                        inAddress = null,
+                        outAddress = "",
+                        showLocation = false,
+                        isTunnelInfoExpanded = false
+                    )
+            )
+        }
+
+        // Assert
+        composeTestRule.apply {
+            onNodeWithTag(CIRCULAR_PROGRESS_INDICATOR).assertExists()
+            onNodeWithText("CREATING SECURE CONNECTION").assertExists()
+            onNodeWithText("Switch location").assertExists()
+            onNodeWithText("Cancel").assertExists()
+        }
+    }
+
+    @Test
+    fun testConnectingStateQuantumSecured() {
+        // Arrange
+        val mockTunnelEndpoint: TunnelEndpoint = mockk(relaxed = true)
+        every { mockTunnelEndpoint.quantumResistant } returns true
+        composeTestRule.setContent {
+            ConnectScreen(
+                uiState =
+                    ConnectUiState(
+                        location = null,
+                        relayLocation = null,
+                        versionInfo = null,
+                        tunnelUiState = TunnelState.Connecting(endpoint = mockTunnelEndpoint, null),
+                        tunnelRealState =
+                            TunnelState.Connecting(endpoint = mockTunnelEndpoint, null),
+                        inAddress = null,
+                        outAddress = "",
+                        showLocation = false,
+                        isTunnelInfoExpanded = false
+                    )
+            )
+        }
+
+        // Assert
+        composeTestRule.apply {
+            onNodeWithTag(CIRCULAR_PROGRESS_INDICATOR).assertExists()
+            onNodeWithText("CREATING QUANTUM SECURE CONNECTION").assertExists()
+            onNodeWithText("Switch location").assertExists()
+            onNodeWithText("Cancel").assertExists()
+        }
+    }
+
+    @Test
+    fun testConnectedState() {
+        // Arrange
+        val mockTunnelEndpoint: TunnelEndpoint = mockk(relaxed = true)
+        composeTestRule.setContent {
+            ConnectScreen(
+                uiState =
+                    ConnectUiState(
+                        location = null,
+                        relayLocation = null,
+                        versionInfo = null,
+                        tunnelUiState = TunnelState.Connected(mockTunnelEndpoint, null),
+                        tunnelRealState = TunnelState.Connected(mockTunnelEndpoint, null),
+                        inAddress = null,
+                        outAddress = "",
+                        showLocation = false,
+                        isTunnelInfoExpanded = false
+                    )
+            )
+        }
+
+        // Assert
+        composeTestRule.apply {
+            onNodeWithText("SECURE CONNECTION").assertExists()
+            onNodeWithText("Switch location").assertExists()
+            onNodeWithText("Disconnect").assertExists()
+        }
+    }
+
+    @Test
+    fun testConnectedStateQuantumSecured() {
+        // Arrange
+        val mockTunnelEndpoint: TunnelEndpoint = mockk(relaxed = true)
+        every { mockTunnelEndpoint.quantumResistant } returns true
+        composeTestRule.setContent {
+            ConnectScreen(
+                uiState =
+                    ConnectUiState(
+                        location = null,
+                        relayLocation = null,
+                        versionInfo = null,
+                        tunnelUiState = TunnelState.Connected(mockTunnelEndpoint, null),
+                        tunnelRealState = TunnelState.Connected(mockTunnelEndpoint, null),
+                        inAddress = null,
+                        outAddress = "",
+                        showLocation = false,
+                        isTunnelInfoExpanded = false
+                    )
+            )
+        }
+
+        // Assert
+        composeTestRule.apply {
+            onNodeWithText("QUANTUM SECURE CONNECTION").assertExists()
+            onNodeWithText("Switch location").assertExists()
+            onNodeWithText("Disconnect").assertExists()
+        }
+    }
+
+    @Test
+    fun testDisconnectingState() {
+        // Arrange
+        val mockRelayLocation: RelayItem = mockk(relaxed = true)
+        val mockLocationName = "Home"
+        every { mockRelayLocation.locationName } returns mockLocationName
+        composeTestRule.setContent {
+            ConnectScreen(
+                uiState =
+                    ConnectUiState(
+                        location = null,
+                        relayLocation = mockRelayLocation,
+                        versionInfo = null,
+                        tunnelUiState = TunnelState.Disconnecting(ActionAfterDisconnect.Nothing),
+                        tunnelRealState = TunnelState.Disconnecting(ActionAfterDisconnect.Nothing),
+                        inAddress = null,
+                        outAddress = "",
+                        showLocation = true,
+                        isTunnelInfoExpanded = false
+                    )
+            )
+        }
+
+        // Assert
+        composeTestRule.apply {
+            onNodeWithText("UNSECURED CONNECTION").assertExists()
+            onNodeWithText(mockLocationName).assertExists()
+            onNodeWithText("Secure my connection").assertExists()
+        }
+    }
+
+    @Test
+    fun testDisconnectedState() {
+        // Arrange
+        val mockRelayLocation: RelayItem = mockk(relaxed = true)
+        val mockLocationName = "Home"
+        every { mockRelayLocation.locationName } returns mockLocationName
+        composeTestRule.setContent {
+            ConnectScreen(
+                uiState =
+                    ConnectUiState(
+                        location = null,
+                        relayLocation = mockRelayLocation,
+                        versionInfo = null,
+                        tunnelUiState = TunnelState.Disconnected,
+                        tunnelRealState = TunnelState.Disconnected,
+                        inAddress = null,
+                        outAddress = "",
+                        showLocation = true,
+                        isTunnelInfoExpanded = false
+                    )
+            )
+        }
+
+        // Assert
+        composeTestRule.apply {
+            onNodeWithText("UNSECURED CONNECTION").assertExists()
+            onNodeWithText(mockLocationName).assertExists()
+            onNodeWithText("Secure my connection").assertExists()
+        }
+    }
+
+    @Test
+    fun testErrorStateBlocked() {
+        // Arrange
+        val mockRelayLocation: RelayItem = mockk(relaxed = true)
+        val mockLocationName = "Home"
+        every { mockRelayLocation.locationName } returns mockLocationName
+        composeTestRule.setContent {
+            ConnectScreen(
+                uiState =
+                    ConnectUiState(
+                        location = null,
+                        relayLocation = mockRelayLocation,
+                        versionInfo = null,
+                        tunnelUiState =
+                            TunnelState.Error(ErrorState(ErrorStateCause.StartTunnelError, true)),
+                        tunnelRealState =
+                            TunnelState.Error(ErrorState(ErrorStateCause.StartTunnelError, true)),
+                        inAddress = null,
+                        outAddress = "",
+                        showLocation = true,
+                        isTunnelInfoExpanded = false
+                    )
+            )
+        }
+
+        // Assert
+        composeTestRule.apply {
+            onNodeWithText("BLOCKED CONNECTION").assertExists()
+            onNodeWithText(mockLocationName).assertExists()
+            onNodeWithText("Disconnect").assertExists()
+        }
+    }
+
+    @Test
+    fun testErrorStateNotBlocked() {
+        // Arrange
+        val mockRelayLocation: RelayItem = mockk(relaxed = true)
+        val mockLocationName = "Home"
+        every { mockRelayLocation.locationName } returns mockLocationName
+        composeTestRule.setContent {
+            ConnectScreen(
+                uiState =
+                    ConnectUiState(
+                        location = null,
+                        relayLocation = mockRelayLocation,
+                        versionInfo = null,
+                        tunnelUiState =
+                            TunnelState.Error(ErrorState(ErrorStateCause.StartTunnelError, false)),
+                        tunnelRealState =
+                            TunnelState.Error(ErrorState(ErrorStateCause.StartTunnelError, false)),
+                        inAddress = null,
+                        outAddress = "",
+                        showLocation = true,
+                        isTunnelInfoExpanded = false
+                    )
+            )
+        }
+
+        // Assert
+        composeTestRule.apply {
+            onNodeWithText("FAILED TO SECURE CONNECTION").assertExists()
+            onNodeWithText(mockLocationName).assertExists()
+            onNodeWithText("Dismiss").assertExists()
+        }
+    }
+
+    @Test
+    fun testReconnectingState() {
+        // Arrange
+        composeTestRule.setContent {
+            ConnectScreen(
+                uiState =
+                    ConnectUiState(
+                        location = null,
+                        relayLocation = null,
+                        versionInfo = null,
+                        tunnelUiState = TunnelState.Disconnecting(ActionAfterDisconnect.Reconnect),
+                        tunnelRealState =
+                            TunnelState.Disconnecting(ActionAfterDisconnect.Reconnect),
+                        inAddress = null,
+                        outAddress = "",
+                        showLocation = false,
+                        isTunnelInfoExpanded = false
+                    )
+            )
+        }
+
+        // Assert
+        composeTestRule.apply {
+            onNodeWithTag(CIRCULAR_PROGRESS_INDICATOR).assertExists()
+            onNodeWithText("CREATING SECURE CONNECTION").assertExists()
+            onNodeWithText("Switch location").assertExists()
+            onNodeWithText("Disconnect").assertExists()
+        }
+    }
+
+    @Test
+    fun testDisconnectingBlockState() {
+        // Arrange
+        val mockRelayLocation: RelayItem = mockk(relaxed = true)
+        val mockLocationName = "Home"
+        every { mockRelayLocation.locationName } returns mockLocationName
+        composeTestRule.setContent {
+            ConnectScreen(
+                uiState =
+                    ConnectUiState(
+                        location = null,
+                        relayLocation = mockRelayLocation,
+                        versionInfo = null,
+                        tunnelUiState = TunnelState.Disconnecting(ActionAfterDisconnect.Block),
+                        tunnelRealState = TunnelState.Disconnecting(ActionAfterDisconnect.Block),
+                        inAddress = null,
+                        outAddress = "",
+                        showLocation = true,
+                        isTunnelInfoExpanded = false
+                    )
+            )
+        }
+
+        // Assert
+        composeTestRule.apply {
+            onNodeWithText("SECURE CONNECTION").assertExists()
+            onNodeWithText(mockLocationName).assertExists()
+            onNodeWithText("Disconnect").assertExists()
+        }
+    }
+
+    @Test
+    fun testClickSelectLocationButton() {
+        // Arrange
+        val mockRelayLocation: RelayItem = mockk(relaxed = true)
+        val mockLocationName = "Home"
+        every { mockRelayLocation.locationName } returns mockLocationName
+        val mockedClickHandler: () -> Unit = mockk(relaxed = true)
+        composeTestRule.setContent {
+            ConnectScreen(
+                uiState =
+                    ConnectUiState(
+                        location = null,
+                        relayLocation = mockRelayLocation,
+                        versionInfo = null,
+                        tunnelUiState = TunnelState.Disconnected,
+                        tunnelRealState = TunnelState.Disconnected,
+                        inAddress = null,
+                        outAddress = "",
+                        showLocation = false,
+                        isTunnelInfoExpanded = false
+                    ),
+                onSwitchLocationClick = mockedClickHandler
+            )
+        }
+
+        // Act
+        composeTestRule.onNodeWithTag(SELECT_LOCATION_BUTTON_TEST_TAG).performClick()
+
+        // Assert
+        verify { mockedClickHandler.invoke() }
+    }
+
+    @Test
+    fun testOnDisconnectClick() {
+        // Arrange
+        val mockTunnelEndpoint: TunnelEndpoint = mockk(relaxed = true)
+        val mockedClickHandler: () -> Unit = mockk(relaxed = true)
+        composeTestRule.setContent {
+            ConnectScreen(
+                uiState =
+                    ConnectUiState(
+                        location = null,
+                        relayLocation = null,
+                        versionInfo = null,
+                        tunnelUiState = TunnelState.Connected(mockTunnelEndpoint, null),
+                        tunnelRealState = TunnelState.Connected(mockTunnelEndpoint, null),
+                        inAddress = null,
+                        outAddress = "",
+                        showLocation = false,
+                        isTunnelInfoExpanded = false
+                    ),
+                onDisconnectClick = mockedClickHandler
+            )
+        }
+
+        // Act
+        composeTestRule.onNodeWithTag(CONNECT_BUTTON_TEST_TAG).performClick()
+
+        // Assert
+        verify { mockedClickHandler.invoke() }
+    }
+
+    @Test
+    fun testOnReconnectClick() {
+        // Arrange
+        val mockTunnelEndpoint: TunnelEndpoint = mockk(relaxed = true)
+        val mockedClickHandler: () -> Unit = mockk(relaxed = true)
+        composeTestRule.setContent {
+            ConnectScreen(
+                uiState =
+                    ConnectUiState(
+                        location = null,
+                        relayLocation = null,
+                        versionInfo = null,
+                        tunnelUiState = TunnelState.Connected(mockTunnelEndpoint, null),
+                        tunnelRealState = TunnelState.Connected(mockTunnelEndpoint, null),
+                        inAddress = null,
+                        outAddress = "",
+                        showLocation = false,
+                        isTunnelInfoExpanded = false
+                    ),
+                onReconnectClick = mockedClickHandler
+            )
+        }
+
+        // Act
+        composeTestRule.onNodeWithTag(RECONNECT_BUTTON_TEST_TAG).performClick()
+
+        // Assert
+        verify { mockedClickHandler.invoke() }
+    }
+
+    @Test
+    fun testOnConnectClick() {
+        // Arrange
+        val mockedClickHandler: () -> Unit = mockk(relaxed = true)
+        composeTestRule.setContent {
+            ConnectScreen(
+                uiState =
+                    ConnectUiState(
+                        location = null,
+                        relayLocation = null,
+                        versionInfo = null,
+                        tunnelUiState = TunnelState.Disconnected,
+                        tunnelRealState = TunnelState.Disconnected,
+                        inAddress = null,
+                        outAddress = "",
+                        showLocation = false,
+                        isTunnelInfoExpanded = false
+                    ),
+                onConnectClick = mockedClickHandler
+            )
+        }
+
+        // Act
+        composeTestRule.onNodeWithTag(CONNECT_BUTTON_TEST_TAG).performClick()
+
+        // Assert
+        verify { mockedClickHandler.invoke() }
+    }
+
+    @Test
+    fun testOnCancelClick() {
+        // Arrange
+        val mockedClickHandler: () -> Unit = mockk(relaxed = true)
+        composeTestRule.setContent {
+            ConnectScreen(
+                uiState =
+                    ConnectUiState(
+                        location = null,
+                        relayLocation = null,
+                        versionInfo = null,
+                        tunnelUiState = TunnelState.Connecting(null, null),
+                        tunnelRealState = TunnelState.Connecting(null, null),
+                        inAddress = null,
+                        outAddress = "",
+                        showLocation = false,
+                        isTunnelInfoExpanded = false
+                    ),
+                onCancelClick = mockedClickHandler
+            )
+        }
+
+        // Act
+        composeTestRule.onNodeWithTag(CONNECT_BUTTON_TEST_TAG).performClick()
+
+        // Assert
+        verify { mockedClickHandler.invoke() }
+    }
+
+    @Test
+    fun testToggleTunnelInfo() {
+        // Arrange
+        val mockedClickHandler: () -> Unit = mockk(relaxed = true)
+        composeTestRule.setContent {
+            ConnectScreen(
+                uiState =
+                    ConnectUiState(
+                        location = null,
+                        relayLocation = null,
+                        versionInfo = null,
+                        tunnelUiState = TunnelState.Connecting(null, null),
+                        tunnelRealState = TunnelState.Connecting(null, null),
+                        inAddress = null,
+                        outAddress = "",
+                        showLocation = false,
+                        isTunnelInfoExpanded = false
+                    ),
+                onToggleTunnelInfo = mockedClickHandler
+            )
+        }
+
+        // Act
+        composeTestRule.onNodeWithTag(LOCATION_INFO_TEST_TAG).performClick()
+
+        // Assert
+        verify { mockedClickHandler.invoke() }
+    }
+
+    @Test
+    fun showLocationInfo() {
+        // Arrange
+        val mockLocation: GeoIpLocation = mockk(relaxed = true)
+        val mockTunnelEndpoint: TunnelEndpoint = mockk(relaxed = true)
+        val mockHostName = "Host-Name"
+        val mockPort = 99
+        val mockHost = "Host"
+        val mockProtocol = TransportProtocol.Udp
+        val mockInAddress = Triple(mockHost, mockPort, mockProtocol)
+        val mockOutAddress = "HostAddressV4 / HostAddressV4"
+        every { mockLocation.hostname } returns mockHostName
+        composeTestRule.setContent {
+            ConnectScreen(
+                uiState =
+                    ConnectUiState(
+                        location = mockLocation,
+                        relayLocation = null,
+                        versionInfo = null,
+                        tunnelUiState = TunnelState.Connected(mockTunnelEndpoint, null),
+                        tunnelRealState = TunnelState.Connected(mockTunnelEndpoint, null),
+                        inAddress = mockInAddress,
+                        outAddress = mockOutAddress,
+                        showLocation = false,
+                        isTunnelInfoExpanded = true
+                    )
+            )
+        }
+
+        // Assert
+        composeTestRule.apply {
+            onNodeWithText(mockHostName).assertExists()
+            onNodeWithText("WireGuard").assertExists()
+            onNodeWithText("In $mockHost:$mockPort UDP").assertExists()
+            onNodeWithText("Out $mockOutAddress").assertExists()
+        }
+    }
+}