mirror of https://github.com/docker/cli.git
28 lines
767 B
Go
28 lines
767 B
Go
|
package idresolver
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/docker/docker/api/types/swarm"
|
||
|
|
"github.com/docker/docker/client"
|
||
|
|
"golang.org/x/net/context"
|
||
|
|
)
|
||
|
|
|
||
|
|
type fakeClient struct {
|
||
|
|
client.Client
|
||
|
|
nodeInspectFunc func(string) (swarm.Node, []byte, error)
|
||
|
|
serviceInspectFunc func(string) (swarm.Service, []byte, error)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (cli *fakeClient) NodeInspectWithRaw(ctx context.Context, nodeID string) (swarm.Node, []byte, error) {
|
||
|
|
if cli.nodeInspectFunc != nil {
|
||
|
|
return cli.nodeInspectFunc(nodeID)
|
||
|
|
}
|
||
|
|
return swarm.Node{}, []byte{}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (cli *fakeClient) ServiceInspectWithRaw(ctx context.Context, serviceID string) (swarm.Service, []byte, error) {
|
||
|
|
if cli.serviceInspectFunc != nil {
|
||
|
|
return cli.serviceInspectFunc(serviceID)
|
||
|
|
}
|
||
|
|
return swarm.Service{}, []byte{}, nil
|
||
|
|
}
|