timerticker/ticker_test.go at main · VarusHsu/timerticker · GitHub

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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
package timerticker

import (
	"context"
	"github.com/stretchr/testify/assert"
	"testing"
	"time"
)

func TestTimerTicker_SetLaunchTimeTwice(t1 *testing.T) {
	defer func() {
		if err := recover(); err != nil {
			assert.Equal(t1, "not allow set start time twice", err)
		}
	}()
	ticker := NewTimerTicker(time.Second)
	ticker.SetLaunchTime(context.Background(), time.Now().Add(time.Second))
	ticker.SetLaunchTime(context.Background(), time.Now().Add(time.Second))
}

func TestTimerTicker_SetLaunchTimeBeforeNow(t1 *testing.T) {
	defer func() {
		if err := recover(); err != nil {
			assert.Equal(t1, "set start time before now", err)
		}
	}()
	ticker := NewTimerTicker(time.Second)
	ticker.SetLaunchTime(context.Background(), time.Now())
}

func TestNewTimerTicker(t *testing.T) {
	defer func() {
		if err := recover(); err != nil {
			assert.Equal(t, "non-positive interval for NewTicker", err)
		}
	}()
	var duration = -time.Second
	_ = NewTimerTicker(duration)
}

func TestNewTimerTicker2(t *testing.T) {
	now := time.Now()
	stamp := now.UnixMilli()
	t.Log(now)
	closeChan := make(chan struct{})
	go func() {
		time.Sleep(30 * time.Second)
		closeChan <- struct{}{}
	}()
	ticker := NewTimerTicker(time.Second)
	ticker.SetLaunchTime(context.Background(), now.Add(5*time.Second))
	ch := ticker.GetChan()

	exceptStamp := stamp + 5000
	for {
		select {
		case tm := <-ch:
			t.Log(tm)
			// 60ms 误差
			assert.LessOrEqual(t, tm.UnixMilli(), exceptStamp+30)
			assert.GreaterOrEqual(t, tm.UnixMilli(), exceptStamp-30)
			exceptStamp += 1000
		case <-closeChan:
			ticker.Stop()
			return
		}
	}
}

func TestTimerTicker_Reset(t *testing.T) {
	resetChan := make(chan struct{})
	closeChan := make(chan struct{})
	go func() {
		time.Sleep(time.Second * 10)
		resetChan <- struct{}{}
	}()
	go func() {
		time.Sleep(time.Second * 28)
		closeChan <- struct{}{}
	}()
	now := time.Now()
	ticker := NewTimerTicker(2 * time.Second)
	ticker.SetLaunchTime(context.Background(), now.Add(time.Second))
	hitArray := []int64{
		now.UnixMilli() + 1000,
		now.UnixMilli() + 3000,
		now.UnixMilli() + 5000,
		now.UnixMilli() + 7000,
		now.UnixMilli() + 9000,
		now.UnixMilli() + 15000,
		now.UnixMilli() + 20000,
		now.UnixMilli() + 25000,
	}
	i := 0
	ch := ticker.GetChan()
	for {
		select {
		case tm := <-ch:
			t.Log(tm)
			assert.LessOrEqual(t, tm.UnixMilli(), hitArray[i]+30)
			assert.GreaterOrEqual(t, tm.UnixMilli(), hitArray[i]-30)
			i++
		case <-resetChan:
			ticker.Reset(time.Second * 5)
		case <-closeChan:
			return
		}
	}

}