@@ -4,6 +4,7 @@ module NixSerializerSpec (spec) where
44
55import Crypto.Hash (MD5 , SHA1 , SHA256 , SHA512 )
66import Data.Dependent.Sum (DSum ((:=>) ))
7+ import Data.Some (Some (Some ))
78import Data.Time (UTCTime )
89import Data.Word (Word64 )
910import Test.Hspec (Expectation , Spec , describe , it , parallel , shouldBe )
@@ -21,11 +22,8 @@ import System.Nix.Store.Remote.Serializer
2122import System.Nix.Store.Remote.Types.Logger (Logger (.. ))
2223import System.Nix.Store.Remote.Types.ProtoVersion (HasProtoVersion (.. ), ProtoVersion (.. ))
2324import System.Nix.Store.Remote.Types.StoreConfig (TestStoreConfig )
24- import System.Nix.Store.Remote.Types.WorkerOp (WorkerOp (.. ))
25-
26- -- WIP
27- import Data.Some (Some (Some ))
2825import System.Nix.Store.Remote.Types.StoreRequest (StoreRequest (.. ))
26+ import System.Nix.Store.Remote.Types.WorkerOp (WorkerOp (.. ))
2927
3028-- | Test for roundtrip using @NixSerializer@
3129roundtripSReader
@@ -139,17 +137,17 @@ spec = parallel $ do
139137
140138 prop " StoreRequest"
141139 $ \ testStoreConfig ->
142- forAll (arbitrary `suchThat` (hacks (hasProtoVersion testStoreConfig)))
140+ forAll (arbitrary `suchThat` (restrictProtoVersion (hasProtoVersion testStoreConfig)))
143141 $ roundtripSReader @ TestStoreConfig storeRequest testStoreConfig
144142
145- hacks :: ProtoVersion -> Some StoreRequest -> Bool
146- hacks v (Some (BuildPaths _ _)) | v < ProtoVersion 1 30 = False
147- hacks _ (Some (BuildDerivation _ drv _)) = inputDrvs drv == mempty
148- hacks v (Some (QueryMissing _)) | v < ProtoVersion 1 30 = False
149- hacks _ _ = True
143+ restrictProtoVersion :: ProtoVersion -> Some StoreRequest -> Bool
144+ restrictProtoVersion v (Some (BuildPaths _ _)) | v < ProtoVersion 1 30 = False
145+ restrictProtoVersion _ (Some (BuildDerivation _ drv _)) = inputDrvs drv == mempty
146+ restrictProtoVersion v (Some (QueryMissing _)) | v < ProtoVersion 1 30 = False
147+ restrictProtoVersion _ _ = True
150148
151149errorInfoIf :: Bool -> Logger -> Bool
152- errorInfoIf True (Logger_Error (Right _)) = True
153- errorInfoIf False (Logger_Error (Left _)) = True
154- errorInfoIf _ (Logger_Error _) = False
155- errorInfoIf _ _ = True
150+ errorInfoIf True (Logger_Error (Right _)) = True
151+ errorInfoIf False (Logger_Error (Left _)) = True
152+ errorInfoIf _ (Logger_Error _) = False
153+ errorInfoIf _ _ = True
0 commit comments