Skip to content

Commit

Permalink
fix unit-test
Browse files Browse the repository at this point in the history
Signed-off-by: Christian Kotzbauer <[email protected]>
  • Loading branch information
ckotzbauer committed Aug 14, 2023
1 parent c296da5 commit fb22621
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions cmd/kured/main_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,20 +4,21 @@ import (
"reflect"
"testing"

"github.com/kubereboot/kured/pkg/alerts"
log "github.com/sirupsen/logrus"
"github.com/spf13/cobra"
"github.com/kubereboot/kured/pkg/alerts"
assert "gotest.tools/v3/assert"

papi "github.com/prometheus/client_golang/api"
)

type BlockingChecker struct {
blocking bool
blocking bool
releaseNodeLock bool
}

func (fbc BlockingChecker) isBlocked() bool {
return fbc.blocking
func (fbc BlockingChecker) isBlocked() (bool, bool) {
return fbc.blocking, fbc.releaseNodeLock
}

var _ RebootBlocker = BlockingChecker{} // Verify that Type implements Interface.
Expand Down Expand Up @@ -155,7 +156,7 @@ func Test_rebootBlocked(t *testing.T) {
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
if got := rebootBlocked(tt.args.blockers...); got != tt.want {
if got, _ := rebootBlocked(tt.args.blockers...); got != tt.want {
t.Errorf("rebootBlocked() = %v, want %v", got, tt.want)
}
})
Expand Down

0 comments on commit fb22621

Please sign in to comment.