forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-08-17 17:33:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

p2p/enr: fixup tests

Browse source
This commit is contained in:
Felix Lange 2017-12-08 21:35:21 +01:00
parent 73923634b7
commit d97b48d0d2
1 changed files with 15 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
p2p/enr/enr_test.go
View file

@ -41,6 +41,12 @@ var (
var rnd = rand.New(rand.NewSource(time.Now().UnixNano())) var rnd = rand.New(rand.NewSource(time.Now().UnixNano()))
func randomString(strlen int) string {
b := make([]byte, strlen)
rnd.Read(b)
return string(b)
}
// TestGetSetID tests encoding/decoding and setting/getting of the ID key. // TestGetSetID tests encoding/decoding and setting/getting of the ID key.
func TestGetSetID(t *testing.T) { func TestGetSetID(t *testing.T) {
id := ID("someid") id := ID("someid")
@ -112,10 +118,10 @@ func TestLoadErrors(t *testing.T) {
// Check error for invalid keys. // Check error for invalid keys.
var list []uint var list []uint
err = r.Load(WithKey(IP4{}.ENRKey(), &list)) err = r.Load(WithKey(ip4.ENRKey(), &list))
kerr, ok := err.(*KeyError) kerr, ok := err.(*KeyError)
if !ok { if !ok {
t.Fatal("expected KeyError, got %T", err) t.Fatalf("expected KeyError, got %T", err)
} }
assert.Equal(t, kerr.Key, ip4.ENRKey()) assert.Equal(t, kerr.Key, ip4.ENRKey())
assert.Error(t, kerr.Err) assert.Error(t, kerr.Err)
@ -184,7 +190,7 @@ func TestDirty(t *testing.T) {
t.Error("Signed returned true for modified record") t.Error("Signed returned true for modified record")
} }
if _, err := rlp.EncodeToBytes(r); err != errEncodeUnsigned { if _, err := rlp.EncodeToBytes(r); err != errEncodeUnsigned {
t.Errorf("expected errEncodeUnsigned, got %#v") t.Errorf("expected errEncodeUnsigned, got %#v", err)
} }
} }
@ -192,10 +198,10 @@ func TestDirty(t *testing.T) {
func TestGetSetOverwrite(t *testing.T) { func TestGetSetOverwrite(t *testing.T) {
var r Record var r Record
ip := IP4(net.IP{192, 168, 0, 3}) ip := IP4{192, 168, 0, 3}
r.Set(ip) r.Set(ip)
ip2 := IP4(net.IP{192, 168, 0, 4}) ip2 := IP4{192, 168, 0, 4}
r.Set(ip2) r.Set(ip2)
var ip3 IP4 var ip3 IP4
@ -207,21 +213,18 @@ func TestGetSetOverwrite(t *testing.T) {
func TestSignEncodeAndDecode(t *testing.T) { func TestSignEncodeAndDecode(t *testing.T) {
var r Record var r Record
r.Set(DiscPort(30303)) r.Set(DiscPort(30303))
r.Set(IP4(net.ParseIP("127.0.0.1"))) r.Set(IP4{127, 0, 0, 1})
require.NoError(t, r.Sign(privkey)) require.NoError(t, r.Sign(privkey))
blob, err := rlp.EncodeToBytes(r) blob, err := rlp.EncodeToBytes(r)
if err != nil { require.NoError(t, err)
t.Fatal(err)
}
var r2 Record var r2 Record
require.NoError(t, rlp.DecodeBytes(blob, &r2)) require.NoError(t, rlp.DecodeBytes(blob, &r2))
assert.Equal(t, r, r2) assert.Equal(t, r, r2)
blob2, err := rlp.EncodeToBytes(r2) blob2, err := rlp.EncodeToBytes(r2)
if err != nil { require.NoError(t, err)
t.Fatal(err)
}
assert.Equal(t, blob, blob2) assert.Equal(t, blob, blob2)
} }
@ -307,9 +310,3 @@ func TestSignEncodeAndDecodeRandom(t *testing.T) {
require.Equal(t, v, got, desc) require.Equal(t, v, got, desc)
} }
} }
func randomString(strlen int) string {
b := make([]byte, strlen)
rnd.Read(b)
return string(b)
}