mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
27 lines
767 B
Go
27 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
|
|
}
|