Disconnect commands

1 files changed, 5 insertions, 1 deletions

diff --git a/src/Run/Monad.hs b/src/Run/Monad.hs
index 221f6d7..5b55897 100644
--- a/src/Run/Monad.hs
+++ b/src/Run/Monad.hs
@@ -16,11 +16,13 @@ import Control.Monad.Except
 import Control.Monad.Reader
 
 import Data.Map (Map)
+import Data.Set (Set)
 import Data.Scientific
 import qualified Data.Text as T
 
 import {-# SOURCE #-} GDB
 import {-# SOURCE #-} Network
+import Network.Ip
 import Output
 import {-# SOURCE #-} Process
 import Test
@@ -39,7 +41,9 @@ data TestEnv = TestEnv
 data TestState = TestState
     { tsNetwork :: Network
     , tsVars :: [(VarName, SomeVarValue)]
-    , tsNodePacketLoss :: Map NodeName Scientific
+    , tsDisconnectedUp :: Set NetworkNamespace
+    , tsDisconnectedBridge :: Set NetworkNamespace
+    , tsNodePacketLoss :: Map NetworkNamespace Scientific
     }
 
 data TestOptions = TestOptions