@@ -7,44 +7,36 @@ import (
77 ma "github.com/multiformats/go-multiaddr"
88)
99
10- func TestRegisterSpec (t * testing.T ) {
10+ func TestRegisterFrom (t * testing.T ) {
1111 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 {
2317 t .Fatal ("myproto not properly registered" )
2418 }
25-
26- _ , ok = cm .addrParsers ["iptest" ]
27- if ! ok {
19+ if _ , ok := cm .addrParsers ["iptest" ]; ! ok {
2820 t .Fatal ("myproto not properly registered" )
2921 }
30-
31- _ , ok = cm .addrParsers ["blahtest" ]
32- if ! ok {
22+ if _ , ok := cm .addrParsers ["blahtest" ]; ! ok {
3323 t .Fatal ("myproto not properly registered" )
3424 }
25+ }
3526
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 {
3834 t .Fatal ("myproto not properly registered" )
3935 }
40-
41- _ , ok = cm .maddrParsers ["iptest" ]
42- if ok {
36+ if _ , ok := cm .maddrParsers ["iptest" ]; ! ok {
4337 t .Fatal ("myproto not properly registered" )
4438 }
45-
46- _ , ok = cm .maddrParsers ["blahtest" ]
47- if ok {
39+ if _ , ok := cm .maddrParsers ["blahtest" ]; ! ok {
4840 t .Fatal ("myproto not properly registered" )
4941 }
5042}
0 commit comments