Benoit Marty
4 months ago
1 changed files with 89 additions and 0 deletions
@ -0,0 +1,89 @@ |
|||||||
|
/* |
||||||
|
* Copyright (c) 2024 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 io.element.android.libraries.pushproviders.unifiedpush |
||||||
|
|
||||||
|
import androidx.test.platform.app.InstrumentationRegistry |
||||||
|
import com.google.common.truth.Truth.assertThat |
||||||
|
import io.element.android.libraries.matrix.test.AN_EXCEPTION |
||||||
|
import io.element.android.libraries.matrix.test.A_SECRET |
||||||
|
import io.element.android.libraries.pushproviders.api.Distributor |
||||||
|
import io.element.android.libraries.pushproviders.unifiedpush.registration.EndpointRegistrationHandler |
||||||
|
import io.element.android.libraries.pushproviders.unifiedpush.registration.RegistrationResult |
||||||
|
import kotlinx.coroutines.delay |
||||||
|
import kotlinx.coroutines.launch |
||||||
|
import kotlinx.coroutines.test.TestScope |
||||||
|
import kotlinx.coroutines.test.runTest |
||||||
|
import org.junit.Ignore |
||||||
|
import org.junit.Test |
||||||
|
import org.junit.runner.RunWith |
||||||
|
import org.robolectric.RobolectricTestRunner |
||||||
|
|
||||||
|
@RunWith(RobolectricTestRunner::class) |
||||||
|
class DefaultRegisterUnifiedPushUseCaseTest { |
||||||
|
@Test |
||||||
|
fun `test registration successful`() = runTest { |
||||||
|
val endpointRegistrationHandler = EndpointRegistrationHandler() |
||||||
|
val useCase = createDefaultRegisterUnifiedPushUseCase( |
||||||
|
endpointRegistrationHandler = endpointRegistrationHandler |
||||||
|
) |
||||||
|
val aDistributor = Distributor("aValue", "aName") |
||||||
|
launch { |
||||||
|
delay(100) |
||||||
|
endpointRegistrationHandler.registrationDone(RegistrationResult(A_SECRET, Result.success(Unit))) |
||||||
|
} |
||||||
|
val result = useCase.execute(aDistributor, A_SECRET) |
||||||
|
assertThat(result.isSuccess).isTrue() |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
fun `test registration error`() = runTest { |
||||||
|
val endpointRegistrationHandler = EndpointRegistrationHandler() |
||||||
|
val useCase = createDefaultRegisterUnifiedPushUseCase( |
||||||
|
endpointRegistrationHandler = endpointRegistrationHandler |
||||||
|
) |
||||||
|
val aDistributor = Distributor("aValue", "aName") |
||||||
|
launch { |
||||||
|
delay(100) |
||||||
|
endpointRegistrationHandler.registrationDone(RegistrationResult(A_SECRET, Result.failure(AN_EXCEPTION))) |
||||||
|
} |
||||||
|
val result = useCase.execute(aDistributor, A_SECRET) |
||||||
|
assertThat(result.isSuccess).isFalse() |
||||||
|
} |
||||||
|
|
||||||
|
@Ignore("Find a solution to test timeout") |
||||||
|
@Test |
||||||
|
fun `test registration timeout`() = runTest { |
||||||
|
val endpointRegistrationHandler = EndpointRegistrationHandler() |
||||||
|
val useCase = createDefaultRegisterUnifiedPushUseCase( |
||||||
|
endpointRegistrationHandler = endpointRegistrationHandler |
||||||
|
) |
||||||
|
val aDistributor = Distributor("aValue", "aName") |
||||||
|
val result = useCase.execute(aDistributor, A_SECRET) |
||||||
|
assertThat(result.isSuccess).isFalse() |
||||||
|
} |
||||||
|
|
||||||
|
private fun TestScope.createDefaultRegisterUnifiedPushUseCase( |
||||||
|
endpointRegistrationHandler: EndpointRegistrationHandler |
||||||
|
): DefaultRegisterUnifiedPushUseCase { |
||||||
|
val context = InstrumentationRegistry.getInstrumentation().context |
||||||
|
return DefaultRegisterUnifiedPushUseCase( |
||||||
|
context = context, |
||||||
|
endpointRegistrationHandler = endpointRegistrationHandler, |
||||||
|
coroutineScope = this@createDefaultRegisterUnifiedPushUseCase |
||||||
|
) |
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue