diff --git a/test/FediChordSpec.hs b/test/FediChordSpec.hs index bcc2eaf..ed1f3c8 100644 --- a/test/FediChordSpec.hs +++ b/test/FediChordSpec.hs @@ -1,4 +1,6 @@ -{-# LANGUAGE OverloadedStrings #-} +{-# LANGUAGE FlexibleInstances #-} +{-# LANGUAGE MultiParamTypeClasses #-} +{-# LANGUAGE OverloadedStrings #-} module FediChordSpec where import Control.Concurrent.STM.TVar @@ -292,12 +294,15 @@ exampleNodeState = RemoteNodeState { , vServerID = 0 } -exampleLocalNode :: IO (LocalNodeState s) -exampleLocalNode = nodeStateInit =<< (newTVarIO $ RealNode { +exampleLocalNode :: IO (LocalNodeState MockService) +exampleLocalNode = do + realNode <- newTVarIO $ RealNode { vservers = [] , nodeConfig = exampleFediConf , bootstrapNodes = confBootstrapNodes exampleFediConf - }) + , nodeService = MockService + } + nodeStateInit realNode exampleFediConf :: FediChordConf @@ -313,3 +318,9 @@ exampleVs :: (Integral i) => i exampleVs = 4 exampleIp :: HostAddress6 exampleIp = tupleToHostAddress6 (0x2001, 0x16b8, 0x755a, 0xb110, 0x7d6a, 0x12ab, 0xf0c5, 0x386e) + +data MockService d = MockService + +instance DHT d => Service MockService d where + runService _ _ = pure MockService + getListeningPortFromService = const 1337