6f8070deb2
Since go 1.7, "context" is a standard package. Since go 1.9,
x/net/context merely provides some types aliased to those in
the standard context package.
The changes were performed by the following script:
for f in $(git ls-files \*.go | grep -v ^vendor/); do
sed -i 's|golang.org/x/net/context|context|' $f
goimports -w $f
for i in 1 2; do
awk '/^$/ {e=1; next;}
/\t"context"$/ {e=0;}
{if (e) {print ""; e=0}; print;}' < $f > $f.new && \
mv $f.new $f
goimports -w $f
done
done
[v2: do awk/goimports fixup twice]
Signed-off-by: Kir Kolyshkin <kolyshkin@gmail.com>
70 lines
2.0 KiB
Go
70 lines
2.0 KiB
Go
package node
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/docker/docker/api/types/swarm"
|
|
"github.com/docker/docker/client"
|
|
)
|
|
|
|
type fakeClient struct {
|
|
client.Client
|
|
infoFunc func() (types.Info, error)
|
|
nodeInspectFunc func() (swarm.Node, []byte, error)
|
|
nodeListFunc func() ([]swarm.Node, error)
|
|
nodeRemoveFunc func() error
|
|
nodeUpdateFunc func(nodeID string, version swarm.Version, node swarm.NodeSpec) error
|
|
taskInspectFunc func(taskID string) (swarm.Task, []byte, error)
|
|
taskListFunc func(options types.TaskListOptions) ([]swarm.Task, error)
|
|
}
|
|
|
|
func (cli *fakeClient) NodeInspectWithRaw(ctx context.Context, ref string) (swarm.Node, []byte, error) {
|
|
if cli.nodeInspectFunc != nil {
|
|
return cli.nodeInspectFunc()
|
|
}
|
|
return swarm.Node{}, []byte{}, nil
|
|
}
|
|
|
|
func (cli *fakeClient) NodeList(ctx context.Context, options types.NodeListOptions) ([]swarm.Node, error) {
|
|
if cli.nodeListFunc != nil {
|
|
return cli.nodeListFunc()
|
|
}
|
|
return []swarm.Node{}, nil
|
|
}
|
|
|
|
func (cli *fakeClient) NodeRemove(ctx context.Context, nodeID string, options types.NodeRemoveOptions) error {
|
|
if cli.nodeRemoveFunc != nil {
|
|
return cli.nodeRemoveFunc()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (cli *fakeClient) NodeUpdate(ctx context.Context, nodeID string, version swarm.Version, node swarm.NodeSpec) error {
|
|
if cli.nodeUpdateFunc != nil {
|
|
return cli.nodeUpdateFunc(nodeID, version, node)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (cli *fakeClient) Info(ctx context.Context) (types.Info, error) {
|
|
if cli.infoFunc != nil {
|
|
return cli.infoFunc()
|
|
}
|
|
return types.Info{}, nil
|
|
}
|
|
|
|
func (cli *fakeClient) TaskInspectWithRaw(ctx context.Context, taskID string) (swarm.Task, []byte, error) {
|
|
if cli.taskInspectFunc != nil {
|
|
return cli.taskInspectFunc(taskID)
|
|
}
|
|
return swarm.Task{}, []byte{}, nil
|
|
}
|
|
|
|
func (cli *fakeClient) TaskList(ctx context.Context, options types.TaskListOptions) ([]swarm.Task, error) {
|
|
if cli.taskListFunc != nil {
|
|
return cli.taskListFunc(options)
|
|
}
|
|
return []swarm.Task{}, nil
|
|
}
|