mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
56a68c15f8
This introduces `NoTasksForService` and `NoTasks` poller checks, that can be used to check if no tasks are left in general, or for a specific service. Some redundant checks were also removed from some tests. Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
47 lines
1.4 KiB
Go
47 lines
1.4 KiB
Go
package swarm
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/docker/docker/api/types/filters"
|
|
"github.com/docker/docker/client"
|
|
"gotest.tools/poll"
|
|
)
|
|
|
|
// NoTasksForService verifies that there are no more tasks for the given service
|
|
func NoTasksForService(ctx context.Context, client client.ServiceAPIClient, serviceID string) func(log poll.LogT) poll.Result {
|
|
return func(log poll.LogT) poll.Result {
|
|
tasks, err := client.TaskList(ctx, types.TaskListOptions{
|
|
Filters: filters.NewArgs(
|
|
filters.Arg("service", serviceID),
|
|
),
|
|
})
|
|
if err == nil {
|
|
if len(tasks) == 0 {
|
|
return poll.Success()
|
|
}
|
|
if len(tasks) > 0 {
|
|
return poll.Continue("task count for service %s at %d waiting for 0", serviceID, len(tasks))
|
|
}
|
|
return poll.Continue("waiting for tasks for service %s to be deleted", serviceID)
|
|
}
|
|
// TODO we should not use an error as indication that the tasks are gone. There may be other reasons for an error to occur.
|
|
return poll.Success()
|
|
}
|
|
}
|
|
|
|
// NoTasks verifies that all tasks are gone
|
|
func NoTasks(ctx context.Context, client client.ServiceAPIClient) func(log poll.LogT) poll.Result {
|
|
return func(log poll.LogT) poll.Result {
|
|
tasks, err := client.TaskList(ctx, types.TaskListOptions{})
|
|
switch {
|
|
case err != nil:
|
|
return poll.Error(err)
|
|
case len(tasks) == 0:
|
|
return poll.Success()
|
|
default:
|
|
return poll.Continue("waiting for all tasks to be removed: task count at %d", len(tasks))
|
|
}
|
|
}
|
|
}
|