| package main |
| |
| import ( |
| "fmt" |
| "os/exec" |
| "strings" |
| "testing" |
| ) |
| |
| func TestTopNonPrivileged(t *testing.T) { |
| runCmd := exec.Command(dockerBinary, "run", "-i", "-d", "busybox", "sleep", "20") |
| out, _, err := runCommandWithOutput(runCmd) |
| errorOut(err, t, fmt.Sprintf("failed to start the container: %v", err)) |
| |
| cleanedContainerID := stripTrailingCharacters(out) |
| |
| topCmd := exec.Command(dockerBinary, "top", cleanedContainerID) |
| out, _, err = runCommandWithOutput(topCmd) |
| errorOut(err, t, fmt.Sprintf("failed to run top: %v %v", out, err)) |
| |
| topCmd = exec.Command(dockerBinary, "top", cleanedContainerID) |
| out2, _, err2 := runCommandWithOutput(topCmd) |
| errorOut(err, t, fmt.Sprintf("failed to run top: %v %v", out2, err2)) |
| |
| killCmd := exec.Command(dockerBinary, "kill", cleanedContainerID) |
| _, err = runCommand(killCmd) |
| errorOut(err, t, fmt.Sprintf("failed to kill container: %v", err)) |
| |
| deleteContainer(cleanedContainerID) |
| |
| if !strings.Contains(out, "sleep 20") && !strings.Contains(out2, "sleep 20") { |
| t.Fatal("top should've listed `sleep 20` in the process list, but failed twice") |
| } else if !strings.Contains(out, "sleep 20") { |
| t.Fatal("top should've listed `sleep 20` in the process list, but failed the first time") |
| } else if !strings.Contains(out2, "sleep 20") { |
| t.Fatal("top should've listed `sleep 20` in the process list, but failed the second itime") |
| } |
| |
| logDone("top - sleep process should be listed in non privileged mode") |
| } |
| |
| func TestTopPrivileged(t *testing.T) { |
| runCmd := exec.Command(dockerBinary, "run", "--privileged", "-i", "-d", "busybox", "sleep", "20") |
| out, _, err := runCommandWithOutput(runCmd) |
| errorOut(err, t, fmt.Sprintf("failed to start the container: %v", err)) |
| |
| cleanedContainerID := stripTrailingCharacters(out) |
| |
| topCmd := exec.Command(dockerBinary, "top", cleanedContainerID) |
| out, _, err = runCommandWithOutput(topCmd) |
| errorOut(err, t, fmt.Sprintf("failed to run top: %v %v", out, err)) |
| |
| topCmd = exec.Command(dockerBinary, "top", cleanedContainerID) |
| out2, _, err2 := runCommandWithOutput(topCmd) |
| errorOut(err, t, fmt.Sprintf("failed to run top: %v %v", out2, err2)) |
| |
| killCmd := exec.Command(dockerBinary, "kill", cleanedContainerID) |
| _, err = runCommand(killCmd) |
| errorOut(err, t, fmt.Sprintf("failed to kill container: %v", err)) |
| |
| deleteContainer(cleanedContainerID) |
| |
| if !strings.Contains(out, "sleep 20") && !strings.Contains(out2, "sleep 20") { |
| t.Fatal("top should've listed `sleep 20` in the process list, but failed twice") |
| } else if !strings.Contains(out, "sleep 20") { |
| t.Fatal("top should've listed `sleep 20` in the process list, but failed the first time") |
| } else if !strings.Contains(out2, "sleep 20") { |
| t.Fatal("top should've listed `sleep 20` in the process list, but failed the second itime") |
| } |
| |
| logDone("top - sleep process should be listed in privileged mode") |
| } |