forked from tokenized/bitcoin_reader
-
Notifications
You must be signed in to change notification settings - Fork 0
/
node_manager_test.go
54 lines (45 loc) · 1.25 KB
/
node_manager_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
package bitcoin_reader
import (
"testing"
"time"
"github.com/tokenized/bitcoin_reader/internal/platform/tests"
"github.com/tokenized/pkg/bitcoin"
"github.com/tokenized/pkg/storage"
"github.com/tokenized/threads"
)
func Test_NodeManager(t *testing.T) {
if !testing.Verbose() {
t.Skip() // Don't want to redownload the block all the time
}
ctx := tests.Context()
store := storage.NewMockStorage()
peers := NewPeerRepository(store, "")
headers := NewMockHeaderRepository()
config := &Config{
Network: bitcoin.MainNet,
DesiredNodeCount: 20,
}
peers.LoadSeeds(ctx, bitcoin.MainNet)
manager := NewNodeManager("/Tokenized/Spynode:Test/", config, headers, peers)
runThread := threads.NewInterruptableThread("Run", manager.Run)
runComplete := runThread.GetCompleteChannel()
runThread.Start(ctx)
select {
case <-runComplete:
if err := runThread.Error(); err != nil {
t.Errorf("Failed to run : %s", err)
}
t.Errorf("Completed without interrupt")
case <-time.After(20 * time.Second):
t.Logf("Shutting down")
runThread.Stop(ctx)
select {
case <-runComplete:
if err := runThread.Error(); err != nil {
t.Errorf("Failed to run : %s", err)
}
case <-time.After(time.Second):
t.Fatalf("Failed to shut down")
}
}
}