-
Notifications
You must be signed in to change notification settings - Fork 54
/
dial_test.go
76 lines (63 loc) · 1.71 KB
/
dial_test.go
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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
package p2p
import (
"crypto/ecdsa"
"net"
"testing"
"time"
"xlibp2p/crypto"
"xlibp2p/discover"
)
var boots = []string{
"xfsnode://127.0.0.1:9092/?id=ff929a9b96a52935b3b65808c73e645b7bbf13dc53f6e0ce140079f106b67fa48d4690500db818c28618f70a04125d6707e38578904c7187fd755b8184f0375f",
"xfsnode://127.0.0.1:9092/?id=ff729a9b96a52935b3b65808c73e649b7bbf13dc53f6e0ce140079f106b67fa48d4690500db818c28618f70a04125d6707e38578904c7187fd755b8184f0375f",
"xfsnode://127.0.0.1:9092/?id=ff929a9b96a52935b3b65809c73e645b7bbf13dc53f6e0ce140079f106b67fa48d4690500db818c28618f70a04125d6707e38578904c7187fd755b8184f0375f",
}
type testTable struct {
key *ecdsa.PrivateKey
self *discover.Node
t *testing.T
}
func (t *testTable) Self() *discover.Node {
return t.self
}
func (t *testTable) Close() {
}
func (t *testTable) Bootstrap(ns []*discover.Node) {
}
func (t *testTable) Lookup(nid discover.NodeId) []*discover.Node {
return nil
}
func (t *testTable) ReadRandomNodes([]*discover.Node) int {
return 0
}
func newTestTable(t *testing.T, key *ecdsa.PrivateKey) *testTable {
if key == nil {
t.Fatal("key not set")
}
id := discover.PubKey2NodeId(key.PublicKey)
node := discover.NewNode(net.IP{127,0,0,1}.To4(),1024,1024, id)
return &testTable{
key: key,
t: t,
self: node,
}
}
func Test_dialtask_newDialState(t *testing.T) {
key := crypto.MustGenPrvKey()
bootNs := make([]*discover.Node, 0)
for _, nRaw := range boots {
n, err := discover.ParseNode(nRaw)
if err != nil {
t.Fatal(err)
}
bootNs = append(bootNs, n)
}
dynPeers := 10 / 2
ds := newDialState(bootNs,newTestTable(t,key),dynPeers)
ps := make(map[discover.NodeId]Peer)
for {
now := time.Now()
ts := ds.newTasks(1,ps,now)
t.Logf("ts len: %d", len(ts))
}
}