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
115
116
|
// Copyright (c) 2012-2016 Eli Janssen
// Use of this source code is governed by an MIT-style
// license that can be found in the LICENSE file.
package statsd
import (
"bytes"
"testing"
"time"
)
type mockSender struct {
closeCallCount int
}
func (m *mockSender) Send(data []byte) (int, error) {
return 0, nil
}
func (m *mockSender) Close() error {
m.closeCallCount++
return nil
}
func TestClose(t *testing.T) {
mockSender := &mockSender{}
sender := &BufferedSender{
flushBytes: 512,
flushInterval: 1 * time.Second,
sender: mockSender,
buffer: bytes.NewBuffer(make([]byte, 0, 512)),
shutdown: make(chan chan error),
}
sender.Close()
if mockSender.closeCallCount != 0 {
t.Fatalf("expected close to have been called zero times, but got %d", mockSender.closeCallCount)
}
sender.Start()
if !sender.running {
t.Fatal("sender failed to start")
}
sender.Close()
if mockSender.closeCallCount != 1 {
t.Fatalf("expected close to have been called once, but got %d", mockSender.closeCallCount)
}
}
func TestCloseConcurrent(t *testing.T) {
mockSender := &mockSender{}
sender := &BufferedSender{
flushBytes: 512,
flushInterval: 1 * time.Second,
sender: mockSender,
buffer: bytes.NewBuffer(make([]byte, 0, 512)),
shutdown: make(chan chan error),
}
sender.Start()
const N = 10
c := make(chan struct{}, N)
for i := 0; i < N; i++ {
go func() {
sender.Close()
c <- struct{}{}
}()
}
for i := 0; i < N; i++ {
<-c
}
if mockSender.closeCallCount != 1 {
t.Errorf("expected close to have been called once, but got %d", mockSender.closeCallCount)
}
}
func TestCloseDuringSendConcurrent(t *testing.T) {
mockSender := &mockSender{}
sender := &BufferedSender{
flushBytes: 512,
flushInterval: 1 * time.Second,
sender: mockSender,
buffer: bytes.NewBuffer(make([]byte, 0, 512)),
shutdown: make(chan chan error),
}
sender.Start()
const N = 10
c := make(chan struct{}, N)
for i := 0; i < N; i++ {
go func() {
for {
_, err := sender.Send([]byte("stat:1|c"))
if err != nil {
c <- struct{}{}
return
}
}
}()
}
// senders should error out now
// we should not receive any panics
sender.Close()
for i := 0; i < N; i++ {
<-c
}
if mockSender.closeCallCount != 1 {
t.Errorf("expected close to have been called once, but got %d", mockSender.closeCallCount)
}
}
|
