// Copyright (c) Tailscale Inc & AUTHORS // SPDX-License-Identifier: BSD-3-Clause package tailfs import ( "log" "net" "testing" ) func TestConnListener(t *testing.T) { l, err := net.Listen("tcp", "127.0.0.1:") if err != nil { t.Fatalf("failed to Listen: %s", err) } cl := newConnListener() // Test that we can accept a connection cc, err := net.Dial("tcp", l.Addr().String()) if err != nil { t.Fatalf("failed to Dial: %s", err) } defer cc.Close() sc, err := l.Accept() if err != nil { t.Fatalf("failed to Accept: %s", err) } remoteAddr := &net.TCPAddr{IP: net.ParseIP("10.10.10.10"), Port: 1234} go func() { err := cl.HandleConn(sc, remoteAddr) if err != nil { log.Printf("failed to HandleConn: %s", err) } }() clc, err := cl.Accept() if err != nil { t.Fatalf("failed to Accept: %s", err) } defer clc.Close() if clc.RemoteAddr().String() != remoteAddr.String() { t.Fatalf("ConnListener accepted the wrong connection, got %q, want %q", clc.RemoteAddr(), remoteAddr) } err = cl.Close() if err != nil { t.Fatalf("failed to Close: %s", err) } err = cl.Close() if err == nil { t.Fatal("should have failed on second Close") } err = cl.HandleConn(sc, remoteAddr) if err == nil { t.Fatal("should have failed on HandleConn after Close") } _, err = cl.Accept() if err == nil { t.Fatal("should have failed on Accept after Close") } }