mirror of
https://github.com/simplex-chat/simplex-chat.git
synced 2026-04-01 03:05:55 +00:00
c782e6762b
* core: remove contact merge, obsolete group code (#6292) * core: remove file connections (#6298) * core: rename migration * corre: remove unused contacts merged event (#6308) * core: optimize subscriptions (#6304) * core: remove conn ids map from subscriptions (#6309) * rename migration (#6316) --------- Co-authored-by: Evgeny Poberezkin <evgeny@poberezkin.com>
94 lines
3.3 KiB
Haskell
94 lines
3.3 KiB
Haskell
{-# LANGUAGE CPP #-}
|
|
{-# LANGUAGE DuplicateRecordFields #-}
|
|
{-# LANGUAGE NamedFieldPuns #-}
|
|
{-# LANGUAGE OverloadedStrings #-}
|
|
|
|
module Bots.BroadcastTests where
|
|
|
|
import Broadcast.Bot
|
|
import Broadcast.Options
|
|
import ChatClient
|
|
import ChatTests.DBUtils
|
|
import ChatTests.Utils
|
|
import Control.Concurrent (forkIO, killThread, threadDelay)
|
|
import Control.Exception (bracket)
|
|
import Simplex.Chat.Bot.KnownContacts
|
|
import Simplex.Chat.Core
|
|
import Simplex.Chat.Options (CoreChatOpts (..))
|
|
import Simplex.Chat.Options.DB
|
|
import Simplex.Chat.Types (ChatPeerType (..), Profile (..))
|
|
import Test.Hspec hiding (it)
|
|
#if !defined(dbPostgres)
|
|
import System.FilePath ((</>))
|
|
#endif
|
|
|
|
broadcastBotTests :: SpecWith TestParams
|
|
broadcastBotTests = do
|
|
it "should broadcast message" testBroadcastMessages
|
|
|
|
withBroadcastBot :: BroadcastBotOpts -> IO () -> IO ()
|
|
withBroadcastBot opts test =
|
|
bracket (forkIO bot) killThread (\_ -> threadDelay 500000 >> test)
|
|
where
|
|
bot = simplexChatCore testCfg (mkChatOpts opts) $ broadcastBot opts
|
|
|
|
broadcastBotProfile :: Profile
|
|
broadcastBotProfile = Profile {displayName = "broadcast_bot", fullName = "Broadcast Bot", shortDescr = Nothing, image = Nothing, contactLink = Nothing, peerType = Just CPTBot, preferences = Nothing}
|
|
|
|
mkBotOpts :: TestParams -> [KnownContact] -> BroadcastBotOpts
|
|
mkBotOpts ps publishers =
|
|
BroadcastBotOpts
|
|
{ coreOptions =
|
|
testCoreOpts
|
|
{ dbOptions =
|
|
(dbOptions testCoreOpts)
|
|
#if defined(dbPostgres)
|
|
{dbSchemaPrefix = "client_" <> botDbPrefix}
|
|
#else
|
|
{dbFilePrefix = tmpPath ps </> botDbPrefix}
|
|
#endif
|
|
|
|
},
|
|
botDisplayName = "broadcast_bot",
|
|
publishers,
|
|
welcomeMessage = defaultWelcomeMessage publishers,
|
|
prohibitedMessage = defaultWelcomeMessage publishers
|
|
}
|
|
|
|
botDbPrefix :: FilePath
|
|
botDbPrefix = "broadcast_bot"
|
|
|
|
testBroadcastMessages :: HasCallStack => TestParams -> IO ()
|
|
testBroadcastMessages ps = do
|
|
botLink <-
|
|
withNewTestChat ps botDbPrefix broadcastBotProfile $ \bc_bot ->
|
|
withNewTestChat ps "alice" aliceProfile $ \alice -> do
|
|
connectUsers bc_bot alice
|
|
bc_bot ##> "/ad"
|
|
getContactLink bc_bot True
|
|
let botOpts = mkBotOpts ps [KnownContact 2 "alice"]
|
|
withBroadcastBot botOpts $
|
|
withTestChat ps "alice" $ \alice ->
|
|
withNewTestChat ps "bob" bobProfile $ \bob ->
|
|
withNewTestChat ps "cath" cathProfile $ \cath -> do
|
|
alice <## "1 connections subscribed"
|
|
bob `connectVia` botLink
|
|
bob #> "@broadcast_bot hello"
|
|
bob <# "broadcast_bot> > hello"
|
|
bob <## " Hello! I am a broadcast bot."
|
|
bob <## "I broadcast messages to all connected users from @alice."
|
|
cath `connectVia` botLink
|
|
alice #> "@broadcast_bot hello all!"
|
|
alice <# "broadcast_bot> hello all!" -- we broadcast to the sender too, /feed is used by bot
|
|
bob <# "broadcast_bot> hello all!"
|
|
cath <# "broadcast_bot> hello all!"
|
|
alice <# "broadcast_bot> > hello all!"
|
|
alice <## " Forwarded to 3 contact(s), 0 errors"
|
|
where
|
|
cc `connectVia` botLink = do
|
|
cc ##> ("/c " <> botLink)
|
|
cc <## "connection request sent!"
|
|
cc <## "broadcast_bot (Broadcast Bot): contact is connected"
|
|
cc <# "broadcast_bot> Hello! I am a broadcast bot."
|
|
cc <## "I broadcast messages to all connected users from @alice."
|