mirrors_public/schollz_croc
1
1
Fork 0
mirror of https://github.com/schollz/croc.git synced 2025-10-10 21:01:02 +02:00
Code Activity

fix: use dynamic port

Browse source
This commit is contained in:
Zack Scholl 2025-09-26 13:43:59 -07:00
parent 990696e419
commit 3e68fa1897
1 changed files with 14 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

19
src/comm/comm_test.go
View file

@ -16,12 +16,21 @@ func TestComm(t *testing.T) {
t.Error(err)
}
port := "8001"
// Use dynamic port allocation to avoid conflicts
listener, err := net.Listen("tcp", "127.0.0.1:0")
if err != nil {
t.Fatal(err)
}
port := listener.Addr().(*net.TCPAddr).Port
portStr := listener.Addr().String()
listener.Close() // Close the listener so we can reopen it in the goroutine
go func() {
log.Debug("starting TCP server on " + port)
server, err := net.Listen("tcp", "0.0.0.0:"+port)
log.Debug("starting TCP server on " + portStr)
server, err := net.Listen("tcp", portStr)
if err != nil {
log.Error(err)
return
}
defer server.Close()
// spawn a new goroutine whenever a client connects
@ -31,7 +40,7 @@ func TestComm(t *testing.T) {
log.Error(err)
}
log.Debugf("client %s connected", connection.RemoteAddr().String())
go func(_ string, connection net.Conn) {
go func(_ int, connection net.Conn) {
c := New(connection)
err = c.Send([]byte("hello, world"))
assert.Nil(t, err)
@ -49,7 +58,7 @@ func TestComm(t *testing.T) {
}()
time.Sleep(300 * time.Millisecond)
a, err := NewConnection("127.0.0.1:"+port, 10*time.Minute)
a, err := NewConnection(portStr, 10*time.Minute)
assert.Nil(t, err)
data, err := a.Receive()
assert.Equal(t, []byte("hello, world"), data)