mirror of https://github.com/docker/cli.git
30 lines
863 B
Go
30 lines
863 B
Go
package task
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/docker/docker/api/types/swarm"
|
|
"github.com/docker/docker/client"
|
|
)
|
|
|
|
type fakeClient struct {
|
|
client.APIClient
|
|
nodeInspectWithRaw func(ref string) (swarm.Node, []byte, error)
|
|
serviceInspectWithRaw func(ref string, options types.ServiceInspectOptions) (swarm.Service, []byte, error)
|
|
}
|
|
|
|
func (cli *fakeClient) NodeInspectWithRaw(_ context.Context, ref string) (swarm.Node, []byte, error) {
|
|
if cli.nodeInspectWithRaw != nil {
|
|
return cli.nodeInspectWithRaw(ref)
|
|
}
|
|
return swarm.Node{}, nil, nil
|
|
}
|
|
|
|
func (cli *fakeClient) ServiceInspectWithRaw(_ context.Context, ref string, options types.ServiceInspectOptions) (swarm.Service, []byte, error) {
|
|
if cli.serviceInspectWithRaw != nil {
|
|
return cli.serviceInspectWithRaw(ref, options)
|
|
}
|
|
return swarm.Service{}, nil, nil
|
|
}
|