package transport import ( "context" "os" "testing" ) func TestMemoryTransport(t *testing.T) { tr := NewTransport() ctx := context.Background() // bind / listen l, err := tr.Listen(ctx, "127.0.0.1:8080") if err != nil { t.Fatalf("Unexpected error listening %v", err) } defer l.Close() cherr := make(chan error, 1) // accept go func() { if nerr := l.Accept(func(sock Socket) { for { var m Message if rerr := sock.Recv(&m); rerr != nil { cherr <- rerr return } if len(os.Getenv("INTEGRATION_TESTS")) == 0 { t.Logf("Server Received %s", string(m.Body)) } if cerr := sock.Send(&Message{ Body: []byte(`pong`), }); cerr != nil { cherr <- cerr return } } }); nerr != nil { cherr <- err } }() // dial c, err := tr.Dial(ctx, "127.0.0.1:8080") if err != nil { t.Fatalf("Unexpected error dialing %v", err) } defer c.Close() select { case err := <-cherr: t.Fatal(err) default: // send <=> receive for i := 0; i < 3; i++ { if err := c.Send(&Message{ Body: []byte(`ping`), }); err != nil { return } var m Message if err := c.Recv(&m); err != nil { return } if len(os.Getenv("INTEGRATION_TESTS")) == 0 { t.Logf("Client Received %s", string(m.Body)) } } } } func TestListener(t *testing.T) { tr := NewTransport() ctx := context.Background() // bind / listen on random port l, err := tr.Listen(ctx, ":0") if err != nil { t.Fatalf("Unexpected error listening %v", err) } defer l.Close() // try again l2, err := tr.Listen(ctx, ":0") if err != nil { t.Fatalf("Unexpected error listening %v", err) } defer l2.Close() // now make sure it still fails l3, err := tr.Listen(ctx, ":8080") if err != nil { t.Fatalf("Unexpected error listening %v", err) } defer l3.Close() if _, err := tr.Listen(ctx, ":8080"); err == nil { t.Fatal("Expected error binding to :8080 got nil") } }