mirror of https://github.com/docker/cli.git
29 lines
883 B
Go
29 lines
883 B
Go
|
package task
|
||
|
|
||
|
import (
|
||
|
"github.com/docker/docker/api/types"
|
||
|
"github.com/docker/docker/api/types/swarm"
|
||
|
"github.com/docker/docker/client"
|
||
|
"golang.org/x/net/context"
|
||
|
)
|
||
|
|
||
|
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(ctx 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(ctx 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
|
||
|
}
|