mirror of
https://git.vectorsigma.ru/public/k3s.git
synced 2026-07-29 05:19:18 +00:00
1ba19856de
Use mocked executor to ensure the correct args are being passed to components Signed-off-by: Brad Davidson <brad.davidson@rancher.com>
35 lines
1.3 KiB
Go
35 lines
1.3 KiB
Go
package mock
|
|
|
|
import (
|
|
"testing"
|
|
|
|
executor "github.com/k3s-io/k3s/pkg/daemons/executor"
|
|
"go.uber.org/mock/gomock"
|
|
)
|
|
|
|
// NewExecutorWithEmbeddedETCD creates a new mock executor, and sets it as the current executor.
|
|
// The executor exepects calls to ETCD(), and wraps the embedded executor method of the same name.
|
|
// The various ready channels are also mocked with immediate channel closure.
|
|
func NewExecutorWithEmbeddedETCD(t *testing.T) *Executor {
|
|
mockController := gomock.NewController(t)
|
|
mockExecutor := NewExecutor(mockController)
|
|
|
|
embed := &executor.Embedded{}
|
|
mockExecutor.EXPECT().Bootstrap(gomock.Any(), gomock.Any(), gomock.Any()).AnyTimes().DoAndReturn(embed.Bootstrap)
|
|
mockExecutor.EXPECT().CurrentETCDOptions().AnyTimes().DoAndReturn(embed.CurrentETCDOptions)
|
|
mockExecutor.EXPECT().ETCD(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).AnyTimes().DoAndReturn(embed.ETCD)
|
|
mockExecutor.EXPECT().ETCDReadyChan().AnyTimes().DoAndReturn(embed.ETCDReadyChan)
|
|
|
|
closedChannel := func() <-chan struct{} {
|
|
c := make(chan struct{})
|
|
close(c)
|
|
return c
|
|
}
|
|
mockExecutor.EXPECT().APIServerReadyChan().AnyTimes().DoAndReturn(closedChannel)
|
|
mockExecutor.EXPECT().CRIReadyChan().AnyTimes().DoAndReturn(closedChannel)
|
|
|
|
executor.Set(mockExecutor)
|
|
|
|
return mockExecutor
|
|
}
|