breaker/breaker_test.go at master · asnelzin/breaker · 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
package breaker

import (
	"fmt"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
)

func TestBreakerInitialStat(t *testing.T) {
	b := prep()
	assert.Equal(t, Closed, b.GetState())
}

func TestBreakerOpened(t *testing.T) {
	b := prep()
	_, err := b.Call(errorFn)
	assert.Equal(t, Closed, b.GetState())
	assert.EqualError(t, err, "error")
	_, err = b.Call(errorFn)
	assert.Equal(t, Open, b.GetState())
	assert.EqualError(t, err, "error")

	_, err = b.Call(errorFn)
	assert.Equal(t, ErrBreakerOpen, err)
}

func TestBreakerHalfOpened(t *testing.T) {
	b := prep()

	b.Call(errorFn)
	b.Call(errorFn)

	time.Sleep(3 * time.Second)
	assert.Equal(t, HalfOpen, b.GetState())

	_, err := b.Call(errorFn)
	assert.EqualError(t, err, "error")
	assert.Equal(t, Open, b.GetState())
}

func TestBreakerReset(t *testing.T) {
	b := prep()

	b.Call(errorFn)
	b.Call(errorFn)

	time.Sleep(3 * time.Second)
	assert.Equal(t, HalfOpen, b.GetState())

	b.Call(func() (interface{}, error) {
		return nil, nil
	})
	assert.Equal(t, Closed, b.GetState())
}

func TestBreakerTimeout(t *testing.T) {
	b := prep()
	start := time.Now()
	_, err := b.Call(func() (interface{}, error) {
		time.Sleep(3 * time.Second)
		return nil, nil
	})
	end := time.Now()
	assert.Equal(t, ErrTimeout, err)
	assert.Equal(t, 2*time.Second, end.Sub(start).Truncate(time.Second))
}

func prep() Breaker {
	return Breaker{
		Threshold:         2,
		InvocationTimeout: 2 * time.Second,
		ResetTimeout:      3 * time.Second,
	}
}

func errorFn() (interface{}, error) {
	return nil, fmt.Errorf("error")
}