Unit tests for interception of signout process result
This commit is contained in:
Maxime NATUREL
parent
8d870b8220
commit
0fe5ee68f2
|
|
@ -28,7 +28,6 @@ import javax.inject.Inject
|
|||
import kotlin.coroutines.Continuation
|
||||
import kotlin.coroutines.resume
|
||||
|
||||
// TODO add unit tests
|
||||
class InterceptSignoutFlowResponseUseCase @Inject constructor(
|
||||
private val reAuthHelper: ReAuthHelper,
|
||||
private val activeSessionHolder: ActiveSessionHolder,
|
||||
|
|
|
|||
|
|
@ -0,0 +1,174 @@
|
|||
/*
|
||||
* Copyright (c) 2022 New Vector Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package im.vector.app.features.settings.devices.v2.signout
|
||||
|
||||
import im.vector.app.test.fakes.FakeActiveSessionHolder
|
||||
import im.vector.app.test.fakes.FakeReAuthHelper
|
||||
import io.mockk.every
|
||||
import io.mockk.just
|
||||
import io.mockk.mockk
|
||||
import io.mockk.mockkStatic
|
||||
import io.mockk.runs
|
||||
import io.mockk.unmockkAll
|
||||
import org.amshove.kluent.shouldBeEqualTo
|
||||
import org.amshove.kluent.shouldBeInstanceOf
|
||||
import org.junit.After
|
||||
import org.junit.Before
|
||||
import org.junit.Test
|
||||
import org.matrix.android.sdk.api.auth.UIABaseAuth
|
||||
import org.matrix.android.sdk.api.auth.UserPasswordAuth
|
||||
import org.matrix.android.sdk.api.auth.data.LoginFlowTypes
|
||||
import org.matrix.android.sdk.api.auth.registration.RegistrationFlowResponse
|
||||
import org.matrix.android.sdk.api.auth.registration.nextUncompletedStage
|
||||
import org.matrix.android.sdk.api.session.uia.DefaultBaseAuth
|
||||
import kotlin.coroutines.Continuation
|
||||
import kotlin.coroutines.resume
|
||||
|
||||
private const val A_PASSWORD = "password"
|
||||
private const val A_SESSION_ID = "session-id"
|
||||
private const val AN_ERROR_CODE = "error-code"
|
||||
|
||||
class InterceptSignoutFlowResponseUseCaseTest {
|
||||
|
||||
private val fakeReAuthHelper = FakeReAuthHelper()
|
||||
private val fakeActiveSessionHolder = FakeActiveSessionHolder()
|
||||
|
||||
private val interceptSignoutFlowResponseUseCase = InterceptSignoutFlowResponseUseCase(
|
||||
reAuthHelper = fakeReAuthHelper.instance,
|
||||
activeSessionHolder = fakeActiveSessionHolder.instance,
|
||||
)
|
||||
|
||||
@Before
|
||||
fun setUp() {
|
||||
mockkStatic("org.matrix.android.sdk.api.auth.registration.RegistrationFlowResponseKt")
|
||||
}
|
||||
|
||||
@After
|
||||
fun tearDown() {
|
||||
unmockkAll()
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `given no error and a stored password and a next stage as password when intercepting then promise is resumed and success is returned`() {
|
||||
// Given
|
||||
val registrationFlowResponse = givenNextUncompletedStage(LoginFlowTypes.PASSWORD, A_SESSION_ID)
|
||||
fakeReAuthHelper.givenStoredPassword(A_PASSWORD)
|
||||
val errorCode: String? = null
|
||||
val promise = mockk<Continuation<UIABaseAuth>>()
|
||||
every { promise.resume(any()) } just runs
|
||||
val expectedAuth = UserPasswordAuth(
|
||||
session = null,
|
||||
user = fakeActiveSessionHolder.fakeSession.myUserId,
|
||||
password = A_PASSWORD,
|
||||
)
|
||||
|
||||
// When
|
||||
val result = interceptSignoutFlowResponseUseCase.execute(
|
||||
flowResponse = registrationFlowResponse,
|
||||
errCode = errorCode,
|
||||
promise = promise,
|
||||
)
|
||||
|
||||
// Then
|
||||
result shouldBeInstanceOf (SignoutSessionResult.Completed::class)
|
||||
every {
|
||||
promise.resume(expectedAuth)
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `given an error when intercepting then reAuthNeeded result is returned`() {
|
||||
// Given
|
||||
val registrationFlowResponse = givenNextUncompletedStage(LoginFlowTypes.PASSWORD, A_SESSION_ID)
|
||||
fakeReAuthHelper.givenStoredPassword(A_PASSWORD)
|
||||
val errorCode = AN_ERROR_CODE
|
||||
val promise = mockk<Continuation<UIABaseAuth>>()
|
||||
val expectedResult = SignoutSessionResult.ReAuthNeeded(
|
||||
pendingAuth = DefaultBaseAuth(session = A_SESSION_ID),
|
||||
uiaContinuation = promise,
|
||||
flowResponse = registrationFlowResponse,
|
||||
errCode = errorCode
|
||||
)
|
||||
|
||||
// When
|
||||
val result = interceptSignoutFlowResponseUseCase.execute(
|
||||
flowResponse = registrationFlowResponse,
|
||||
errCode = errorCode,
|
||||
promise = promise,
|
||||
)
|
||||
|
||||
// Then
|
||||
result shouldBeEqualTo expectedResult
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `given next stage is not password when intercepting then reAuthNeeded result is returned`() {
|
||||
// Given
|
||||
val registrationFlowResponse = givenNextUncompletedStage(LoginFlowTypes.SSO, A_SESSION_ID)
|
||||
fakeReAuthHelper.givenStoredPassword(A_PASSWORD)
|
||||
val errorCode: String? = null
|
||||
val promise = mockk<Continuation<UIABaseAuth>>()
|
||||
val expectedResult = SignoutSessionResult.ReAuthNeeded(
|
||||
pendingAuth = DefaultBaseAuth(session = A_SESSION_ID),
|
||||
uiaContinuation = promise,
|
||||
flowResponse = registrationFlowResponse,
|
||||
errCode = errorCode
|
||||
)
|
||||
|
||||
// When
|
||||
val result = interceptSignoutFlowResponseUseCase.execute(
|
||||
flowResponse = registrationFlowResponse,
|
||||
errCode = errorCode,
|
||||
promise = promise,
|
||||
)
|
||||
|
||||
// Then
|
||||
result shouldBeEqualTo expectedResult
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `given no existing stored password when intercepting then reAuthNeeded result is returned`() {
|
||||
// Given
|
||||
val registrationFlowResponse = givenNextUncompletedStage(LoginFlowTypes.PASSWORD, A_SESSION_ID)
|
||||
fakeReAuthHelper.givenStoredPassword(null)
|
||||
val errorCode: String? = null
|
||||
val promise = mockk<Continuation<UIABaseAuth>>()
|
||||
val expectedResult = SignoutSessionResult.ReAuthNeeded(
|
||||
pendingAuth = DefaultBaseAuth(session = A_SESSION_ID),
|
||||
uiaContinuation = promise,
|
||||
flowResponse = registrationFlowResponse,
|
||||
errCode = errorCode
|
||||
)
|
||||
|
||||
// When
|
||||
val result = interceptSignoutFlowResponseUseCase.execute(
|
||||
flowResponse = registrationFlowResponse,
|
||||
errCode = errorCode,
|
||||
promise = promise,
|
||||
)
|
||||
|
||||
// Then
|
||||
result shouldBeEqualTo expectedResult
|
||||
}
|
||||
|
||||
private fun givenNextUncompletedStage(nextStage: String, sessionId: String): RegistrationFlowResponse {
|
||||
val registrationFlowResponse = mockk<RegistrationFlowResponse>()
|
||||
every { registrationFlowResponse.nextUncompletedStage() } returns nextStage
|
||||
every { registrationFlowResponse.session } returns sessionId
|
||||
return registrationFlowResponse
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,30 @@
|
|||
/*
|
||||
* Copyright (c) 2022 New Vector Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package im.vector.app.test.fakes
|
||||
|
||||
import im.vector.app.features.login.ReAuthHelper
|
||||
import io.mockk.every
|
||||
import io.mockk.mockk
|
||||
|
||||
class FakeReAuthHelper {
|
||||
|
||||
val instance = mockk<ReAuthHelper>()
|
||||
|
||||
fun givenStoredPassword(pwd: String?) {
|
||||
every { instance.data } returns pwd
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue