create_test.go 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129
  1. package service
  2. import (
  3. "testing"
  4. "time"
  5. "github.com/docker/docker/api/types"
  6. "github.com/docker/docker/api/types/filters"
  7. "github.com/docker/docker/api/types/swarm"
  8. "github.com/docker/docker/client"
  9. "github.com/docker/docker/integration-cli/request"
  10. "github.com/gotestyourself/gotestyourself/poll"
  11. "github.com/stretchr/testify/assert"
  12. "github.com/stretchr/testify/require"
  13. "golang.org/x/net/context"
  14. )
  15. func TestCreateWithLBSandbox(t *testing.T) {
  16. defer setupTest(t)()
  17. d := newSwarm(t)
  18. defer d.Stop(t)
  19. client, err := request.NewClientForHost(d.Sock())
  20. require.NoError(t, err)
  21. overlayName := "overlay1"
  22. networkCreate := types.NetworkCreate{
  23. CheckDuplicate: true,
  24. Driver: "overlay",
  25. }
  26. netResp, err := client.NetworkCreate(context.Background(), overlayName, networkCreate)
  27. require.NoError(t, err)
  28. overlayID := netResp.ID
  29. var instances uint64 = 1
  30. serviceSpec := swarmServiceSpec("TestService", instances)
  31. serviceSpec.TaskTemplate.Networks = append(serviceSpec.TaskTemplate.Networks, swarm.NetworkAttachmentConfig{Target: overlayName})
  32. serviceResp, err := client.ServiceCreate(context.Background(), serviceSpec, types.ServiceCreateOptions{
  33. QueryRegistry: false,
  34. })
  35. require.NoError(t, err)
  36. serviceID := serviceResp.ID
  37. poll.WaitOn(t, serviceRunningTasksCount(client, serviceID, instances))
  38. _, _, err = client.ServiceInspectWithRaw(context.Background(), serviceID, types.ServiceInspectOptions{})
  39. require.NoError(t, err)
  40. network, err := client.NetworkInspect(context.Background(), overlayID, types.NetworkInspectOptions{})
  41. require.NoError(t, err)
  42. assert.Contains(t, network.Containers, overlayName+"-sbox")
  43. err = client.ServiceRemove(context.Background(), serviceID)
  44. require.NoError(t, err)
  45. poll.WaitOn(t, serviceIsRemoved(client, serviceID))
  46. err = client.NetworkRemove(context.Background(), overlayID)
  47. require.NoError(t, err)
  48. poll.WaitOn(t, networkIsRemoved(client, overlayID), poll.WithTimeout(1*time.Minute), poll.WithDelay(10*time.Second))
  49. }
  50. func swarmServiceSpec(name string, replicas uint64) swarm.ServiceSpec {
  51. return swarm.ServiceSpec{
  52. Annotations: swarm.Annotations{
  53. Name: name,
  54. },
  55. TaskTemplate: swarm.TaskSpec{
  56. ContainerSpec: &swarm.ContainerSpec{
  57. Image: "busybox:latest",
  58. Command: []string{"/bin/top"},
  59. },
  60. },
  61. Mode: swarm.ServiceMode{
  62. Replicated: &swarm.ReplicatedService{
  63. Replicas: &replicas,
  64. },
  65. },
  66. }
  67. }
  68. func serviceRunningTasksCount(client client.ServiceAPIClient, serviceID string, instances uint64) func(log poll.LogT) poll.Result {
  69. return func(log poll.LogT) poll.Result {
  70. filter := filters.NewArgs()
  71. filter.Add("service", serviceID)
  72. tasks, err := client.TaskList(context.Background(), types.TaskListOptions{
  73. Filters: filter,
  74. })
  75. switch {
  76. case err != nil:
  77. return poll.Error(err)
  78. case len(tasks) == int(instances):
  79. for _, task := range tasks {
  80. if task.Status.State != swarm.TaskStateRunning {
  81. return poll.Continue("waiting for tasks to enter run state")
  82. }
  83. }
  84. return poll.Success()
  85. default:
  86. return poll.Continue("task count at %d waiting for %d", len(tasks), instances)
  87. }
  88. }
  89. }
  90. func serviceIsRemoved(client client.ServiceAPIClient, serviceID string) func(log poll.LogT) poll.Result {
  91. return func(log poll.LogT) poll.Result {
  92. filter := filters.NewArgs()
  93. filter.Add("service", serviceID)
  94. _, err := client.TaskList(context.Background(), types.TaskListOptions{
  95. Filters: filter,
  96. })
  97. if err == nil {
  98. return poll.Continue("waiting for service %s to be deleted", serviceID)
  99. }
  100. return poll.Success()
  101. }
  102. }
  103. func networkIsRemoved(client client.NetworkAPIClient, networkID string) func(log poll.LogT) poll.Result {
  104. return func(log poll.LogT) poll.Result {
  105. _, err := client.NetworkInspect(context.Background(), networkID, types.NetworkInspectOptions{})
  106. if err == nil {
  107. return poll.Continue("waiting for network %s to be removed", networkID)
  108. }
  109. return poll.Success()
  110. }
  111. }