Use internal/pkg structure

1 files changed, 0 insertions, 52 deletions

diff --git a/network/network_test.go b/network/network_test.go
deleted file mode 100644
index 9dbc416..0000000
--- a/network/network_test.go
+++ /dev/null
@@ -1,52 +0,0 @@
-package network
-
-import (
-	"testing"
-	"time"
-
-	"github.com/stretchr/testify/assert"
-)
-
-// TestPeerToPeerCommunication tests if two peers can communicate.
-func TestPeerToPeerCommunication(t *testing.T) {
-	// Create a mock of the first peer (peer-1)
-	peer1IP := "127.0.0.1"
-	peer1Port := 9001
-	peer1 := NewTCPNetwork("peer-1", peer1IP, peer1Port)
-
-	// Create a mock of the second peer (peer-2)
-	peer2IP := "127.0.0.1"
-	peer2Port := 9002
-	peer2 := NewTCPNetwork("peer-2", peer2IP, peer2Port)
-
-	// Register a message handler on peer-2 to receive the message from peer-1
-	peer2.RegisterHandler("chat", func(msg Message) {
-		assert.Equal(t, "peer-1", msg.Source.ID)
-		assert.Equal(t, "Hey from peer-1!", string(msg.Payload))
-	})
-
-	// Start the first peer and add the second peer
-	go peer1.AddPeer("peer-2", peer2IP, peer2Port)
-	go peer2.AddPeer("peer-1", peer1IP, peer1Port)
-
-	// Wait for the connections to be established
-	// You might need a little more time based on network delay and retry logic
-	time.Sleep(5 * time.Second)
-
-	// Send a message from peer-1 to peer-2
-	err := peer1.Send("peer-2", "chat", []byte("Hey from peer-1!"))
-	assert.NoError(t, err)
-
-	// Allow some time for the message to be received and handled
-	time.Sleep(2 * time.Second)
-}
-
-// TestSendFailure tests if sending a message fails when no connection exists.
-func TestSendFailure(t *testing.T) {
-	peer1 := NewTCPNetwork("peer-1", "127.0.0.1", 9001)
-	_ = NewTCPNetwork("peer-2", "127.0.0.1", 9002)
-
-	// Attempt to send a message without establishing a connection first
-	err := peer1.Send("peer-2", "chat", []byte("Message without connection"))
-	assert.Error(t, err, "Expected error when sending to a non-connected peer")
-}