80 lines
2.9 KiB
Go
80 lines
2.9 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/docker/cli/internal/test/builders"
|
|
"github.com/moby/moby/api/types/swarm"
|
|
"github.com/moby/moby/api/types/system"
|
|
"github.com/moby/moby/client"
|
|
)
|
|
|
|
type fakeClient struct {
|
|
client.Client
|
|
serviceInspectFunc func(ctx context.Context, serviceID string, options client.ServiceInspectOptions) (client.ServiceInspectResult, error)
|
|
serviceUpdateFunc func(ctx context.Context, serviceID string, options client.ServiceUpdateOptions) (client.ServiceUpdateResult, error)
|
|
serviceListFunc func(context.Context, client.ServiceListOptions) (client.ServiceListResult, error)
|
|
taskListFunc func(context.Context, client.TaskListOptions) (client.TaskListResult, error)
|
|
infoFunc func(ctx context.Context) (system.Info, error)
|
|
networkInspectFunc func(ctx context.Context, networkID string, options client.NetworkInspectOptions) (client.NetworkInspectResult, error)
|
|
nodeListFunc func(ctx context.Context, options client.NodeListOptions) (client.NodeListResult, error)
|
|
}
|
|
|
|
func (f *fakeClient) NodeList(ctx context.Context, options client.NodeListOptions) (client.NodeListResult, error) {
|
|
if f.nodeListFunc != nil {
|
|
return f.nodeListFunc(ctx, options)
|
|
}
|
|
return client.NodeListResult{}, nil
|
|
}
|
|
|
|
func (f *fakeClient) TaskList(ctx context.Context, options client.TaskListOptions) (client.TaskListResult, error) {
|
|
if f.taskListFunc != nil {
|
|
return f.taskListFunc(ctx, options)
|
|
}
|
|
return client.TaskListResult{}, nil
|
|
}
|
|
|
|
func (f *fakeClient) ServiceInspect(ctx context.Context, serviceID string, options client.ServiceInspectOptions) (client.ServiceInspectResult, error) {
|
|
if f.serviceInspectFunc != nil {
|
|
return f.serviceInspectFunc(ctx, serviceID, options)
|
|
}
|
|
|
|
return client.ServiceInspectResult{
|
|
Service: *builders.Service(builders.ServiceID(serviceID)),
|
|
}, nil
|
|
}
|
|
|
|
func (f *fakeClient) ServiceList(ctx context.Context, options client.ServiceListOptions) (client.ServiceListResult, error) {
|
|
if f.serviceListFunc != nil {
|
|
return f.serviceListFunc(ctx, options)
|
|
}
|
|
|
|
return client.ServiceListResult{}, nil
|
|
}
|
|
|
|
func (f *fakeClient) ServiceUpdate(ctx context.Context, serviceID string, options client.ServiceUpdateOptions) (client.ServiceUpdateResult, error) {
|
|
if f.serviceUpdateFunc != nil {
|
|
return f.serviceUpdateFunc(ctx, serviceID, options)
|
|
}
|
|
|
|
return client.ServiceUpdateResult{}, nil
|
|
}
|
|
|
|
func (f *fakeClient) Info(ctx context.Context) (system.Info, error) {
|
|
if f.infoFunc == nil {
|
|
return system.Info{}, nil
|
|
}
|
|
return f.infoFunc(ctx)
|
|
}
|
|
|
|
func (f *fakeClient) NetworkInspect(ctx context.Context, networkID string, options client.NetworkInspectOptions) (client.NetworkInspectResult, error) {
|
|
if f.networkInspectFunc != nil {
|
|
return f.networkInspectFunc(ctx, networkID, options)
|
|
}
|
|
return client.NetworkInspectResult{}, nil
|
|
}
|
|
|
|
func newService(id string, name string) swarm.Service {
|
|
return *builders.Service(builders.ServiceID(id), builders.ServiceName(name))
|
|
}
|