fix: test cov

wireapp · Sep 21, 2023 · 34e3863 · 34e3863
1 parent 6da1c31
commit 34e3863
Showing 1 changed file with 27 additions and 1 deletion.
diff --git a/logic/src/commonTest/kotlin/com/wire/kalium/logic/feature/user/SearchKnownUserUseCaseTest.kt b/logic/src/commonTest/kotlin/com/wire/kalium/logic/feature/user/SearchKnownUserUseCaseTest.kt
@@ -251,6 +251,32 @@ class SearchKnownUserUseCaseTest {
 
     }
 
+    @Test
+    fun givenAnInputStartingWithAtSymbolAndDomainPresent_whenSearchingUsers_thenSearchBySanitizedHandle() = runTest {
+        // given
+        val handlePartOfQuery = "somehandle"
+        val handleSearchQuery = "@$handlePartOfQuery@bella.wire.link"
+        val (arrangement, searchKnownUsersUseCase) = Arrangement()
+            .withSuccessFullSelfUserRetrieve()
+            .withSearchByHandle(handleSearchQuery)
+            .arrange()
+
+        // when
+        searchKnownUsersUseCase(handleSearchQuery)
+
+        // then
+        verify(arrangement.searchUserRepository)
+            .suspendFunction(arrangement.searchUserRepository::searchKnownUsersByHandle)
+            .with(eq(handlePartOfQuery), anything())
+            .wasInvoked(exactly = once)
+
+        verify(arrangement.searchUserRepository)
+            .suspendFunction(arrangement.searchUserRepository::searchKnownUsersByNameOrHandleOrEmail)
+            .with(anything(), anything())
+            .wasNotInvoked()
+    }
+
+
     private class Arrangement {
 
         @Mock
@@ -292,7 +318,7 @@ class SearchKnownUserUseCaseTest {
             given(searchUserRepository)
                 .suspendFunction(searchUserRepository::searchKnownUsersByHandle)
                 .whenInvokedWith(
-                    if (searchQuery == null) any() else eq(searchQuery.removePrefix("@")),
+                    if (searchQuery == null) any() else eq(searchQuery.substringAfter("@").substringBeforeLast("@")),
                     if (searchUsersOptions == null) any() else eq(searchUsersOptions)
                 )
                 .thenReturn(