1
0
Fork 0
mirror of https://github.com/kyverno/kyverno.git synced 2024-12-14 11:57:48 +00:00
kyverno/pkg/breaker/breaker_test.go

78 lines
1.3 KiB
Go
Raw Normal View History

package breaker
import (
"context"
"errors"
"testing"
"github.com/stretchr/testify/assert"
)
func Test_breaker_Do(t *testing.T) {
type args struct {
inner func(context.Context) error
}
tests := []struct {
name string
subject *breaker
args args
wantErr bool
}{{
name: "empty",
subject: NewBreaker("", nil),
wantErr: false,
}, {
name: "no error",
subject: NewBreaker("", nil),
args: args{
inner: func(context.Context) error {
return nil
},
},
wantErr: false,
}, {
name: "with error",
subject: NewBreaker("", nil),
args: args{
inner: func(context.Context) error {
return errors.New("foo")
},
},
wantErr: true,
}, {
name: "with break",
subject: NewBreaker("", func(context.Context) bool {
return true
}),
args: args{
inner: func(context.Context) error {
return errors.New("foo")
},
},
wantErr: false,
}, {
name: "with metrics",
subject: &breaker{
open: func(context.Context) bool {
return true
},
},
args: args{
inner: func(context.Context) error {
return errors.New("foo")
},
},
wantErr: false,
}}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
err := tt.subject.Do(context.TODO(), tt.args.inner)
if tt.wantErr {
assert.Error(t, err)
} else {
assert.NoError(t, err)
}
})
}
}