123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228 |
- package service
- import (
- "runtime"
- "testing"
- "time"
- "github.com/docker/docker/api/types"
- "github.com/docker/docker/api/types/filters"
- "github.com/docker/docker/api/types/swarm"
- "github.com/docker/docker/client"
- "github.com/docker/docker/integration-cli/request"
- "github.com/gotestyourself/gotestyourself/poll"
- "github.com/stretchr/testify/assert"
- "github.com/stretchr/testify/require"
- "golang.org/x/net/context"
- )
- func TestCreateServiceMultipleTimes(t *testing.T) {
- defer setupTest(t)()
- d := newSwarm(t)
- defer d.Stop(t)
- client, err := request.NewClientForHost(d.Sock())
- require.NoError(t, err)
- overlayName := "overlay1"
- networkCreate := types.NetworkCreate{
- CheckDuplicate: true,
- Driver: "overlay",
- }
- netResp, err := client.NetworkCreate(context.Background(), overlayName, networkCreate)
- require.NoError(t, err)
- overlayID := netResp.ID
- var instances uint64 = 4
- serviceSpec := swarmServiceSpec("TestService", instances)
- serviceSpec.TaskTemplate.Networks = append(serviceSpec.TaskTemplate.Networks, swarm.NetworkAttachmentConfig{Target: overlayName})
- serviceResp, err := client.ServiceCreate(context.Background(), serviceSpec, types.ServiceCreateOptions{
- QueryRegistry: false,
- })
- require.NoError(t, err)
- pollSettings := func(config *poll.Settings) {
- // It takes about ~25s to finish the multi services creation in this case per the pratical observation on arm64/arm platform
- if runtime.GOARCH == "arm64" || runtime.GOARCH == "arm" {
- config.Timeout = 30 * time.Second
- config.Delay = 100 * time.Millisecond
- }
- }
- serviceID := serviceResp.ID
- poll.WaitOn(t, serviceRunningTasksCount(client, serviceID, instances), pollSettings)
- _, _, err = client.ServiceInspectWithRaw(context.Background(), serviceID, types.ServiceInspectOptions{})
- require.NoError(t, err)
- err = client.ServiceRemove(context.Background(), serviceID)
- require.NoError(t, err)
- poll.WaitOn(t, serviceIsRemoved(client, serviceID), pollSettings)
- poll.WaitOn(t, noTasks(client), pollSettings)
- serviceResp, err = client.ServiceCreate(context.Background(), serviceSpec, types.ServiceCreateOptions{
- QueryRegistry: false,
- })
- require.NoError(t, err)
- serviceID2 := serviceResp.ID
- poll.WaitOn(t, serviceRunningTasksCount(client, serviceID2, instances), pollSettings)
- err = client.ServiceRemove(context.Background(), serviceID2)
- require.NoError(t, err)
- poll.WaitOn(t, serviceIsRemoved(client, serviceID2), pollSettings)
- poll.WaitOn(t, noTasks(client), pollSettings)
- err = client.NetworkRemove(context.Background(), overlayID)
- require.NoError(t, err)
- poll.WaitOn(t, networkIsRemoved(client, overlayID), poll.WithTimeout(1*time.Minute), poll.WithDelay(10*time.Second))
- }
- func TestCreateWithDuplicateNetworkNames(t *testing.T) {
- defer setupTest(t)()
- d := newSwarm(t)
- defer d.Stop(t)
- client, err := request.NewClientForHost(d.Sock())
- require.NoError(t, err)
- name := "foo"
- networkCreate := types.NetworkCreate{
- CheckDuplicate: false,
- Driver: "bridge",
- }
- n1, err := client.NetworkCreate(context.Background(), name, networkCreate)
- require.NoError(t, err)
- n2, err := client.NetworkCreate(context.Background(), name, networkCreate)
- require.NoError(t, err)
- // Dupliates with name but with different driver
- networkCreate.Driver = "overlay"
- n3, err := client.NetworkCreate(context.Background(), name, networkCreate)
- require.NoError(t, err)
- // Create Service with the same name
- var instances uint64 = 1
- serviceSpec := swarmServiceSpec("top", instances)
- serviceSpec.TaskTemplate.Networks = append(serviceSpec.TaskTemplate.Networks, swarm.NetworkAttachmentConfig{Target: name})
- service, err := client.ServiceCreate(context.Background(), serviceSpec, types.ServiceCreateOptions{})
- require.NoError(t, err)
- poll.WaitOn(t, serviceRunningTasksCount(client, service.ID, instances))
- resp, _, err := client.ServiceInspectWithRaw(context.Background(), service.ID, types.ServiceInspectOptions{})
- require.NoError(t, err)
- assert.Equal(t, n3.ID, resp.Spec.TaskTemplate.Networks[0].Target)
- // Remove Service
- err = client.ServiceRemove(context.Background(), service.ID)
- require.NoError(t, err)
- // Make sure task has been destroyed.
- poll.WaitOn(t, serviceIsRemoved(client, service.ID))
- // Remove networks
- err = client.NetworkRemove(context.Background(), n3.ID)
- require.NoError(t, err)
- err = client.NetworkRemove(context.Background(), n2.ID)
- require.NoError(t, err)
- err = client.NetworkRemove(context.Background(), n1.ID)
- require.NoError(t, err)
- // Make sure networks have been destroyed.
- poll.WaitOn(t, networkIsRemoved(client, n3.ID), poll.WithTimeout(1*time.Minute), poll.WithDelay(10*time.Second))
- poll.WaitOn(t, networkIsRemoved(client, n2.ID), poll.WithTimeout(1*time.Minute), poll.WithDelay(10*time.Second))
- poll.WaitOn(t, networkIsRemoved(client, n1.ID), poll.WithTimeout(1*time.Minute), poll.WithDelay(10*time.Second))
- }
- func swarmServiceSpec(name string, replicas uint64) swarm.ServiceSpec {
- return swarm.ServiceSpec{
- Annotations: swarm.Annotations{
- Name: name,
- },
- TaskTemplate: swarm.TaskSpec{
- ContainerSpec: &swarm.ContainerSpec{
- Image: "busybox:latest",
- Command: []string{"/bin/top"},
- },
- },
- Mode: swarm.ServiceMode{
- Replicated: &swarm.ReplicatedService{
- Replicas: &replicas,
- },
- },
- }
- }
- func serviceRunningTasksCount(client client.ServiceAPIClient, serviceID string, instances uint64) func(log poll.LogT) poll.Result {
- return func(log poll.LogT) poll.Result {
- filter := filters.NewArgs()
- filter.Add("service", serviceID)
- tasks, err := client.TaskList(context.Background(), types.TaskListOptions{
- Filters: filter,
- })
- switch {
- case err != nil:
- return poll.Error(err)
- case len(tasks) == int(instances):
- for _, task := range tasks {
- if task.Status.State != swarm.TaskStateRunning {
- return poll.Continue("waiting for tasks to enter run state")
- }
- }
- return poll.Success()
- default:
- return poll.Continue("task count at %d waiting for %d", len(tasks), instances)
- }
- }
- }
- func noTasks(client client.ServiceAPIClient) func(log poll.LogT) poll.Result {
- return func(log poll.LogT) poll.Result {
- filter := filters.NewArgs()
- tasks, err := client.TaskList(context.Background(), types.TaskListOptions{
- Filters: filter,
- })
- switch {
- case err != nil:
- return poll.Error(err)
- case len(tasks) == 0:
- return poll.Success()
- default:
- return poll.Continue("task count at %d waiting for 0", len(tasks))
- }
- }
- }
- func serviceIsRemoved(client client.ServiceAPIClient, serviceID string) func(log poll.LogT) poll.Result {
- return func(log poll.LogT) poll.Result {
- filter := filters.NewArgs()
- filter.Add("service", serviceID)
- _, err := client.TaskList(context.Background(), types.TaskListOptions{
- Filters: filter,
- })
- if err == nil {
- return poll.Continue("waiting for service %s to be deleted", serviceID)
- }
- return poll.Success()
- }
- }
- func networkIsRemoved(client client.NetworkAPIClient, networkID string) func(log poll.LogT) poll.Result {
- return func(log poll.LogT) poll.Result {
- _, err := client.NetworkInspect(context.Background(), networkID, types.NetworkInspectOptions{})
- if err == nil {
- return poll.Continue("waiting for network %s to be removed", networkID)
- }
- return poll.Success()
- }
- }
|