refactor: mock dispatcher

checkout · Jun 5, 2024 · f16ffe4 · f16ffe4
1 parent e10bb76
commit f16ffe4
Show file tree

Hide file tree

Showing 2 changed files with 319 additions and 336 deletions.
diff --git a/checkout/src/main/java/com/checkout/tokenization/repository/TokenRepositoryImpl.kt b/checkout/src/main/java/com/checkout/tokenization/repository/TokenRepositoryImpl.kt
@@ -57,6 +57,9 @@ internal class TokenRepositoryImpl(
                 NonCancellable,
         )
 
+    @VisibleForTesting
+    var dispatcher = Dispatchers.IO
+
     @Suppress("TooGenericExceptionCaught")
     override fun sendCardTokenRequest(cardTokenRequest: CardTokenRequest) {
         var response: NetworkApiResponse<TokenDetailsResponse>
@@ -130,7 +133,7 @@ internal class TokenRepositoryImpl(
                     when (tokenResult) {
                         is TokenResult.Success -> {
                             try {
-                                withContext(Dispatchers.IO) {
+                                withContext(dispatcher) {
                                     riskSdkUseCase.execute(TokenResult.Success(tokenResult.result.token))
                                 }
                             } catch (exception: Exception) {
@@ -211,7 +214,7 @@ internal class TokenRepositoryImpl(
     ) {
         when (tokenResult) {
             is TokenResult.Success -> try {
-                withContext(Dispatchers.IO) {
+                withContext(dispatcher) {
                     riskSdkUseCase.execute(TokenResult.Success(tokenResult.result.token))
                 }
             } catch (exception: Exception) {