mirror of
https://github.com/moby/moby.git
synced 2026-01-11 18:51:37 +00:00
85 lines
2.0 KiB
Go
85 lines
2.0 KiB
Go
package daemon
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/moby/moby/api/types/swarm"
|
|
"github.com/moby/moby/client"
|
|
"gotest.tools/v3/assert"
|
|
)
|
|
|
|
// NodeConstructor defines a swarm node constructor
|
|
type NodeConstructor func(*swarm.Node)
|
|
|
|
// GetNode returns a swarm node identified by the specified id
|
|
func (d *Daemon) GetNode(ctx context.Context, t testing.TB, id string, errCheck ...func(error) bool) *swarm.Node {
|
|
t.Helper()
|
|
cli := d.NewClientT(t)
|
|
defer cli.Close()
|
|
|
|
result, err := cli.NodeInspect(ctx, id, client.NodeInspectOptions{})
|
|
if err != nil {
|
|
for _, f := range errCheck {
|
|
if f(err) {
|
|
return nil
|
|
}
|
|
}
|
|
}
|
|
assert.NilError(t, err, "[%s] (*Daemon).GetNode: NodeInspect(%q) failed", d.id, id)
|
|
assert.Check(t, result.Node.ID == id)
|
|
return &result.Node
|
|
}
|
|
|
|
// RemoveNode removes the specified node
|
|
func (d *Daemon) RemoveNode(ctx context.Context, t testing.TB, id string, force bool) {
|
|
t.Helper()
|
|
cli := d.NewClientT(t)
|
|
defer cli.Close()
|
|
|
|
options := client.NodeRemoveOptions{
|
|
Force: force,
|
|
}
|
|
_, err := cli.NodeRemove(ctx, id, options)
|
|
assert.NilError(t, err)
|
|
}
|
|
|
|
// UpdateNode updates a swarm node with the specified node constructor
|
|
func (d *Daemon) UpdateNode(ctx context.Context, t testing.TB, id string, f ...NodeConstructor) {
|
|
t.Helper()
|
|
cli := d.NewClientT(t)
|
|
defer cli.Close()
|
|
|
|
for i := 0; ; i++ {
|
|
node := d.GetNode(ctx, t, id)
|
|
for _, fn := range f {
|
|
fn(node)
|
|
}
|
|
|
|
_, err := cli.NodeUpdate(ctx, node.ID, client.NodeUpdateOptions{
|
|
Version: node.Version,
|
|
Node: node.Spec,
|
|
})
|
|
if i < 10 && err != nil && strings.Contains(err.Error(), "update out of sequence") {
|
|
time.Sleep(100 * time.Millisecond)
|
|
continue
|
|
}
|
|
assert.NilError(t, err)
|
|
return
|
|
}
|
|
}
|
|
|
|
// ListNodes returns the list of the current swarm nodes
|
|
func (d *Daemon) ListNodes(ctx context.Context, t testing.TB) []swarm.Node {
|
|
t.Helper()
|
|
cli := d.NewClientT(t)
|
|
defer cli.Close()
|
|
|
|
result, err := cli.NodeList(ctx, client.NodeListOptions{})
|
|
assert.NilError(t, err)
|
|
|
|
return result.Items
|
|
}
|