aboutsummaryrefslogtreecommitdiffstats
path: root/p2p/client_identity_test.go
blob: 61c34fbf1167169745d1be552b9c7cbb41d3ace7 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
package p2p

import (
    "bytes"
    "fmt"
    "runtime"
    "testing"
)

func TestClientIdentity(t *testing.T) {
    clientIdentity := NewSimpleClientIdentity("Ethereum(G)", "0.5.16", "test", []byte("privkey"), []byte("pubkey"))
    key := clientIdentity.Privkey()
    if !bytes.Equal(key, []byte("privkey")) {
        t.Errorf("Expected Privkey to be %x, got %x", key, []byte("privkey"))
    }
    key = clientIdentity.Pubkey()
    if !bytes.Equal(key, []byte("pubkey")) {
        t.Errorf("Expected Pubkey to be %x, got %x", key, []byte("pubkey"))
    }
    clientString := clientIdentity.String()
    expected := fmt.Sprintf("Ethereum(G)/v0.5.16/test/%s/%s", runtime.GOOS, runtime.Version())
    if clientString != expected {
        t.Errorf("Expected clientIdentity to be %v, got %v", expected, clientString)
    }
    customIdentifier := clientIdentity.GetCustomIdentifier()
    if customIdentifier != "test" {
        t.Errorf("Expected clientIdentity.GetCustomIdentifier() to be 'test', got %v", customIdentifier)
    }
    clientIdentity.SetCustomIdentifier("test2")
    customIdentifier = clientIdentity.GetCustomIdentifier()
    if customIdentifier != "test2" {
        t.Errorf("Expected clientIdentity.GetCustomIdentifier() to be 'test2', got %v", customIdentifier)
    }
    clientString = clientIdentity.String()
    expected = fmt.Sprintf("Ethereum(G)/v0.5.16/test2/%s/%s", runtime.GOOS, runtime.Version())
    if clientString != expected {
        t.Errorf("Expected clientIdentity to be %v, got %v", expected, clientString)
    }
}