forked from vgarvardt/gue
/
client_option_test.go
57 lines (42 loc) · 1.58 KB
/
client_option_test.go
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
package gue
import (
"reflect"
"testing"
"time"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/mock"
"github.com/vgarvardt/gue/v3/adapter"
"github.com/vgarvardt/gue/v3/adapter/exponential"
)
func TestWithClientID(t *testing.T) {
clientWithDefaultID := NewClient(nil)
assert.NotEmpty(t, clientWithDefaultID.id)
customID := "some-meaningful-id"
clientWithCustomID := NewClient(nil, WithClientID(customID))
assert.Equal(t, customID, clientWithCustomID.id)
}
func TestWithClientLogger(t *testing.T) {
clientWithDefaultLogger := NewClient(nil)
assert.IsType(t, adapter.NoOpLogger{}, clientWithDefaultLogger.logger)
logMessage := "hello"
l := new(mockLogger)
l.On("Info", logMessage, mock.Anything)
// worker sets id as default logger field
l.On("With", mock.Anything).Return(l)
clientWithCustomLogger := NewClient(nil, WithClientLogger(l))
clientWithCustomLogger.logger.Info(logMessage)
l.AssertExpectations(t)
}
func TestWithClientBackoff(t *testing.T) {
customBackoff := func(retries int) time.Duration {
return time.Duration(retries) * time.Second
}
defaultPtr := reflect.ValueOf(exponential.Default).Pointer()
customPtr := reflect.ValueOf(customBackoff).Pointer()
clientWithDefaultBackoff := NewClient(nil)
clientWithDefaultBackoffPtr := reflect.ValueOf(clientWithDefaultBackoff.backoff).Pointer()
assert.Equal(t, defaultPtr, clientWithDefaultBackoffPtr)
clientWithCustomBackoff := NewClient(nil, WithClientBackoff(customBackoff))
assert.Equal(t, customBackoff(123), clientWithCustomBackoff.backoff(123))
assert.NotEqual(t, defaultPtr, customPtr)
}