mirror of
https://github.com/opencontainers/runc.git
synced 2026-07-11 06:03:57 +08:00
Daniel, Dao Quang Minh
2f1b2ce204
when the test binary starts, it detects whether it should run the container initialization code or the execin initialization code based on the suppplied arguments. The execin initialization code is taken from docker. also added a sample test for execin process. Docker-DCO-1.1-Signed-off-by: Daniel, Dao Quang Minh <dqminh89@gmail.com> (github: dqminh)
84 lines
1.9 KiB
Go
84 lines
1.9 KiB
Go
package integration
|
|
|
|
import (
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/docker/libcontainer"
|
|
"github.com/docker/libcontainer/namespaces"
|
|
)
|
|
|
|
func TestExecIn(t *testing.T) {
|
|
if testing.Short() {
|
|
return
|
|
}
|
|
|
|
rootfs, err := newRootFs()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer remove(rootfs)
|
|
|
|
config := newTemplateConfig(rootfs)
|
|
if err := writeConfig(config); err != nil {
|
|
t.Fatalf("failed to write config %s", err)
|
|
}
|
|
|
|
// start the container
|
|
containerErr := make(chan error, 1)
|
|
containerCmd := &exec.Cmd{}
|
|
var statePath string
|
|
createCmd := func(container *libcontainer.Config, console, dataPath, init string,
|
|
pipe *os.File, args []string) *exec.Cmd {
|
|
containerCmd = namespaces.DefaultCreateCommand(container, console, dataPath, init, pipe, args)
|
|
statePath = dataPath
|
|
return containerCmd
|
|
}
|
|
var containerStart sync.WaitGroup
|
|
containerStart.Add(1)
|
|
go func() {
|
|
buffers := newStdBuffers()
|
|
_, err := namespaces.Exec(config,
|
|
buffers.Stdin, buffers.Stdout, buffers.Stderr,
|
|
"", config.RootFs, []string{"sleep", "10"},
|
|
createCmd, containerStart.Done)
|
|
containerErr <- err
|
|
}()
|
|
containerStart.Wait()
|
|
|
|
defer func() {
|
|
// kill the container
|
|
if containerCmd.Process != nil {
|
|
containerCmd.Process.Kill()
|
|
}
|
|
if err := <-containerErr; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}()
|
|
|
|
// start the exec process
|
|
state, err := libcontainer.GetState(statePath)
|
|
if err != nil {
|
|
t.Fatalf("failed to get state %s", err)
|
|
}
|
|
buffers := newStdBuffers()
|
|
execErr := make(chan error, 1)
|
|
go func() {
|
|
_, err := namespaces.ExecIn(config, state, []string{"ps"},
|
|
os.Args[0], "exec", buffers.Stdin, buffers.Stdout, buffers.Stderr,
|
|
"", nil)
|
|
execErr <- err
|
|
}()
|
|
if err := <-execErr; err != nil {
|
|
t.Fatalf("exec finished with error %s", err)
|
|
}
|
|
|
|
out := buffers.Stdout.String()
|
|
if !strings.Contains(out, "sleep 10") || !strings.Contains(out, "ps") {
|
|
t.Fatalf("unexpected running process, output %q", out)
|
|
}
|
|
}
|