Test: refactor successful attach to exported function

1 files changed, 11 insertions, 0 deletions

diff --git a/main/Test.hs b/main/Test.hs
index 6ee75bd..b06ed61 100644
--- a/main/Test.hs
+++ b/main/Test.hs
@@ -320,6 +320,7 @@ commands =
     , ( "peer-add", cmdPeerAdd )
     , ( "peer-drop", cmdPeerDrop )
     , ( "peer-list", cmdPeerList )
+    , ( "peer-get-by-addr", cmdPeerGetByAddr )
     , ( "test-message-send", cmdTestMessageSend )
     , ( "test-stream-open", cmdTestStreamOpen )
     , ( "test-stream-close", cmdTestStreamClose )
@@ -748,6 +749,16 @@ cmdPeerList = do
             ]
     cmdOut "peer-list-done"
 
+cmdPeerGetByAddr :: Command
+cmdPeerGetByAddr = do
+    Just RunningServer {..} <- gets tsServer
+    addr <- T.unwords <$> asks tiParams
+    ( _, peers ) <- liftIO $ readMVar rsPeers
+    forM_ peers $ \tpeer -> do
+        paddr <- getPeerAddress $ tpPeer tpeer
+        when (T.pack (show paddr) == addr) $ do
+            cmdOut $ "peer-get-by-addr " <> show (tpIndex tpeer)
+
 
 cmdTestMessageSend :: Command
 cmdTestMessageSend = do