@@ -7,44 +7,36 @@ import (
7
7
ma "github.com/multiformats/go-multiaddr"
8
8
)
9
9
10
- func TestRegisterSpec (t * testing.T ) {
10
+ func TestRegisterFrom (t * testing.T ) {
11
11
cm := NewCodecMap ()
12
- myproto := & NetCodec {
13
- ProtocolName : "test" ,
14
- NetAddrNetworks : []string {"test" , "iptest" , "blahtest" },
15
- ConvertMultiaddr : func (a ma.Multiaddr ) (net.Addr , error ) { return nil , nil },
16
- ParseNetAddr : func (a net.Addr ) (ma.Multiaddr , error ) { return nil , nil },
17
- }
18
-
19
- cm .RegisterNetCodec (myproto )
20
-
21
- _ , ok := cm .addrParsers ["test" ]
22
- if ! ok {
12
+ cm .RegisterFromNetAddr (
13
+ func (a net.Addr ) (ma.Multiaddr , error ) { return nil , nil },
14
+ "test" , "iptest" , "blahtest" ,
15
+ )
16
+ if _ , ok := cm .addrParsers ["test" ]; ! ok {
23
17
t .Fatal ("myproto not properly registered" )
24
18
}
25
-
26
- _ , ok = cm .addrParsers ["iptest" ]
27
- if ! ok {
19
+ if _ , ok := cm .addrParsers ["iptest" ]; ! ok {
28
20
t .Fatal ("myproto not properly registered" )
29
21
}
30
-
31
- _ , ok = cm .addrParsers ["blahtest" ]
32
- if ! ok {
22
+ if _ , ok := cm .addrParsers ["blahtest" ]; ! ok {
33
23
t .Fatal ("myproto not properly registered" )
34
24
}
25
+ }
35
26
36
- _ , ok = cm .maddrParsers ["test" ]
37
- if ! ok {
27
+ func TestRegisterTo (t * testing.T ) {
28
+ cm := NewCodecMap ()
29
+ cm .RegisterToNetAddr (
30
+ func (a ma.Multiaddr ) (net.Addr , error ) { return nil , nil },
31
+ "test" , "iptest" , "blahtest" ,
32
+ )
33
+ if _ , ok := cm .maddrParsers ["test" ]; ! ok {
38
34
t .Fatal ("myproto not properly registered" )
39
35
}
40
-
41
- _ , ok = cm .maddrParsers ["iptest" ]
42
- if ok {
36
+ if _ , ok := cm .maddrParsers ["iptest" ]; ! ok {
43
37
t .Fatal ("myproto not properly registered" )
44
38
}
45
-
46
- _ , ok = cm .maddrParsers ["blahtest" ]
47
- if ok {
39
+ if _ , ok := cm .maddrParsers ["blahtest" ]; ! ok {
48
40
t .Fatal ("myproto not properly registered" )
49
41
}
50
42
}
0 commit comments