refactoring #40 | use Millisecond instead of Second for shorter tests

cardano-foundation · Mar 11, 2019 · 66f24a6 · 66f24a6
1 parent ab904ec
commit 66f24a6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/src/Cardano/Wallet/BlockSyncer.hs b/src/Cardano/Wallet/BlockSyncer.hs
@@ -22,7 +22,7 @@ import Cardano.Wallet.Primitive
 import Control.Concurrent
     ( threadDelay )
 import Data.Time.Units
-    ( Second, toMicroseconds )
+    ( Millisecond, toMicroseconds )
 
 import qualified Data.List as L
 
@@ -37,7 +37,7 @@ tickingFunction
     -- ^ a way to get a new block
     -> (Block -> IO ())
     -- ^ action taken on a new block
-    -> Second
+    -> Millisecond
     -- ^ tick time
     -> BlockHeadersConsumed
     -> IO ()

diff --git a/test/unit/Cardano/Wallet/BlockSyncerSpec.hs b/test/unit/Cardano/Wallet/BlockSyncerSpec.hs
@@ -28,7 +28,7 @@ import Data.ByteString
 import Data.Map.Strict
     ( Map )
 import Data.Time.Units
-    ( Second, fromMicroseconds )
+    ( Millisecond, fromMicroseconds )
 import Data.Tuple
     ( swap )
 import Test.Hspec
@@ -68,6 +68,7 @@ tickingFunctionTest
     :: (TickingTime, Blocks)
     -> Property
 tickingFunctionTest (TickingTime tickTime, Blocks blocks) = monadicIO $ liftIO $ do
+    print blocks
     (readerChan, reader) <- mkReader
     (writerChan, writer) <- mkWriter blocks
     waitFor writerChan $ tickingFunction writer reader tickTime (BlockHeadersConsumed [])
@@ -118,13 +119,13 @@ mkReader = do
 -------------------------------------------------------------------------------}
 
 
-newtype TickingTime = TickingTime Second
+newtype TickingTime = TickingTime Millisecond
     deriving (Show)
 
 instance Arbitrary TickingTime where
     -- No shrinking
     arbitrary = do
-        tickTime <- fromMicroseconds . (* (1000 * 1000)) <$> choose (1, 3)
+        tickTime <- fromMicroseconds . (* 1000) <$> choose (50, 100)
         return $ TickingTime tickTime