mirror of
https://github.com/waku-org/go-multiaddr.git
synced 2025-02-23 03:28:12 +00:00
c90ef4472f
This commit changes the struct to a new Multiaddr interface:
```Go
type Multiaddr interface {
Equal(Multiaddr) bool
Bytes() []byte
String() string
Protocols() []*Protocol
Encapsulate(Multiaddr) Multiaddr
Decapsulate(Multiaddr) Multiaddr
}
```
This means a few things have changed:
- use Multiaddr interface, struct not exported
- Bytes returns a copy of the internal bytes
- Some methods no longer return errors (catch errors in NewMultiaddr)
- String (panics if malformed)
- Protocols (panics if malformed)
- Decapsulate (no-op if not prefix)
- Moved net-specific functions to package
- Multiaddr.DialArgs() -> DialArgs(Multiaddr)
- Multiaddr.IsThinWaist() -> IsThinWaist(Multiaddr)
cc @whyrusleeping @perfmode
70 lines
1.4 KiB
Go
70 lines
1.4 KiB
Go
package multiaddr
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
)
|
|
|
|
type GenFunc func() (Multiaddr, error)
|
|
|
|
func testConvert(t *testing.T, s string, gen GenFunc) {
|
|
m, err := gen()
|
|
if err != nil {
|
|
t.Fatal("failed to generate.")
|
|
}
|
|
|
|
if s2 := m.String(); err != nil || s2 != s {
|
|
t.Fatal("failed to convert: " + s + " != " + s2)
|
|
}
|
|
}
|
|
|
|
func TestFromIP4(t *testing.T) {
|
|
testConvert(t, "/ip4/10.20.30.40", func() (Multiaddr, error) {
|
|
return FromIP(net.ParseIP("10.20.30.40"))
|
|
})
|
|
}
|
|
|
|
func TestFromIP6(t *testing.T) {
|
|
testConvert(t, "/ip6/2001:4860:0:2001::68", func() (Multiaddr, error) {
|
|
return FromIP(net.ParseIP("2001:4860:0:2001::68"))
|
|
})
|
|
}
|
|
|
|
func TestFromTCP(t *testing.T) {
|
|
testConvert(t, "/ip4/10.20.30.40/tcp/1234", func() (Multiaddr, error) {
|
|
return FromNetAddr(&net.TCPAddr{
|
|
IP: net.ParseIP("10.20.30.40"),
|
|
Port: 1234,
|
|
})
|
|
})
|
|
}
|
|
|
|
func TestFromUDP(t *testing.T) {
|
|
testConvert(t, "/ip4/10.20.30.40/udp/1234", func() (Multiaddr, error) {
|
|
return FromNetAddr(&net.UDPAddr{
|
|
IP: net.ParseIP("10.20.30.40"),
|
|
Port: 1234,
|
|
})
|
|
})
|
|
}
|
|
|
|
func TestDialArgs(t *testing.T) {
|
|
m, err := NewMultiaddr("/ip4/127.0.0.1/udp/1234")
|
|
if err != nil {
|
|
t.Fatal("failed to construct", "/ip4/127.0.0.1/udp/1234")
|
|
}
|
|
|
|
nw, host, err := DialArgs(m)
|
|
if err != nil {
|
|
t.Fatal("failed to get dial args", "/ip4/127.0.0.1/udp/1234", err)
|
|
}
|
|
|
|
if nw != "udp" {
|
|
t.Error("failed to get udp network Dial Arg")
|
|
}
|
|
|
|
if host != "127.0.0.1:1234" {
|
|
t.Error("failed to get host:port Dial Arg")
|
|
}
|
|
}
|