1 // Copyright (C) The Arvados Authors. All rights reserved.
3 // SPDX-License-Identifier: AGPL-3.0
16 "golang.org/x/net/context"
20 var _ = Suite(&dockerSuite{})
22 func busyboxDockerImage(c *C) string {
23 fnm := "busybox_uclibc.tar"
24 cachedir := "/var/lib/arvados/tmp"
25 cachefile := cachedir + "/" + fnm
26 if _, err := os.Stat(cachefile); err == nil {
30 f, err := ioutil.TempFile(cachedir, "")
33 defer os.Remove(f.Name())
35 resp, err := http.Get("http://cache.arvados.org/" + fnm)
37 defer resp.Body.Close()
38 _, err = io.Copy(f, resp.Body)
42 err = os.Rename(f.Name(), cachefile)
48 type nopWriteCloser struct{ io.Writer }
50 func (nopWriteCloser) Close() error { return nil }
52 // embedded by dockerSuite and singularitySuite so they can share
54 type executorSuite struct {
55 newExecutor func(*C) // embedding struct's SetUpSuite method must set this
56 executor containerExecutor
62 func (s *executorSuite) SetUpTest(c *C) {
64 s.stdout = bytes.Buffer{}
65 s.stderr = bytes.Buffer{}
66 s.spec = containerSpec{
67 Image: "busybox:uclibc",
70 Env: map[string]string{"PATH": "/bin:/usr/bin"},
71 NetworkMode: "default",
72 Stdout: nopWriteCloser{&s.stdout},
73 Stderr: nopWriteCloser{&s.stderr},
75 err := s.executor.LoadImage(busyboxDockerImage(c))
79 func (s *executorSuite) TearDownTest(c *C) {
83 func (s *executorSuite) TestExecTrivialContainer(c *C) {
84 s.spec.Command = []string{"echo", "ok"}
86 c.Check(s.stdout.String(), Equals, "ok\n")
87 c.Check(s.stderr.String(), Equals, "")
90 func (s *executorSuite) TestExecStop(c *C) {
91 s.spec.Command = []string{"sh", "-c", "sleep 10; echo ok"}
92 err := s.executor.Create(s.spec)
94 err = s.executor.Start()
97 time.Sleep(time.Second / 10)
100 ctx, cancel := context.WithDeadline(context.Background(), time.Now().Add(10*time.Second))
102 code, err := s.executor.Wait(ctx)
103 c.Check(code, Not(Equals), 0)
105 c.Check(s.stdout.String(), Equals, "")
106 c.Check(s.stderr.String(), Equals, "")
109 func (s *executorSuite) TestExecCleanEnv(c *C) {
110 s.spec.Command = []string{"env"}
112 c.Check(s.stderr.String(), Equals, "")
113 got := map[string]string{}
114 for _, kv := range strings.Split(s.stdout.String(), "\n") {
118 kv := strings.SplitN(kv, "=", 2)
120 case "HOSTNAME", "HOME":
121 // docker sets these by itself
122 case "LD_LIBRARY_PATH", "SINGULARITY_NAME", "PWD", "LANG", "SHLVL", "SINGULARITY_INIT", "SINGULARITY_CONTAINER":
123 // singularity sets these by itself (cf. https://sylabs.io/guides/3.5/user-guide/environment_and_metadata.html)
124 case "PROMPT_COMMAND", "PS1", "SINGULARITY_APPNAME":
125 // singularity also sets these by itself (as of v3.5.2)
130 c.Check(got, DeepEquals, s.spec.Env)
132 func (s *executorSuite) TestExecEnableNetwork(c *C) {
133 for _, enable := range []bool{false, true} {
135 s.spec.Command = []string{"ip", "route"}
136 s.spec.EnableNetwork = enable
139 c.Check(s.stdout.String(), Matches, "(?ms).*default via.*")
141 c.Check(s.stdout.String(), Equals, "")
146 func (s *executorSuite) TestExecStdoutStderr(c *C) {
147 s.spec.Command = []string{"sh", "-c", "echo foo; echo -n bar >&2; echo baz; echo waz >&2"}
149 c.Check(s.stdout.String(), Equals, "foo\nbaz\n")
150 c.Check(s.stderr.String(), Equals, "barwaz\n")
153 func (s *executorSuite) checkRun(c *C, expectCode int) {
154 c.Assert(s.executor.Create(s.spec), IsNil)
155 c.Assert(s.executor.Start(), IsNil)
156 ctx, cancel := context.WithDeadline(context.Background(), time.Now().Add(10*time.Second))
158 code, err := s.executor.Wait(ctx)
160 c.Check(code, Equals, expectCode)