aboutsummaryrefslogtreecommitdiffstats
path: root/eth/filters/filter_system_test.go
blob: 7ddeb02bc92399dfb6e68e15b70501bc97239efd (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
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
77
78
79
80
81
82
83
84
85
86
87
package filters

import (
    "testing"
    "time"

    "github.com/ethereum/go-ethereum/core"
    "github.com/ethereum/go-ethereum/core/types"
    "github.com/ethereum/go-ethereum/core/vm"
    "github.com/ethereum/go-ethereum/event"
)

func TestCallbacks(t *testing.T) {
    var (
        mux            event.TypeMux
        fs             = NewFilterSystem(&mux)
        blockDone      = make(chan struct{})
        txDone         = make(chan struct{})
        logDone        = make(chan struct{})
        removedLogDone = make(chan struct{})
    )

    blockFilter := &Filter{
        BlockCallback: func(*types.Block, vm.Logs) {
            close(blockDone)
        },
    }
    txFilter := &Filter{
        TransactionCallback: func(*types.Transaction) {
            close(txDone)
        },
    }
    logFilter := &Filter{
        LogCallback: func(l *vm.Log, oob bool) {
            if !oob {
                close(logDone)
            }
        },
    }

    removedLogFilter := &Filter{
        LogCallback: func(l *vm.Log, oob bool) {
            if oob {
                close(removedLogDone)
            }
        },
    }

    fs.Add(blockFilter)
    fs.Add(txFilter)
    fs.Add(logFilter)
    fs.Add(removedLogFilter)

    mux.Post(core.ChainEvent{})
    mux.Post(core.TxPreEvent{})
    mux.Post(core.RemovedLogEvent{vm.Logs{&vm.Log{}}})
    mux.Post(vm.Logs{&vm.Log{}})

    const dura = 5 * time.Second
    failTimer := time.NewTimer(dura)
    select {
    case <-blockDone:
    case <-failTimer.C:
        t.Error("block filter failed to trigger (timeout)")
    }

    failTimer.Reset(dura)
    select {
    case <-txDone:
    case <-failTimer.C:
        t.Error("transaction filter failed to trigger (timeout)")
    }

    failTimer.Reset(dura)
    select {
    case <-logDone:
    case <-failTimer.C:
        t.Error("log filter failed to trigger (timeout)")
    }

    failTimer.Reset(dura)
    select {
    case <-removedLogDone:
    case <-failTimer.C:
        t.Error("removed log filter failed to trigger (timeout)")
    }
}