Fix race condition in new unit tests added to AWSCognitoAuthPluginTest

aws-amplify · Nov 25, 2024 · 7282fe2 · 7282fe2
1 parent 98415ba
commit 7282fe2
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/aws-auth-cognito/src/test/java/com/amplifyframework/auth/cognito/AWSCognitoAuthPluginTest.kt b/aws-auth-cognito/src/test/java/com/amplifyframework/auth/cognito/AWSCognitoAuthPluginTest.kt
@@ -776,7 +776,7 @@ class AWSCognitoAuthPluginTest {
     fun listWebAuthnCredentials() {
         val useCase = authPlugin.useCaseFactory.listWebAuthnCredentials()
         authPlugin.listWebAuthnCredentials({}, {})
-        coVerify {
+        coVerify(timeout = CHANNEL_TIMEOUT) {
             useCase.execute(AuthListWebAuthnCredentialsOptions.defaults())
         }
     }
@@ -786,7 +786,7 @@ class AWSCognitoAuthPluginTest {
         val useCase = authPlugin.useCaseFactory.listWebAuthnCredentials()
         val options = AWSCognitoAuthListWebAuthnCredentialsOptions.builder().build()
         authPlugin.listWebAuthnCredentials(options, {}, {})
-        coVerify {
+        coVerify(timeout = CHANNEL_TIMEOUT) {
             useCase.execute(options)
         }
     }
@@ -796,7 +796,7 @@ class AWSCognitoAuthPluginTest {
         val useCase = authPlugin.useCaseFactory.deleteWebAuthnCredential()
         val credentialId = "someId"
         authPlugin.deleteWebAuthnCredential(credentialId, {}, {})
-        coVerify {
+        coVerify(timeout = CHANNEL_TIMEOUT) {
             useCase.execute(credentialId, AuthDeleteWebAuthnCredentialOptions.defaults())
         }
     }
@@ -807,7 +807,7 @@ class AWSCognitoAuthPluginTest {
         val options: AuthDeleteWebAuthnCredentialOptions = mockk()
         val credentialId = "someId"
         authPlugin.deleteWebAuthnCredential(credentialId, options, {}, {})
-        coVerify {
+        coVerify(timeout = CHANNEL_TIMEOUT) {
             useCase.execute(credentialId, options)
         }
     }