108 lines
3.4 KiB
Go
108 lines
3.4 KiB
Go
package main
|
|
|
|
import (
|
|
"os/exec"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestTopMultipleArgs(t *testing.T) {
|
|
runCmd := exec.Command(dockerBinary, "run", "-i", "-d", "busybox", "top")
|
|
out, _, err := runCommandWithOutput(runCmd)
|
|
if err != nil {
|
|
t.Fatalf("failed to start the container: %s, %v", out, err)
|
|
}
|
|
|
|
cleanedContainerID := strings.TrimSpace(out)
|
|
defer deleteContainer(cleanedContainerID)
|
|
|
|
topCmd := exec.Command(dockerBinary, "top", cleanedContainerID, "-o", "pid")
|
|
out, _, err = runCommandWithOutput(topCmd)
|
|
if err != nil {
|
|
t.Fatalf("failed to run top: %s, %v", out, err)
|
|
}
|
|
|
|
if !strings.Contains(out, "PID") {
|
|
t.Fatalf("did not see PID after top -o pid: %s", out)
|
|
}
|
|
|
|
logDone("top - multiple arguments")
|
|
}
|
|
|
|
func TestTopNonPrivileged(t *testing.T) {
|
|
runCmd := exec.Command(dockerBinary, "run", "-i", "-d", "busybox", "top")
|
|
out, _, err := runCommandWithOutput(runCmd)
|
|
if err != nil {
|
|
t.Fatalf("failed to start the container: %s, %v", out, err)
|
|
}
|
|
|
|
cleanedContainerID := strings.TrimSpace(out)
|
|
|
|
topCmd := exec.Command(dockerBinary, "top", cleanedContainerID)
|
|
out1, _, err := runCommandWithOutput(topCmd)
|
|
if err != nil {
|
|
t.Fatalf("failed to run top: %s, %v", out1, err)
|
|
}
|
|
|
|
topCmd = exec.Command(dockerBinary, "top", cleanedContainerID)
|
|
out2, _, err := runCommandWithOutput(topCmd)
|
|
if err != nil {
|
|
t.Fatalf("failed to run top: %s, %v", out2, err)
|
|
}
|
|
|
|
killCmd := exec.Command(dockerBinary, "kill", cleanedContainerID)
|
|
if out, _, err = runCommandWithOutput(killCmd); err != nil {
|
|
t.Fatalf("failed to kill container: %s, %v", out, err)
|
|
}
|
|
|
|
deleteContainer(cleanedContainerID)
|
|
|
|
if !strings.Contains(out1, "top") && !strings.Contains(out2, "top") {
|
|
t.Fatal("top should've listed `top` in the process list, but failed twice")
|
|
} else if !strings.Contains(out1, "top") {
|
|
t.Fatal("top should've listed `top` in the process list, but failed the first time")
|
|
} else if !strings.Contains(out2, "top") {
|
|
t.Fatal("top should've listed `top` in the process list, but failed the second itime")
|
|
}
|
|
|
|
logDone("top - top process should be listed in non privileged mode")
|
|
}
|
|
|
|
func TestTopPrivileged(t *testing.T) {
|
|
runCmd := exec.Command(dockerBinary, "run", "--privileged", "-i", "-d", "busybox", "top")
|
|
out, _, err := runCommandWithOutput(runCmd)
|
|
if err != nil {
|
|
t.Fatalf("failed to start the container: %s, %v", out, err)
|
|
}
|
|
|
|
cleanedContainerID := strings.TrimSpace(out)
|
|
|
|
topCmd := exec.Command(dockerBinary, "top", cleanedContainerID)
|
|
out1, _, err := runCommandWithOutput(topCmd)
|
|
if err != nil {
|
|
t.Fatalf("failed to run top: %s, %v", out1, err)
|
|
}
|
|
|
|
topCmd = exec.Command(dockerBinary, "top", cleanedContainerID)
|
|
out2, _, err := runCommandWithOutput(topCmd)
|
|
if err != nil {
|
|
t.Fatalf("failed to run top: %s, %v", out2, err)
|
|
}
|
|
|
|
killCmd := exec.Command(dockerBinary, "kill", cleanedContainerID)
|
|
if out, _, err = runCommandWithOutput(killCmd); err != nil {
|
|
t.Fatalf("failed to kill container: %s, %v", out, err)
|
|
}
|
|
|
|
deleteContainer(cleanedContainerID)
|
|
|
|
if !strings.Contains(out1, "top") && !strings.Contains(out2, "top") {
|
|
t.Fatal("top should've listed `top` in the process list, but failed twice")
|
|
} else if !strings.Contains(out1, "top") {
|
|
t.Fatal("top should've listed `top` in the process list, but failed the first time")
|
|
} else if !strings.Contains(out2, "top") {
|
|
t.Fatal("top should've listed `top` in the process list, but failed the second itime")
|
|
}
|
|
|
|
logDone("top - top process should be listed in privileged mode")
|
|
}
|