m/go-agwpe
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use fake server for engine connect and disconnect tests

Browse Source
This commit is contained in:
Michael Smith
2026-05-20 11:16:59 +02:00
parent 36192cbd89
commit 64f902ba9d
1 changed files with 21 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
agwpe_test.go
View File

@@ -1,6 +1,7 @@
package agwpe package agwpe
import ( import (
"net"
"testing" "testing"
"time" "time"
) )
@@ -30,10 +31,17 @@ func TestNewPacketEngine(t *testing.T) {
} }
} }
func TestNewPacketEngine_Connect(t *testing.T) { func TestPacketEngine_Connect(t *testing.T) {
listener, err := net.Listen("tcp", "127.0.0.1:0")
if err != nil {
t.Fatal(err)
}
defer listener.Close()
serverAddr := listener.Addr().String()
// Successful connect // Successful connect
engine, _ := NewPacketEngine("localhost:1234") engine, _ := NewPacketEngine(serverAddr)
err := engine.Connect() err = engine.Connect()
if err != nil { if err != nil {
t.Errorf("Did not expect error: %v", err) t.Errorf("Did not expect error: %v", err)
} }
@@ -46,12 +54,19 @@ func TestNewPacketEngine_Connect(t *testing.T) {
} }
} }
func TestNewPacketEngine_Disconnect(t *testing.T) { func TestPacketEngine_Disconnect(t *testing.T) {
engine, _ := NewPacketEngine("localhost:1234") listener, err := net.Listen("tcp", "127.0.0.1:0")
if err != nil {
t.Fatal(err)
}
defer listener.Close()
serverAddr := listener.Addr().String()
engine, _ := NewPacketEngine(serverAddr)
engine.Connect() engine.Connect()
// Disconnect connected engine // Disconnect connected engine
err := engine.Disconnect() err = engine.Disconnect()
if err != nil { if err != nil {
t.Errorf("Did not expect error: %v", err) t.Errorf("Did not expect error: %v", err)
} }