diff --git a/insdash/insdash_test.go b/insdash/insdash_test.go index 9e62f93..1e02812 100644 --- a/insdash/insdash_test.go +++ b/insdash/insdash_test.go @@ -23,7 +23,7 @@ func Test_createBatches(t *testing.T) { }) t.Run("should_return_batches_by_record_limit", func(t *testing.T) { - batches, err := CreateBatches([]string{"test", "test"}, 10, kb64) + batches, err := CreateBatches([]string{"test", "test"}, 1, kb64) assert.Nil(t, err, "err should be nil") assert.Equal(t, 2, len(batches), "batches length should be equal to 2") diff --git a/insrequester/requester_mock.go b/insrequester/requester_mock.go index f2a8680..8b4d9c2 100644 --- a/insrequester/requester_mock.go +++ b/insrequester/requester_mock.go @@ -7,6 +7,7 @@ package insrequester import ( http "net/http" reflect "reflect" + time "time" gomock "github.com/golang/mock/gomock" ) @@ -151,15 +152,15 @@ func (mr *MockRequesterMockRecorder) WithRetry(config interface{}) *gomock.Call } // WithTimeout mocks base method. -func (m *MockRequester) WithTimeout(timeoutSeconds int) *Request { +func (m *MockRequester) WithTimeout(timeout time.Duration) *Request { m.ctrl.T.Helper() - ret := m.ctrl.Call(m, "WithTimeout", timeoutSeconds) + ret := m.ctrl.Call(m, "WithTimeout", timeout) ret0, _ := ret[0].(*Request) return ret0 } // WithTimeout indicates an expected call of WithTimeout. -func (mr *MockRequesterMockRecorder) WithTimeout(timeoutSeconds interface{}) *gomock.Call { +func (mr *MockRequesterMockRecorder) WithTimeout(timeout interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() - return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WithTimeout", reflect.TypeOf((*MockRequester)(nil).WithTimeout), timeoutSeconds) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WithTimeout", reflect.TypeOf((*MockRequester)(nil).WithTimeout), timeout) } diff --git a/insrequester/requester_test.go b/insrequester/requester_test.go index f41cb0d..d9ebbbf 100644 --- a/insrequester/requester_test.go +++ b/insrequester/requester_test.go @@ -1,7 +1,6 @@ package insrequester import ( - "github.com/slok/goresilience/errors" "github.com/stretchr/testify/assert" "net/http" "net/http/httptest" @@ -66,7 +65,7 @@ func TestRequest_Get(t *testing.T) { _, _ = r.Get(req) } _, err = r.Get(req) - assert.ErrorIs(t, err, errors.ErrCircuitOpen) + assert.ErrorIs(t, err, ErrCircuitBreakerOpen) }) t.Run("it_should_apply_headers_properly", func(t *testing.T) {