Move peer handler integration test

libsv · Jul 19, 2024 · 5131e14 · 5131e14
1 parent c5ef4a2
commit 5131e14
Show file tree

Hide file tree

Showing 2 changed files with 58 additions and 37 deletions.
diff --git a/test/peer_integration_test.go b/test/peer_integration_test.go
@@ -148,41 +148,4 @@ func TestNewPeer(t *testing.T) {
 		t.Log("shutdown finished")
 	})
 
-	t.Run("announce transaction", func(t *testing.T) {
-		logger := slog.New(slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{Level: slog.LevelDebug}))
-
-		pm := p2p.NewPeerManager(logger, wire.TestNet)
-		require.NotNil(t, pm)
-
-		peerHandler := &p2p.PeerHandlerIMock{
-			HandleTransactionsGetFunc: func(msgs []*wire.InvVect, peer p2p.PeerI) ([][]byte, error) {
-				return [][]byte{TX1RawBytes}, nil
-			},
-		}
-
-		peer, err := p2p.NewPeer(logger, "localhost:"+p2pPortBinding, peerHandler, wire.TestNet)
-		require.NoError(t, err)
-
-		err = pm.AddPeer(peer)
-		require.NoError(t, err)
-
-		t.Log("expect that peer has connected")
-	connectLoop:
-		for {
-			select {
-			case <-time.NewTicker(200 * time.Millisecond).C:
-				if peer.Connected() {
-					break connectLoop
-				}
-			case <-time.NewTimer(5 * time.Second).C:
-				t.Fatal("peer did not disconnect")
-			}
-		}
-
-		pm.AnnounceTransaction(TX1Hash, []p2p.PeerI{peer})
-
-		time.Sleep(100 * time.Millisecond)
-
-		peer.Shutdown()
-	})
 }
diff --git a/test/peer_manager_integration_test.go b/test/peer_manager_integration_test.go
@@ -0,0 +1,58 @@
+package test
+
+import (
+	"log/slog"
+	"os"
+	"testing"
+	"time"
+
+	"github.com/libsv/go-p2p"
+
+	"github.com/libsv/go-p2p/wire"
+	"github.com/stretchr/testify/require"
+)
+
+func TestNewPeerManager(t *testing.T) {
+	t.Helper()
+	if testing.Short() {
+		t.Skip("skipping integration test")
+	}
+
+	t.Run("announce transaction", func(t *testing.T) {
+		logger := slog.New(slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{Level: slog.LevelDebug}))
+
+		pm := p2p.NewPeerManager(logger, wire.TestNet)
+		require.NotNil(t, pm)
+
+		peerHandler := &p2p.PeerHandlerIMock{
+			HandleTransactionsGetFunc: func(msgs []*wire.InvVect, peer p2p.PeerI) ([][]byte, error) {
+				return [][]byte{TX1RawBytes}, nil
+			},
+		}
+
+		peer, err := p2p.NewPeer(logger, "localhost:"+p2pPortBinding, peerHandler, wire.TestNet)
+		require.NoError(t, err)
+
+		err = pm.AddPeer(peer)
+		require.NoError(t, err)
+
+		t.Log("expect that peer has connected")
+	connectLoop:
+		for {
+			select {
+			case <-time.NewTicker(200 * time.Millisecond).C:
+				if peer.Connected() {
+					break connectLoop
+				}
+			case <-time.NewTimer(5 * time.Second).C:
+				t.Fatal("peer did not disconnect")
+			}
+		}
+
+		pm.AnnounceTransaction(TX1Hash, []p2p.PeerI{peer})
+
+		time.Sleep(100 * time.Millisecond)
+
+		peer.Shutdown()
+	})
+}