docker_cli_wait_test.go 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121
  1. package main
  2. import (
  3. "os/exec"
  4. "testing"
  5. "time"
  6. )
  7. // non-blocking wait with 0 exit code
  8. func TestWaitNonBlockedExitZero(t *testing.T) {
  9. defer deleteAllContainers()
  10. runCmd := exec.Command(dockerBinary, "run", "-d", "busybox", "sh", "-c", "true")
  11. out, _, err := runCommandWithOutput(runCmd)
  12. if err != nil {
  13. t.Fatal(out, err)
  14. }
  15. containerID := stripTrailingCharacters(out)
  16. status := "true"
  17. for i := 0; status != "false"; i++ {
  18. runCmd = exec.Command(dockerBinary, "inspect", "--format='{{.State.Running}}'", containerID)
  19. status, _, err = runCommandWithOutput(runCmd)
  20. if err != nil {
  21. t.Fatal(status, err)
  22. }
  23. status = stripTrailingCharacters(status)
  24. time.Sleep(time.Second)
  25. if i >= 60 {
  26. t.Fatal("Container should have stopped by now")
  27. }
  28. }
  29. runCmd = exec.Command(dockerBinary, "wait", containerID)
  30. out, _, err = runCommandWithOutput(runCmd)
  31. if err != nil || stripTrailingCharacters(out) != "0" {
  32. t.Fatal("failed to set up container", out, err)
  33. }
  34. logDone("wait - non-blocking wait with 0 exit code")
  35. }
  36. // blocking wait with 0 exit code
  37. func TestWaitBlockedExitZero(t *testing.T) {
  38. defer deleteAllContainers()
  39. runCmd := exec.Command(dockerBinary, "run", "-d", "busybox", "sh", "-c", "sleep 10")
  40. out, _, err := runCommandWithOutput(runCmd)
  41. if err != nil {
  42. t.Fatal(out, err)
  43. }
  44. containerID := stripTrailingCharacters(out)
  45. runCmd = exec.Command(dockerBinary, "wait", containerID)
  46. out, _, err = runCommandWithOutput(runCmd)
  47. if err != nil || stripTrailingCharacters(out) != "0" {
  48. t.Fatal("failed to set up container", out, err)
  49. }
  50. logDone("wait - blocking wait with 0 exit code")
  51. }
  52. // non-blocking wait with random exit code
  53. func TestWaitNonBlockedExitRandom(t *testing.T) {
  54. defer deleteAllContainers()
  55. runCmd := exec.Command(dockerBinary, "run", "-d", "busybox", "sh", "-c", "exit 99")
  56. out, _, err := runCommandWithOutput(runCmd)
  57. if err != nil {
  58. t.Fatal(out, err)
  59. }
  60. containerID := stripTrailingCharacters(out)
  61. status := "true"
  62. for i := 0; status != "false"; i++ {
  63. runCmd = exec.Command(dockerBinary, "inspect", "--format='{{.State.Running}}'", containerID)
  64. status, _, err = runCommandWithOutput(runCmd)
  65. if err != nil {
  66. t.Fatal(status, err)
  67. }
  68. status = stripTrailingCharacters(status)
  69. time.Sleep(time.Second)
  70. if i >= 60 {
  71. t.Fatal("Container should have stopped by now")
  72. }
  73. }
  74. runCmd = exec.Command(dockerBinary, "wait", containerID)
  75. out, _, err = runCommandWithOutput(runCmd)
  76. if err != nil || stripTrailingCharacters(out) != "99" {
  77. t.Fatal("failed to set up container", out, err)
  78. }
  79. logDone("wait - non-blocking wait with random exit code")
  80. }
  81. // blocking wait with random exit code
  82. func TestWaitBlockedExitRandom(t *testing.T) {
  83. defer deleteAllContainers()
  84. runCmd := exec.Command(dockerBinary, "run", "-d", "busybox", "sh", "-c", "sleep 10; exit 99")
  85. out, _, err := runCommandWithOutput(runCmd)
  86. if err != nil {
  87. t.Fatal(out, err)
  88. }
  89. containerID := stripTrailingCharacters(out)
  90. runCmd = exec.Command(dockerBinary, "wait", containerID)
  91. out, _, err = runCommandWithOutput(runCmd)
  92. if err != nil || stripTrailingCharacters(out) != "99" {
  93. t.Fatal("failed to set up container", out, err)
  94. }
  95. logDone("wait - blocking wait with random exit code")
  96. }