123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008 |
- package docker
- import (
- "bufio"
- "fmt"
- "io"
- "io/ioutil"
- "math/rand"
- "os"
- "regexp"
- "sort"
- "strings"
- "testing"
- "time"
- )
- func TestIdFormat(t *testing.T) {
- runtime, err := newTestRuntime()
- if err != nil {
- t.Fatal(err)
- }
- defer nuke(runtime)
- container1, err := runtime.Create(
- &Config{
- Image: GetTestImage(runtime).Id,
- Cmd: []string{"/bin/sh", "-c", "echo hello world"},
- Memory: 33554432,
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- match, err := regexp.Match("^[0-9a-f]{64}$", []byte(container1.Id))
- if err != nil {
- t.Fatal(err)
- }
- if !match {
- t.Fatalf("Invalid container ID: %s", container1.Id)
- }
- }
- func TestMultipleAttachRestart(t *testing.T) {
- runtime, err := newTestRuntime()
- if err != nil {
- t.Fatal(err)
- }
- defer nuke(runtime)
- container, err := runtime.Create(
- &Config{
- Image: GetTestImage(runtime).Id,
- Cmd: []string{"/bin/sh", "-c",
- "i=1; while [ $i -le 5 ]; do i=`expr $i + 1`; echo hello; done"},
- Memory: 33554432,
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container)
- // Simulate 3 client attaching to the container and stop/restart
- stdout1, err := container.StdoutPipe()
- if err != nil {
- t.Fatal(err)
- }
- stdout2, err := container.StdoutPipe()
- if err != nil {
- t.Fatal(err)
- }
- stdout3, err := container.StdoutPipe()
- if err != nil {
- t.Fatal(err)
- }
- if err := container.Start(); err != nil {
- t.Fatal(err)
- }
- l1, err := bufio.NewReader(stdout1).ReadString('\n')
- if err != nil {
- t.Fatal(err)
- }
- if strings.Trim(l1, " \r\n") != "hello" {
- t.Fatalf("Unexpected output. Expected [%s], received [%s]", "hello", l1)
- }
- l2, err := bufio.NewReader(stdout2).ReadString('\n')
- if err != nil {
- t.Fatal(err)
- }
- if strings.Trim(l2, " \r\n") != "hello" {
- t.Fatalf("Unexpected output. Expected [%s], received [%s]", "hello", l2)
- }
- l3, err := bufio.NewReader(stdout3).ReadString('\n')
- if err != nil {
- t.Fatal(err)
- }
- if strings.Trim(l3, " \r\n") != "hello" {
- t.Fatalf("Unexpected output. Expected [%s], received [%s]", "hello", l3)
- }
- if err := container.Stop(); err != nil {
- t.Fatal(err)
- }
- stdout1, err = container.StdoutPipe()
- if err != nil {
- t.Fatal(err)
- }
- stdout2, err = container.StdoutPipe()
- if err != nil {
- t.Fatal(err)
- }
- stdout3, err = container.StdoutPipe()
- if err != nil {
- t.Fatal(err)
- }
- if err := container.Start(); err != nil {
- t.Fatal(err)
- }
- timeout := make(chan bool)
- go func() {
- l1, err = bufio.NewReader(stdout1).ReadString('\n')
- if err != nil {
- t.Fatal(err)
- }
- if strings.Trim(l1, " \r\n") != "hello" {
- t.Fatalf("Unexpected output. Expected [%s], received [%s]", "hello", l1)
- }
- l2, err = bufio.NewReader(stdout2).ReadString('\n')
- if err != nil {
- t.Fatal(err)
- }
- if strings.Trim(l2, " \r\n") != "hello" {
- t.Fatalf("Unexpected output. Expected [%s], received [%s]", "hello", l2)
- }
- l3, err = bufio.NewReader(stdout3).ReadString('\n')
- if err != nil {
- t.Fatal(err)
- }
- if strings.Trim(l3, " \r\n") != "hello" {
- t.Fatalf("Unexpected output. Expected [%s], received [%s]", "hello", l3)
- }
- timeout <- false
- }()
- go func() {
- time.Sleep(3 * time.Second)
- timeout <- true
- }()
- if <-timeout {
- t.Fatalf("Timeout reading from the process")
- }
- }
- func TestCommitRun(t *testing.T) {
- runtime, err := newTestRuntime()
- if err != nil {
- t.Fatal(err)
- }
- defer nuke(runtime)
- container1, err := runtime.Create(
- &Config{
- Image: GetTestImage(runtime).Id,
- Cmd: []string{"/bin/sh", "-c", "echo hello > /world"},
- Memory: 33554432,
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container1)
- if container1.State.Running {
- t.Errorf("Container shouldn't be running")
- }
- if err := container1.Run(); err != nil {
- t.Fatal(err)
- }
- if container1.State.Running {
- t.Errorf("Container shouldn't be running")
- }
- rwTar, err := container1.ExportRw()
- if err != nil {
- t.Error(err)
- }
- img, err := runtime.graph.Create(rwTar, container1, "unit test commited image")
- if err != nil {
- t.Error(err)
- }
- // FIXME: Make a TestCommit that stops here and check docker.root/layers/img.id/world
- container2, err := runtime.Create(
- &Config{
- Image: img.Id,
- Memory: 33554432,
- Cmd: []string{"cat", "/world"},
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container2)
- stdout, err := container2.StdoutPipe()
- if err != nil {
- t.Fatal(err)
- }
- stderr, err := container2.StderrPipe()
- if err != nil {
- t.Fatal(err)
- }
- if err := container2.Start(); err != nil {
- t.Fatal(err)
- }
- container2.Wait()
- output, err := ioutil.ReadAll(stdout)
- if err != nil {
- t.Fatal(err)
- }
- output2, err := ioutil.ReadAll(stderr)
- if err != nil {
- t.Fatal(err)
- }
- if err := stdout.Close(); err != nil {
- t.Fatal(err)
- }
- if err := stderr.Close(); err != nil {
- t.Fatal(err)
- }
- if string(output) != "hello\n" {
- t.Fatalf("Unexpected output. Expected %s, received: %s (err: %s)", "hello\n", output, output2)
- }
- }
- func TestStart(t *testing.T) {
- runtime, err := newTestRuntime()
- if err != nil {
- t.Fatal(err)
- }
- defer nuke(runtime)
- container, err := runtime.Create(
- &Config{
- Image: GetTestImage(runtime).Id,
- Memory: 33554432,
- Cmd: []string{"/bin/cat"},
- OpenStdin: true,
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container)
- if err := container.Start(); err != nil {
- t.Fatal(err)
- }
- // Give some time to the process to start
- container.WaitTimeout(500 * time.Millisecond)
- if !container.State.Running {
- t.Errorf("Container should be running")
- }
- if err := container.Start(); err == nil {
- t.Fatalf("A running containter should be able to be started")
- }
- // Try to avoid the timeoout in destroy. Best effort, don't check error
- cStdin, _ := container.StdinPipe()
- cStdin.Close()
- container.WaitTimeout(2 * time.Second)
- }
- func TestRun(t *testing.T) {
- runtime, err := newTestRuntime()
- if err != nil {
- t.Fatal(err)
- }
- defer nuke(runtime)
- container, err := runtime.Create(
- &Config{
- Image: GetTestImage(runtime).Id,
- Memory: 33554432,
- Cmd: []string{"ls", "-al"},
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container)
- if container.State.Running {
- t.Errorf("Container shouldn't be running")
- }
- if err := container.Run(); err != nil {
- t.Fatal(err)
- }
- if container.State.Running {
- t.Errorf("Container shouldn't be running")
- }
- }
- func TestOutput(t *testing.T) {
- runtime, err := newTestRuntime()
- if err != nil {
- t.Fatal(err)
- }
- defer nuke(runtime)
- container, err := runtime.Create(
- &Config{
- Image: GetTestImage(runtime).Id,
- Cmd: []string{"echo", "-n", "foobar"},
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container)
- output, err := container.Output()
- if err != nil {
- t.Fatal(err)
- }
- if string(output) != "foobar" {
- t.Error(string(output))
- }
- }
- func TestKillDifferentUser(t *testing.T) {
- runtime, err := newTestRuntime()
- if err != nil {
- t.Fatal(err)
- }
- defer nuke(runtime)
- container, err := runtime.Create(&Config{
- Image: GetTestImage(runtime).Id,
- Cmd: []string{"tail", "-f", "/etc/resolv.conf"},
- User: "daemon",
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container)
- if container.State.Running {
- t.Errorf("Container shouldn't be running")
- }
- if err := container.Start(); err != nil {
- t.Fatal(err)
- }
- // Give some time to lxc to spawn the process (setuid might take some time)
- container.WaitTimeout(500 * time.Millisecond)
- if !container.State.Running {
- t.Errorf("Container should be running")
- }
- if err := container.Kill(); err != nil {
- t.Fatal(err)
- }
- if container.State.Running {
- t.Errorf("Container shouldn't be running")
- }
- container.Wait()
- if container.State.Running {
- t.Errorf("Container shouldn't be running")
- }
- // Try stopping twice
- if err := container.Kill(); err != nil {
- t.Fatal(err)
- }
- }
- func TestKill(t *testing.T) {
- runtime, err := newTestRuntime()
- if err != nil {
- t.Fatal(err)
- }
- defer nuke(runtime)
- container, err := runtime.Create(&Config{
- Image: GetTestImage(runtime).Id,
- Cmd: []string{"cat", "/dev/zero"},
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container)
- if container.State.Running {
- t.Errorf("Container shouldn't be running")
- }
- if err := container.Start(); err != nil {
- t.Fatal(err)
- }
- // Give some time to lxc to spawn the process
- container.WaitTimeout(500 * time.Millisecond)
- if !container.State.Running {
- t.Errorf("Container should be running")
- }
- if err := container.Kill(); err != nil {
- t.Fatal(err)
- }
- if container.State.Running {
- t.Errorf("Container shouldn't be running")
- }
- container.Wait()
- if container.State.Running {
- t.Errorf("Container shouldn't be running")
- }
- // Try stopping twice
- if err := container.Kill(); err != nil {
- t.Fatal(err)
- }
- }
- func TestExitCode(t *testing.T) {
- runtime, err := newTestRuntime()
- if err != nil {
- t.Fatal(err)
- }
- defer nuke(runtime)
- trueContainer, err := runtime.Create(&Config{
- Image: GetTestImage(runtime).Id,
- Cmd: []string{"/bin/true", ""},
- })
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(trueContainer)
- if err := trueContainer.Run(); err != nil {
- t.Fatal(err)
- }
- if trueContainer.State.ExitCode != 0 {
- t.Errorf("Unexpected exit code %d (expected 0)", trueContainer.State.ExitCode)
- }
- falseContainer, err := runtime.Create(&Config{
- Image: GetTestImage(runtime).Id,
- Cmd: []string{"/bin/false", ""},
- })
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(falseContainer)
- if err := falseContainer.Run(); err != nil {
- t.Fatal(err)
- }
- if falseContainer.State.ExitCode != 1 {
- t.Errorf("Unexpected exit code %d (expected 1)", falseContainer.State.ExitCode)
- }
- }
- func TestRestart(t *testing.T) {
- runtime, err := newTestRuntime()
- if err != nil {
- t.Fatal(err)
- }
- defer nuke(runtime)
- container, err := runtime.Create(&Config{
- Image: GetTestImage(runtime).Id,
- Cmd: []string{"echo", "-n", "foobar"},
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container)
- output, err := container.Output()
- if err != nil {
- t.Fatal(err)
- }
- if string(output) != "foobar" {
- t.Error(string(output))
- }
- // Run the container again and check the output
- output, err = container.Output()
- if err != nil {
- t.Fatal(err)
- }
- if string(output) != "foobar" {
- t.Error(string(output))
- }
- }
- func TestRestartStdin(t *testing.T) {
- runtime, err := newTestRuntime()
- if err != nil {
- t.Fatal(err)
- }
- defer nuke(runtime)
- container, err := runtime.Create(&Config{
- Image: GetTestImage(runtime).Id,
- Cmd: []string{"cat"},
- OpenStdin: true,
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container)
- stdin, err := container.StdinPipe()
- if err != nil {
- t.Fatal(err)
- }
- stdout, err := container.StdoutPipe()
- if err != nil {
- t.Fatal(err)
- }
- if err := container.Start(); err != nil {
- t.Fatal(err)
- }
- if _, err := io.WriteString(stdin, "hello world"); err != nil {
- t.Fatal(err)
- }
- if err := stdin.Close(); err != nil {
- t.Fatal(err)
- }
- container.Wait()
- output, err := ioutil.ReadAll(stdout)
- if err != nil {
- t.Fatal(err)
- }
- if err := stdout.Close(); err != nil {
- t.Fatal(err)
- }
- if string(output) != "hello world" {
- t.Fatalf("Unexpected output. Expected %s, received: %s", "hello world", string(output))
- }
- // Restart and try again
- stdin, err = container.StdinPipe()
- if err != nil {
- t.Fatal(err)
- }
- stdout, err = container.StdoutPipe()
- if err != nil {
- t.Fatal(err)
- }
- if err := container.Start(); err != nil {
- t.Fatal(err)
- }
- if _, err := io.WriteString(stdin, "hello world #2"); err != nil {
- t.Fatal(err)
- }
- if err := stdin.Close(); err != nil {
- t.Fatal(err)
- }
- container.Wait()
- output, err = ioutil.ReadAll(stdout)
- if err != nil {
- t.Fatal(err)
- }
- if err := stdout.Close(); err != nil {
- t.Fatal(err)
- }
- if string(output) != "hello world #2" {
- t.Fatalf("Unexpected output. Expected %s, received: %s", "hello world #2", string(output))
- }
- }
- func TestUser(t *testing.T) {
- runtime, err := newTestRuntime()
- if err != nil {
- t.Fatal(err)
- }
- defer nuke(runtime)
- // Default user must be root
- container, err := runtime.Create(&Config{
- Image: GetTestImage(runtime).Id,
- Cmd: []string{"id"},
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container)
- output, err := container.Output()
- if err != nil {
- t.Fatal(err)
- }
- if !strings.Contains(string(output), "uid=0(root) gid=0(root)") {
- t.Error(string(output))
- }
- // Set a username
- container, err = runtime.Create(&Config{
- Image: GetTestImage(runtime).Id,
- Cmd: []string{"id"},
- User: "root",
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container)
- output, err = container.Output()
- if err != nil || container.State.ExitCode != 0 {
- t.Fatal(err)
- }
- if !strings.Contains(string(output), "uid=0(root) gid=0(root)") {
- t.Error(string(output))
- }
- // Set a UID
- container, err = runtime.Create(&Config{
- Image: GetTestImage(runtime).Id,
- Cmd: []string{"id"},
- User: "0",
- },
- )
- if err != nil || container.State.ExitCode != 0 {
- t.Fatal(err)
- }
- defer runtime.Destroy(container)
- output, err = container.Output()
- if err != nil || container.State.ExitCode != 0 {
- t.Fatal(err)
- }
- if !strings.Contains(string(output), "uid=0(root) gid=0(root)") {
- t.Error(string(output))
- }
- // Set a different user by uid
- container, err = runtime.Create(&Config{
- Image: GetTestImage(runtime).Id,
- Cmd: []string{"id"},
- User: "1",
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container)
- output, err = container.Output()
- if err != nil {
- t.Fatal(err)
- } else if container.State.ExitCode != 0 {
- t.Fatalf("Container exit code is invalid: %d\nOutput:\n%s\n", container.State.ExitCode, output)
- }
- if !strings.Contains(string(output), "uid=1(daemon) gid=1(daemon)") {
- t.Error(string(output))
- }
- // Set a different user by username
- container, err = runtime.Create(&Config{
- Image: GetTestImage(runtime).Id,
- Cmd: []string{"id"},
- User: "daemon",
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container)
- output, err = container.Output()
- if err != nil || container.State.ExitCode != 0 {
- t.Fatal(err)
- }
- if !strings.Contains(string(output), "uid=1(daemon) gid=1(daemon)") {
- t.Error(string(output))
- }
- }
- func TestMultipleContainers(t *testing.T) {
- runtime, err := newTestRuntime()
- if err != nil {
- t.Fatal(err)
- }
- defer nuke(runtime)
- container1, err := runtime.Create(&Config{
- Image: GetTestImage(runtime).Id,
- Cmd: []string{"cat", "/dev/zero"},
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container1)
- container2, err := runtime.Create(&Config{
- Image: GetTestImage(runtime).Id,
- Cmd: []string{"cat", "/dev/zero"},
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container2)
- // Start both containers
- if err := container1.Start(); err != nil {
- t.Fatal(err)
- }
- if err := container2.Start(); err != nil {
- t.Fatal(err)
- }
- // Make sure they are running before trying to kill them
- container1.WaitTimeout(250 * time.Millisecond)
- container2.WaitTimeout(250 * time.Millisecond)
- // If we are here, both containers should be running
- if !container1.State.Running {
- t.Fatal("Container not running")
- }
- if !container2.State.Running {
- t.Fatal("Container not running")
- }
- // Kill them
- if err := container1.Kill(); err != nil {
- t.Fatal(err)
- }
- if err := container2.Kill(); err != nil {
- t.Fatal(err)
- }
- }
- func TestStdin(t *testing.T) {
- runtime, err := newTestRuntime()
- if err != nil {
- t.Fatal(err)
- }
- defer nuke(runtime)
- container, err := runtime.Create(&Config{
- Image: GetTestImage(runtime).Id,
- Cmd: []string{"cat"},
- OpenStdin: true,
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container)
- stdin, err := container.StdinPipe()
- if err != nil {
- t.Fatal(err)
- }
- stdout, err := container.StdoutPipe()
- if err != nil {
- t.Fatal(err)
- }
- if err := container.Start(); err != nil {
- t.Fatal(err)
- }
- defer stdin.Close()
- defer stdout.Close()
- if _, err := io.WriteString(stdin, "hello world"); err != nil {
- t.Fatal(err)
- }
- if err := stdin.Close(); err != nil {
- t.Fatal(err)
- }
- container.Wait()
- output, err := ioutil.ReadAll(stdout)
- if err != nil {
- t.Fatal(err)
- }
- if string(output) != "hello world" {
- t.Fatalf("Unexpected output. Expected %s, received: %s", "hello world", string(output))
- }
- }
- func TestTty(t *testing.T) {
- runtime, err := newTestRuntime()
- if err != nil {
- t.Fatal(err)
- }
- defer nuke(runtime)
- container, err := runtime.Create(&Config{
- Image: GetTestImage(runtime).Id,
- Cmd: []string{"cat"},
- OpenStdin: true,
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container)
- stdin, err := container.StdinPipe()
- if err != nil {
- t.Fatal(err)
- }
- stdout, err := container.StdoutPipe()
- if err != nil {
- t.Fatal(err)
- }
- if err := container.Start(); err != nil {
- t.Fatal(err)
- }
- defer stdin.Close()
- defer stdout.Close()
- if _, err := io.WriteString(stdin, "hello world"); err != nil {
- t.Fatal(err)
- }
- if err := stdin.Close(); err != nil {
- t.Fatal(err)
- }
- container.Wait()
- output, err := ioutil.ReadAll(stdout)
- if err != nil {
- t.Fatal(err)
- }
- if string(output) != "hello world" {
- t.Fatalf("Unexpected output. Expected %s, received: %s", "hello world", string(output))
- }
- }
- func TestEnv(t *testing.T) {
- runtime, err := newTestRuntime()
- if err != nil {
- t.Fatal(err)
- }
- defer nuke(runtime)
- container, err := runtime.Create(&Config{
- Image: GetTestImage(runtime).Id,
- Cmd: []string{"/usr/bin/env"},
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container)
- stdout, err := container.StdoutPipe()
- if err != nil {
- t.Fatal(err)
- }
- defer stdout.Close()
- if err := container.Start(); err != nil {
- t.Fatal(err)
- }
- container.Wait()
- output, err := ioutil.ReadAll(stdout)
- if err != nil {
- t.Fatal(err)
- }
- actualEnv := strings.Split(string(output), "\n")
- if actualEnv[len(actualEnv)-1] == "" {
- actualEnv = actualEnv[:len(actualEnv)-1]
- }
- sort.Strings(actualEnv)
- goodEnv := []string{
- "PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin",
- "HOME=/",
- }
- sort.Strings(goodEnv)
- if len(goodEnv) != len(actualEnv) {
- t.Fatalf("Wrong environment: should be %d variables, not: '%s'\n", len(goodEnv), strings.Join(actualEnv, ", "))
- }
- for i := range goodEnv {
- if actualEnv[i] != goodEnv[i] {
- t.Fatalf("Wrong environment variable: should be %s, not %s", goodEnv[i], actualEnv[i])
- }
- }
- }
- func grepFile(t *testing.T, path string, pattern string) {
- f, err := os.Open(path)
- if err != nil {
- t.Fatal(err)
- }
- defer f.Close()
- r := bufio.NewReader(f)
- var (
- line string
- )
- err = nil
- for err == nil {
- line, err = r.ReadString('\n')
- if strings.Contains(line, pattern) == true {
- return
- }
- }
- t.Fatalf("grepFile: pattern \"%s\" not found in \"%s\"", pattern, path)
- }
- func TestLXCConfig(t *testing.T) {
- runtime, err := newTestRuntime()
- if err != nil {
- t.Fatal(err)
- }
- defer nuke(runtime)
- // Memory is allocated randomly for testing
- rand.Seed(time.Now().UTC().UnixNano())
- memMin := 33554432
- memMax := 536870912
- mem := memMin + rand.Intn(memMax-memMin)
- container, err := runtime.Create(&Config{
- Image: GetTestImage(runtime).Id,
- Cmd: []string{"/bin/true"},
- Hostname: "foobar",
- Memory: int64(mem),
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container)
- container.generateLXCConfig()
- grepFile(t, container.lxcConfigPath(), "lxc.utsname = foobar")
- grepFile(t, container.lxcConfigPath(),
- fmt.Sprintf("lxc.cgroup.memory.limit_in_bytes = %d", mem))
- grepFile(t, container.lxcConfigPath(),
- fmt.Sprintf("lxc.cgroup.memory.memsw.limit_in_bytes = %d", mem*2))
- }
- func BenchmarkRunSequencial(b *testing.B) {
- runtime, err := newTestRuntime()
- if err != nil {
- b.Fatal(err)
- }
- defer nuke(runtime)
- for i := 0; i < b.N; i++ {
- container, err := runtime.Create(&Config{
- Image: GetTestImage(runtime).Id,
- Cmd: []string{"echo", "-n", "foo"},
- },
- )
- if err != nil {
- b.Fatal(err)
- }
- defer runtime.Destroy(container)
- output, err := container.Output()
- if err != nil {
- b.Fatal(err)
- }
- if string(output) != "foo" {
- b.Fatalf("Unexpected output: %s", output)
- }
- if err := runtime.Destroy(container); err != nil {
- b.Fatal(err)
- }
- }
- }
- func BenchmarkRunParallel(b *testing.B) {
- runtime, err := newTestRuntime()
- if err != nil {
- b.Fatal(err)
- }
- defer nuke(runtime)
- var tasks []chan error
- for i := 0; i < b.N; i++ {
- complete := make(chan error)
- tasks = append(tasks, complete)
- go func(i int, complete chan error) {
- container, err := runtime.Create(&Config{
- Image: GetTestImage(runtime).Id,
- Cmd: []string{"echo", "-n", "foo"},
- },
- )
- if err != nil {
- complete <- err
- return
- }
- defer runtime.Destroy(container)
- if err := container.Start(); err != nil {
- complete <- err
- return
- }
- if err := container.WaitTimeout(15 * time.Second); err != nil {
- complete <- err
- return
- }
- // if string(output) != "foo" {
- // complete <- fmt.Errorf("Unexecpted output: %v", string(output))
- // }
- if err := runtime.Destroy(container); err != nil {
- complete <- err
- return
- }
- complete <- nil
- }(i, complete)
- }
- var errors []error
- for _, task := range tasks {
- err := <-task
- if err != nil {
- errors = append(errors, err)
- }
- }
- if len(errors) > 0 {
- b.Fatal(errors)
- }
- }
|