add a test for dialing with a local and an unspecified listener

transport_test.go

@@ -84,3 +84,74 @@ func TestTwoLocal(t *testing.T) {
 	<-done
 	c.Close()
 }
+
+func TestLocalAndUnspecified(t *testing.T) {
+	var trA Transport
+	var trB Transport
+	laddr, _ := ma.NewMultiaddr("/ip4/127.0.0.1/tcp/0")
+	unspec, _ := ma.NewMultiaddr("/ip4/0.0.0.0/tcp/0")
+	listenerA, err := trA.Listen(laddr)
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer listenerA.Close()
+
+	listenerB1, err := trB.Listen(laddr)
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer listenerB1.Close()
+
+	listenerB2, err := trB.Listen(unspec)
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer listenerB2.Close()
+
+	done := make(chan struct{})
+	go func() {
+		defer close(done)
+		c, err := listenerA.Accept()
+		if err != nil {
+			t.Fatal(err)
+		}
+		c.Close()
+	}()
+
+	c, err := trB.Dial(listenerA.Multiaddr())
+	if err != nil {
+		t.Fatal(err)
+	}
+	actual := c.LocalAddr().(*net.TCPAddr).Port
+	expected := listenerB1.Addr().(*net.TCPAddr).Port
+	if actual != expected {
+		t.Errorf("expected to use port %d, used port %d", expected, actual)
+	}
+	<-done
+	c.Close()
+
+	// Closing the listener should reset the dialer.
+	listenerB1.Close()
+
+	done = make(chan struct{})
+	go func() {
+		defer close(done)
+		c, err := listenerA.Accept()
+		if err != nil {
+			t.Fatal(err)
+		}
+		c.Close()
+	}()
+
+	c, err = trB.Dial(listenerA.Multiaddr())
+	if err != nil {
+		t.Fatal(err)
+	}
+	actual = c.LocalAddr().(*net.TCPAddr).Port
+	expected = listenerB2.Addr().(*net.TCPAddr).Port
+	if actual != expected {
+		t.Errorf("expected to use port %d, used port %d", expected, actual)
+	}
+	<-done
+	c.Close()
+}