mirror of
https://github.com/simplex-chat/simplexmq.git
synced 2026-04-05 19:46:06 +00:00
330 lines
12 KiB
Haskell
330 lines
12 KiB
Haskell
{-# LANGUAGE BlockArguments #-}
|
|
{-# LANGUAGE DuplicateRecordFields #-}
|
|
{-# LANGUAGE OverloadedStrings #-}
|
|
{-# LANGUAGE QuasiQuotes #-}
|
|
|
|
module AgentTests.SQLiteTests (storeTests) where
|
|
|
|
import Control.Monad.Except (ExceptT, runExceptT)
|
|
import qualified Crypto.PubKey.RSA as R
|
|
import Data.Text.Encoding (encodeUtf8)
|
|
import Data.Time
|
|
import Data.Word (Word32)
|
|
import qualified Database.SQLite.Simple as DB
|
|
import Database.SQLite.Simple.QQ (sql)
|
|
import Simplex.Messaging.Agent.Store
|
|
import Simplex.Messaging.Agent.Store.SQLite
|
|
import Simplex.Messaging.Agent.Transmission
|
|
import qualified Simplex.Messaging.Crypto as C
|
|
import System.Random (Random (randomIO))
|
|
import Test.Hspec
|
|
import UnliftIO.Directory (removeFile)
|
|
|
|
testDB :: String
|
|
testDB = "smp-agent.test.db"
|
|
|
|
withStore :: SpecWith SQLiteStore -> Spec
|
|
withStore = before createStore . after removeStore
|
|
where
|
|
createStore :: IO SQLiteStore
|
|
createStore = do
|
|
-- Randomize DB file name to avoid SQLite IO errors supposedly caused by asynchronous
|
|
-- IO operations on multiple similarly named files; error seems to be environment specific
|
|
r <- randomIO :: IO Word32
|
|
newSQLiteStore $ testDB <> show r
|
|
|
|
removeStore :: SQLiteStore -> IO ()
|
|
removeStore store = do
|
|
DB.close $ dbConn store
|
|
removeFile $ dbFilename store
|
|
|
|
returnsResult :: (Eq a, Eq e, Show a, Show e) => ExceptT e IO a -> a -> Expectation
|
|
action `returnsResult` r = runExceptT action `shouldReturn` Right r
|
|
|
|
throwsError :: (Eq a, Eq e, Show a, Show e) => ExceptT e IO a -> e -> Expectation
|
|
action `throwsError` e = runExceptT action `shouldReturn` Left e
|
|
|
|
-- TODO add null port tests
|
|
storeTests :: Spec
|
|
storeTests = withStore do
|
|
describe "foreign keys enabled" testForeignKeysEnabled
|
|
describe "store methods" do
|
|
describe "createRcvConn" testCreateRcvConn
|
|
describe "createSndConn" testCreateSndConn
|
|
describe "getRcvQueue" testGetRcvQueue
|
|
describe "deleteConn" do
|
|
describe "rcv" testDeleteRcvConn
|
|
describe "snd" testDeleteSndConn
|
|
describe "duplex" testDeleteDuplexConn
|
|
describe "upgradeRcvConnToDuplex" testUpgradeRcvConnToDuplex
|
|
describe "upgradeSndConnToDuplex" testUpgradeSndConnToDuplex
|
|
describe "set queue status" do
|
|
describe "setRcvQueueStatus" testSetRcvQueueStatus
|
|
describe "setSndQueueStatus" testSetSndQueueStatus
|
|
describe "duplex connection" testSetQueueStatusDuplex
|
|
xdescribe "rcv queue doesn't exist" testSetRcvQueueStatusNoQueue
|
|
xdescribe "snd queue doesn't exist" testSetSndQueueStatusNoQueue
|
|
describe "createRcvMsg" do
|
|
describe "rcv queue exists" testCreateRcvMsg
|
|
describe "rcv queue doesn't exist" testCreateRcvMsgNoQueue
|
|
describe "createSndMsg" do
|
|
describe "snd queue exists" testCreateSndMsg
|
|
describe "snd queue doesn't exist" testCreateSndMsgNoQueue
|
|
|
|
testForeignKeysEnabled :: SpecWith SQLiteStore
|
|
testForeignKeysEnabled = do
|
|
it "should throw error if foreign keys are enabled" $ \store -> do
|
|
let inconsistentQuery =
|
|
[sql|
|
|
INSERT INTO connections
|
|
(conn_alias, rcv_host, rcv_port, rcv_id, snd_host, snd_port, snd_id)
|
|
VALUES
|
|
("conn1", "smp.simplex.im", "5223", "1234", "smp.simplex.im", "5223", "2345");
|
|
|]
|
|
DB.execute_ (dbConn store) inconsistentQuery
|
|
`shouldThrow` (\e -> DB.sqlError e == DB.ErrorConstraint)
|
|
|
|
rcvQueue1 :: ReceiveQueue
|
|
rcvQueue1 =
|
|
ReceiveQueue
|
|
{ server = SMPServer "smp.simplex.im" (Just "5223") (Just "1234"),
|
|
rcvId = "1234",
|
|
connAlias = "conn1",
|
|
rcvPrivateKey = C.PrivateKey 1 2 3,
|
|
sndId = Just "2345",
|
|
sndKey = Nothing,
|
|
decryptKey = C.PrivateKey 1 2 3,
|
|
verifyKey = Nothing,
|
|
status = New
|
|
}
|
|
|
|
sndQueue1 :: SendQueue
|
|
sndQueue1 =
|
|
SendQueue
|
|
{ server = SMPServer "smp.simplex.im" (Just "5223") (Just "1234"),
|
|
sndId = "3456",
|
|
connAlias = "conn1",
|
|
sndPrivateKey = C.PrivateKey 1 2 3,
|
|
encryptKey = C.PublicKey $ R.PublicKey 1 2 3,
|
|
signKey = C.PrivateKey 1 2 3,
|
|
status = New
|
|
}
|
|
|
|
testCreateRcvConn :: SpecWith SQLiteStore
|
|
testCreateRcvConn = do
|
|
it "should create receive connection and add send queue" $ \store -> do
|
|
createRcvConn store rcvQueue1
|
|
`returnsResult` ()
|
|
getConn store "conn1"
|
|
`returnsResult` SomeConn SCReceive (ReceiveConnection "conn1" rcvQueue1)
|
|
upgradeRcvConnToDuplex store "conn1" sndQueue1
|
|
`returnsResult` ()
|
|
getConn store "conn1"
|
|
`returnsResult` SomeConn SCDuplex (DuplexConnection "conn1" rcvQueue1 sndQueue1)
|
|
|
|
testCreateSndConn :: SpecWith SQLiteStore
|
|
testCreateSndConn = do
|
|
it "should create send connection and add receive queue" $ \store -> do
|
|
createSndConn store sndQueue1
|
|
`returnsResult` ()
|
|
getConn store "conn1"
|
|
`returnsResult` SomeConn SCSend (SendConnection "conn1" sndQueue1)
|
|
upgradeSndConnToDuplex store "conn1" rcvQueue1
|
|
`returnsResult` ()
|
|
getConn store "conn1"
|
|
`returnsResult` SomeConn SCDuplex (DuplexConnection "conn1" rcvQueue1 sndQueue1)
|
|
|
|
testGetRcvQueue :: SpecWith SQLiteStore
|
|
testGetRcvQueue = do
|
|
it "should get receive queue" $ \store -> do
|
|
let smpServer = SMPServer "smp.simplex.im" (Just "5223") (Just "1234")
|
|
let recipientId = "1234"
|
|
createRcvConn store rcvQueue1
|
|
`returnsResult` ()
|
|
getRcvQueue store smpServer recipientId
|
|
`returnsResult` rcvQueue1
|
|
|
|
testDeleteRcvConn :: SpecWith SQLiteStore
|
|
testDeleteRcvConn = do
|
|
it "should create receive connection and delete it" $ \store -> do
|
|
createRcvConn store rcvQueue1
|
|
`returnsResult` ()
|
|
getConn store "conn1"
|
|
`returnsResult` SomeConn SCReceive (ReceiveConnection "conn1" rcvQueue1)
|
|
deleteConn store "conn1"
|
|
`returnsResult` ()
|
|
-- TODO check queues are deleted as well
|
|
getConn store "conn1"
|
|
`throwsError` SEBadConn
|
|
|
|
testDeleteSndConn :: SpecWith SQLiteStore
|
|
testDeleteSndConn = do
|
|
it "should create send connection and delete it" $ \store -> do
|
|
createSndConn store sndQueue1
|
|
`returnsResult` ()
|
|
getConn store "conn1"
|
|
`returnsResult` SomeConn SCSend (SendConnection "conn1" sndQueue1)
|
|
deleteConn store "conn1"
|
|
`returnsResult` ()
|
|
-- TODO check queues are deleted as well
|
|
getConn store "conn1"
|
|
`throwsError` SEBadConn
|
|
|
|
testDeleteDuplexConn :: SpecWith SQLiteStore
|
|
testDeleteDuplexConn = do
|
|
it "should create duplex connection and delete it" $ \store -> do
|
|
createRcvConn store rcvQueue1
|
|
`returnsResult` ()
|
|
upgradeRcvConnToDuplex store "conn1" sndQueue1
|
|
`returnsResult` ()
|
|
getConn store "conn1"
|
|
`returnsResult` SomeConn SCDuplex (DuplexConnection "conn1" rcvQueue1 sndQueue1)
|
|
deleteConn store "conn1"
|
|
`returnsResult` ()
|
|
-- TODO check queues are deleted as well
|
|
getConn store "conn1"
|
|
`throwsError` SEBadConn
|
|
|
|
testUpgradeRcvConnToDuplex :: SpecWith SQLiteStore
|
|
testUpgradeRcvConnToDuplex = do
|
|
it "should throw error on attempts to add send queue to SendConnection or DuplexConnection" $ \store -> do
|
|
createSndConn store sndQueue1
|
|
`returnsResult` ()
|
|
let anotherSndQueue =
|
|
SendQueue
|
|
{ server = SMPServer "smp.simplex.im" (Just "5223") (Just "1234"),
|
|
sndId = "2345",
|
|
connAlias = "conn1",
|
|
sndPrivateKey = C.PrivateKey 1 2 3,
|
|
encryptKey = C.PublicKey $ R.PublicKey 1 2 3,
|
|
signKey = C.PrivateKey 1 2 3,
|
|
status = New
|
|
}
|
|
upgradeRcvConnToDuplex store "conn1" anotherSndQueue
|
|
`throwsError` SEBadConnType CSend
|
|
upgradeSndConnToDuplex store "conn1" rcvQueue1
|
|
`returnsResult` ()
|
|
upgradeRcvConnToDuplex store "conn1" anotherSndQueue
|
|
`throwsError` SEBadConnType CDuplex
|
|
|
|
testUpgradeSndConnToDuplex :: SpecWith SQLiteStore
|
|
testUpgradeSndConnToDuplex = do
|
|
it "should throw error on attempts to add receive queue to ReceiveConnection or DuplexConnection" $ \store -> do
|
|
createRcvConn store rcvQueue1
|
|
`returnsResult` ()
|
|
let anotherRcvQueue =
|
|
ReceiveQueue
|
|
{ server = SMPServer "smp.simplex.im" (Just "5223") (Just "1234"),
|
|
rcvId = "3456",
|
|
connAlias = "conn1",
|
|
rcvPrivateKey = C.PrivateKey 1 2 3,
|
|
sndId = Just "4567",
|
|
sndKey = Nothing,
|
|
decryptKey = C.PrivateKey 1 2 3,
|
|
verifyKey = Nothing,
|
|
status = New
|
|
}
|
|
upgradeSndConnToDuplex store "conn1" anotherRcvQueue
|
|
`throwsError` SEBadConnType CReceive
|
|
upgradeRcvConnToDuplex store "conn1" sndQueue1
|
|
`returnsResult` ()
|
|
upgradeSndConnToDuplex store "conn1" anotherRcvQueue
|
|
`throwsError` SEBadConnType CDuplex
|
|
|
|
testSetRcvQueueStatus :: SpecWith SQLiteStore
|
|
testSetRcvQueueStatus = do
|
|
it "should update status of receive queue" $ \store -> do
|
|
createRcvConn store rcvQueue1
|
|
`returnsResult` ()
|
|
getConn store "conn1"
|
|
`returnsResult` SomeConn SCReceive (ReceiveConnection "conn1" rcvQueue1)
|
|
setRcvQueueStatus store rcvQueue1 Confirmed
|
|
`returnsResult` ()
|
|
getConn store "conn1"
|
|
`returnsResult` SomeConn SCReceive (ReceiveConnection "conn1" rcvQueue1 {status = Confirmed})
|
|
|
|
testSetSndQueueStatus :: SpecWith SQLiteStore
|
|
testSetSndQueueStatus = do
|
|
it "should update status of send queue" $ \store -> do
|
|
createSndConn store sndQueue1
|
|
`returnsResult` ()
|
|
getConn store "conn1"
|
|
`returnsResult` SomeConn SCSend (SendConnection "conn1" sndQueue1)
|
|
setSndQueueStatus store sndQueue1 Confirmed
|
|
`returnsResult` ()
|
|
getConn store "conn1"
|
|
`returnsResult` SomeConn SCSend (SendConnection "conn1" sndQueue1 {status = Confirmed})
|
|
|
|
testSetQueueStatusDuplex :: SpecWith SQLiteStore
|
|
testSetQueueStatusDuplex = do
|
|
it "should update statuses of receive and send queues in duplex connection" $ \store -> do
|
|
createRcvConn store rcvQueue1
|
|
`returnsResult` ()
|
|
upgradeRcvConnToDuplex store "conn1" sndQueue1
|
|
`returnsResult` ()
|
|
getConn store "conn1"
|
|
`returnsResult` SomeConn SCDuplex (DuplexConnection "conn1" rcvQueue1 sndQueue1)
|
|
setRcvQueueStatus store rcvQueue1 Secured
|
|
`returnsResult` ()
|
|
getConn store "conn1"
|
|
`returnsResult` SomeConn SCDuplex (DuplexConnection "conn1" rcvQueue1 {status = Secured} sndQueue1)
|
|
setSndQueueStatus store sndQueue1 Confirmed
|
|
`returnsResult` ()
|
|
getConn store "conn1"
|
|
`returnsResult` SomeConn
|
|
SCDuplex
|
|
( DuplexConnection "conn1" rcvQueue1 {status = Secured} sndQueue1 {status = Confirmed}
|
|
)
|
|
|
|
testSetRcvQueueStatusNoQueue :: SpecWith SQLiteStore
|
|
testSetRcvQueueStatusNoQueue = do
|
|
it "should throw error on attempt to update status of nonexistent receive queue" $ \store -> do
|
|
setRcvQueueStatus store rcvQueue1 Confirmed
|
|
`throwsError` SEInternal
|
|
|
|
testSetSndQueueStatusNoQueue :: SpecWith SQLiteStore
|
|
testSetSndQueueStatusNoQueue = do
|
|
it "should throw error on attempt to update status of nonexistent send queue" $ \store -> do
|
|
setSndQueueStatus store sndQueue1 Confirmed
|
|
`throwsError` SEInternal
|
|
|
|
testCreateRcvMsg :: SpecWith SQLiteStore
|
|
testCreateRcvMsg = do
|
|
it "should create a rcv message" $ \store -> do
|
|
createRcvConn store rcvQueue1
|
|
`returnsResult` ()
|
|
-- TODO getMsg to check message
|
|
let ts = UTCTime (fromGregorian 2021 02 24) (secondsToDiffTime 0)
|
|
createRcvMsg store "conn1" (encodeUtf8 "Hello world!") 1 ts "1" ts
|
|
`returnsResult` ()
|
|
|
|
testCreateRcvMsgNoQueue :: SpecWith SQLiteStore
|
|
testCreateRcvMsgNoQueue = do
|
|
it "should throw error on attempt to create a rcv message w/t a rcv queue" $ \store -> do
|
|
let ts = UTCTime (fromGregorian 2021 02 24) (secondsToDiffTime 0)
|
|
createRcvMsg store "conn1" (encodeUtf8 "Hello world!") 1 ts "1" ts
|
|
`throwsError` SEBadConn
|
|
createSndConn store sndQueue1
|
|
`returnsResult` ()
|
|
createRcvMsg store "conn1" (encodeUtf8 "Hello world!") 1 ts "1" ts
|
|
`throwsError` SEBadConnType CSend
|
|
|
|
testCreateSndMsg :: SpecWith SQLiteStore
|
|
testCreateSndMsg = do
|
|
it "should create a snd message" $ \store -> do
|
|
createSndConn store sndQueue1
|
|
`returnsResult` ()
|
|
-- TODO getMsg to check message
|
|
createSndMsg store "conn1" (encodeUtf8 "Hello world!")
|
|
`returnsResult` ()
|
|
|
|
testCreateSndMsgNoQueue :: SpecWith SQLiteStore
|
|
testCreateSndMsgNoQueue = do
|
|
it "should throw error on attempt to create a snd message w/t a snd queue" $ \store -> do
|
|
createSndMsg store "conn1" (encodeUtf8 "Hello world!")
|
|
`throwsError` SEBadConn
|
|
createRcvConn store rcvQueue1
|
|
`returnsResult` ()
|
|
createSndMsg store "conn1" (encodeUtf8 "Hello world!")
|
|
`throwsError` SEBadConnType CReceive
|