mirror of https://github.com/voidlizard/hbs2
49 lines
1.1 KiB
Haskell
49 lines
1.1 KiB
Haskell
module TestBlockInfoActor where
|
|
|
|
import HBS2.Hash
|
|
import HBS2.Clock
|
|
import HBS2.Net.Proto
|
|
import HBS2.Net.Proto.Actors.BlockInfo
|
|
import HBS2.Net.PeerLocator
|
|
import HBS2.Net.PeerLocator.Static
|
|
|
|
import Test.Tasty.HUnit
|
|
|
|
import Test.QuickCheck
|
|
import Data.Word
|
|
import Data.Hashable (Hashable)
|
|
import Data.ByteString (ByteString)
|
|
import Data.ByteString qualified as B
|
|
import Control.Concurrent.Async
|
|
|
|
data Fake
|
|
|
|
instance IsPeer Fake where
|
|
newtype instance Peer Fake = FakePeer Int
|
|
deriving stock (Eq,Ord,Show)
|
|
deriving newtype (Hashable,Num,Enum,Real,Integral)
|
|
|
|
testBlockInfoActor :: IO ()
|
|
testBlockInfoActor = do
|
|
|
|
|
|
np <- newStaticPeerLocator @Fake [1..10]
|
|
|
|
a <- createBlockInfoActor (AnyPeerLocator np)
|
|
actor <- async $ runBlockInfoActor a
|
|
|
|
let obj = shrink [0x00 .. 0xFF] :: [[Word8]]
|
|
|
|
forConcurrently_ obj $ \x -> do
|
|
requestBlockInfo @Fake a Nothing (hashObject (B.pack x) :: Hash HbSync)
|
|
|
|
pause ( 1 :: Timeout 'Seconds)
|
|
|
|
stopBlockInfoActor a
|
|
|
|
waitAnyCatchCancel [actor]
|
|
|
|
assertBool "testBlockInfoActor" True
|
|
|
|
|