a4aba23b85
During the refactoring for 18.09 the activate/update flows no longer restart the engine explicitly but let the user do that when they're ready, so the health check logic is no longer required. Signed-off-by: Daniel Hiltgen <daniel.hiltgen@docker.com> (cherry picked from commit f2b2061cc38995a4d16659787d80563d8ec669d0) Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
102 lines
3.1 KiB
Go
102 lines
3.1 KiB
Go
package engine
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/containerd/containerd"
|
|
registryclient "github.com/docker/cli/cli/registry/client"
|
|
clitypes "github.com/docker/cli/types"
|
|
"github.com/docker/docker/api/types"
|
|
)
|
|
|
|
type (
|
|
fakeContainerizedEngineClient struct {
|
|
closeFunc func() error
|
|
activateEngineFunc func(ctx context.Context,
|
|
opts clitypes.EngineInitOptions,
|
|
out clitypes.OutStream,
|
|
authConfig *types.AuthConfig) error
|
|
initEngineFunc func(ctx context.Context,
|
|
opts clitypes.EngineInitOptions,
|
|
out clitypes.OutStream,
|
|
authConfig *types.AuthConfig,
|
|
healthfn func(context.Context) error) error
|
|
doUpdateFunc func(ctx context.Context,
|
|
opts clitypes.EngineInitOptions,
|
|
out clitypes.OutStream,
|
|
authConfig *types.AuthConfig) error
|
|
getEngineVersionsFunc func(ctx context.Context,
|
|
registryClient registryclient.RegistryClient,
|
|
currentVersion,
|
|
imageName string) (clitypes.AvailableVersions, error)
|
|
|
|
getEngineFunc func(ctx context.Context) (containerd.Container, error)
|
|
removeEngineFunc func(ctx context.Context) error
|
|
getCurrentEngineVersionFunc func(ctx context.Context) (clitypes.EngineInitOptions, error)
|
|
}
|
|
)
|
|
|
|
func (w *fakeContainerizedEngineClient) Close() error {
|
|
if w.closeFunc != nil {
|
|
return w.closeFunc()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (w *fakeContainerizedEngineClient) ActivateEngine(ctx context.Context,
|
|
opts clitypes.EngineInitOptions,
|
|
out clitypes.OutStream,
|
|
authConfig *types.AuthConfig) error {
|
|
if w.activateEngineFunc != nil {
|
|
return w.activateEngineFunc(ctx, opts, out, authConfig)
|
|
}
|
|
return nil
|
|
}
|
|
func (w *fakeContainerizedEngineClient) InitEngine(ctx context.Context,
|
|
opts clitypes.EngineInitOptions,
|
|
out clitypes.OutStream,
|
|
authConfig *types.AuthConfig,
|
|
healthfn func(context.Context) error) error {
|
|
if w.initEngineFunc != nil {
|
|
return w.initEngineFunc(ctx, opts, out, authConfig, healthfn)
|
|
}
|
|
return nil
|
|
}
|
|
func (w *fakeContainerizedEngineClient) DoUpdate(ctx context.Context,
|
|
opts clitypes.EngineInitOptions,
|
|
out clitypes.OutStream,
|
|
authConfig *types.AuthConfig) error {
|
|
if w.doUpdateFunc != nil {
|
|
return w.doUpdateFunc(ctx, opts, out, authConfig)
|
|
}
|
|
return nil
|
|
}
|
|
func (w *fakeContainerizedEngineClient) GetEngineVersions(ctx context.Context,
|
|
registryClient registryclient.RegistryClient,
|
|
currentVersion, imageName string) (clitypes.AvailableVersions, error) {
|
|
|
|
if w.getEngineVersionsFunc != nil {
|
|
return w.getEngineVersionsFunc(ctx, registryClient, currentVersion, imageName)
|
|
}
|
|
return clitypes.AvailableVersions{}, nil
|
|
}
|
|
|
|
func (w *fakeContainerizedEngineClient) GetEngine(ctx context.Context) (containerd.Container, error) {
|
|
if w.getEngineFunc != nil {
|
|
return w.getEngineFunc(ctx)
|
|
}
|
|
return nil, nil
|
|
}
|
|
func (w *fakeContainerizedEngineClient) RemoveEngine(ctx context.Context) error {
|
|
if w.removeEngineFunc != nil {
|
|
return w.removeEngineFunc(ctx)
|
|
}
|
|
return nil
|
|
}
|
|
func (w *fakeContainerizedEngineClient) GetCurrentEngineVersion(ctx context.Context) (clitypes.EngineInitOptions, error) {
|
|
if w.getCurrentEngineVersionFunc != nil {
|
|
return w.getCurrentEngineVersionFunc(ctx)
|
|
}
|
|
return clitypes.EngineInitOptions{}, nil
|
|
}
|