45 lines
958 B
Go
45 lines
958 B
Go
|
// Copyright 2018 Mikhail Klementev. All rights reserved.
|
||
|
|
// Use of this source code is governed by a AGPLv3 license
|
||
|
|
// (or later) that can be found in the LICENSE file.
|
||
|
|
|
||
|
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"io/ioutil"
|
||
|
|
"os"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestDockerCommand(t *testing.T) {
|
||
|
|
tmp, err := ioutil.TempDir("/tmp/", "out-of-tree_test_")
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
defer os.RemoveAll(tmp)
|
||
|
|
|
||
|
|
start := time.Now()
|
||
|
|
|
||
|
|
c := dockerCommand("ubuntu", tmp, "1s", "sleep 5s")
|
||
|
|
_, err = c.CombinedOutput()
|
||
|
|
if err == nil {
|
||
|
|
t.Fatal("docker is not killed by timeout")
|
||
|
|
}
|
||
|
|
if time.Since(start) > 3*time.Second {
|
||
|
|
t.Fatal(fmt.Sprintf("timeout failed (%v instead of %v)",
|
||
|
|
time.Since(start), time.Second))
|
||
|
|
}
|
||
|
|
|
||
|
|
c = dockerCommand("ubuntu", tmp, "1m", "echo hello")
|
||
|
|
rawOutput, err := c.CombinedOutput()
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if !strings.Contains(string(rawOutput), "hello") {
|
||
|
|
t.Fatal("wrong output (" + string(rawOutput) + ")")
|
||
|
|
}
|
||
|
|
}
|