go-queue/queue_test.go at main · czasg/go-queue · 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
111
112
113
114
package queue

import (
    "context"
    "errors"
    "reflect"
    "testing"
    "time"
)

func test_queue(name string, queue Queue, t *testing.T) {
    {
        if length := queue.Len(); length != 0 {
            t.Error(name, "空队列-获取长度为0", length)
        }
        if data, err := queue.Get(nil); data != nil || !errors.Is(err, ErrQueueEmpty) {
            t.Error(name, "空队列-Get数据返回ErrQueueEmpty", data, err)
        }
    }
    {
        if err := queue.Put(nil, []byte{}); err != nil {
            t.Error(name, "空队列-Put数据返回nil", err)
        }
        if data, err := queue.Get(nil); !reflect.DeepEqual(data, []byte{}) || err != nil {
            t.Error(name, "非空队列-Get数据返回nil", data, err)
        }
    }
    {
        for i := 0; i < 1024; i++ {
            if err := queue.Put(nil, []byte{}); err != nil {
                t.Error(name, "空队列-推送1024条数据返回nil", err)
            }
        }
        if length := queue.Len(); length != 1024 {
            t.Error(name, "满队列获取长度为1024", length)
        }
        if err := queue.Put(nil, []byte{}); !errors.Is(err, ErrQueueFull) {
            t.Error(name, "满队列推送数据返回ErrQueueFull", err)
        }
        for i := 0; i < 1024; i++ {
            if data, err := queue.Get(nil); !reflect.DeepEqual(data, []byte{}) || err != nil {
                t.Error(name, "满队列Get数据返回nil", data, err)
            }
        }
        if data, err := queue.Get(nil); data != nil || !errors.Is(err, ErrQueueEmpty) {
            t.Error(name, "1024空队列-Get数据返回ErrQueueEmpty", data, err)
        }
    }
    {
        go func() {
            time.Sleep(time.Millisecond)
            if err := queue.Put(context.Background(), []byte{}); err != nil {
                t.Error(name, "阻塞队列Put数据返回nil", err)
            }
        }()
        if data, err := queue.Get(context.Background()); !reflect.DeepEqual(data, []byte{}) || err != nil {
            t.Error(name, "阻塞队列Get数据返回nil", data, err)
        }
        if data, err := queue.Get(nil); data != nil || !errors.Is(err, ErrQueueEmpty) {
            t.Error(name, "阻塞队列-Get数据返回ErrQueueEmpty", data, err)
        }
    }
    {
        ctx, cancel := context.WithCancel(context.Background())
        go func() {
            if data, err := queue.Get(ctx); data != nil || !errors.Is(err, ctx.Err()) {
                t.Error(name, "ctx失效-阻塞队列Get数据返回cancel", data, err)
            }
        }()
        time.Sleep(time.Millisecond)
        cancel()
    }
    {
        for i := 0; i < 1024; i++ {
            if err := queue.Put(nil, []byte{}); err != nil {
                t.Error(name, "阻塞队列-推送1024条数据返回nil", err)
            }
        }
        ctx, cancel := context.WithCancel(context.Background())
        go func() {
            if err := queue.Put(ctx, []byte{}); !errors.Is(err, ctx.Err()) {
                t.Error(name, "ctx失效-阻塞队列-推送1024条数据返回cancel", err)
            }
        }()
        time.Sleep(time.Millisecond)
        cancel()
        time.Sleep(time.Millisecond * 2)
        for i := 0; i < 1024; i++ {
            if data, err := queue.Get(nil); !reflect.DeepEqual(data, []byte{}) || err != nil {
                t.Error(name, "满队列Get数据返回nil", data, err)
            }
        }
    }
    // 队列关闭测试
    {
        if err := queue.Close(); err != nil {
            t.Error(name, "队列关闭返回nil", err)
        }
    }
    {
        if data, err := queue.Get(nil); data != nil || !errors.Is(err, ErrQueueClosed) {
            t.Error(name, "关闭队列-Get数据返回ErrQueueClosed", data, err)
        }
        if data, err := queue.Get(context.Background()); data != nil || !errors.Is(err, ErrQueueClosed) {
            t.Error(name, "关闭队列-Get数据返回ErrQueueClosed", data, err)
        }
        if err := queue.Put(nil, []byte{}); !errors.Is(err, ErrQueueClosed) {
            t.Error(name, "关闭队列-推送数据返回ErrQueueClosed", err)
        }
        if err := queue.Put(context.Background(), []byte{}); !errors.Is(err, ErrQueueClosed) {
            t.Error(name, "关闭队列-推送数据返回ErrQueueClosed", err)
        }
    }
}