@@ -18,7 +18,7 @@ final class ChatMessageActionsVC_Tests: XCTestCase {
1818 chatMessageController = . mock( )
1919 vc = ChatMessageActionsVC ( )
2020 vc. messageController = chatMessageController
21- vc. channel = . mock( cid: . unique, config: . mock( ) , ownCapabilities: [ . sendReply, . quoteMessage, . readEvents] )
21+ vc. channel = . mock( cid: . unique, config: . mock( ) , ownCapabilities: [ . sendReply, . quoteMessage, . readEvents, . flagMessage ] )
2222
2323 chatMessageController. simulateInitial (
2424 message: ChatMessage . mock ( id: . unique, cid: . unique, text: " test " , author: ChatUser . mock ( id: . unique) ) ,
@@ -62,7 +62,7 @@ final class ChatMessageActionsVC_Tests: XCTestCase {
6262
6363 let vc = TestView ( )
6464 vc. messageController = chatMessageController
65- vc. channel = . mock( cid: . unique, config: . mock( ) , ownCapabilities: [ . sendReply, . quoteMessage, . readEvents] )
65+ vc. channel = . mock( cid: . unique, config: . mock( ) , ownCapabilities: [ . sendReply, . quoteMessage, . readEvents, . flagMessage ] )
6666 AssertSnapshot ( vc. embedded ( ) )
6767 }
6868
@@ -336,11 +336,23 @@ final class ChatMessageActionsVC_Tests: XCTestCase {
336336 state: . remoteDataFetched
337337 )
338338
339- vc. channel = . mock( cid: . unique, ownCapabilities: [ ] )
339+ vc. channel = . mock( cid: . unique, ownCapabilities: [ . flagMessage ] )
340340
341341 XCTAssertTrue ( vc. messageActions. contains ( where: { $0 is FlagActionItem } ) )
342342 }
343343
344+ func test_messageActions_whenMessageIsSentByAnotherUser_whenNoCapability_doesNotContainFlagAction( ) {
345+ chatMessageController. simulateInitial (
346+ message: ChatMessage . mock ( isSentByCurrentUser: false ) ,
347+ replies: [ ] ,
348+ state: . remoteDataFetched
349+ )
350+
351+ vc. channel = . mock( cid: . unique, ownCapabilities: [ ] )
352+
353+ XCTAssertFalse ( vc. messageActions. contains ( where: { $0 is FlagActionItem } ) )
354+ }
355+
344356 func test_messageActions_whenSendingFailed_thenContainsResendActionEditActionDeleteAction( ) {
345357 chatMessageController. simulateInitial (
346358 message: . mock( localState: . sendingFailed, isSentByCurrentUser: false ) ,
@@ -419,7 +431,8 @@ final class ChatMessageActionsVC_Tests: XCTestCase {
419431 . sendReply,
420432 . readEvents,
421433 . updateAnyMessage,
422- . deleteAnyMessage
434+ . deleteAnyMessage,
435+ . flagMessage
423436 ]
424437 )
425438
0 commit comments