12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292 |
- package docker
- import (
- "bufio"
- "fmt"
- "io"
- "io/ioutil"
- "math/rand"
- "os"
- "path"
- "regexp"
- "sort"
- "strings"
- "testing"
- "time"
- )
- func TestIDFormat(t *testing.T) {
- runtime := mkRuntime(t)
- defer nuke(runtime)
- container1, err := NewBuilder(runtime).Create(
- &Config{
- Image: GetTestImage(runtime).ID,
- Cmd: []string{"/bin/sh", "-c", "echo hello world"},
- },
- )
- 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 := mkRuntime(t)
- defer nuke(runtime)
- container, hostConfig, _ := mkContainer(
- runtime,
- []string{"_", "/bin/sh", "-c", "i=1; while [ $i -le 5 ]; do i=`expr $i + 1`; echo hello; done"},
- t,
- )
- 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(hostConfig); 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(10); 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(hostConfig); err != nil {
- t.Fatal(err)
- }
- setTimeout(t, "Timeout reading from the process", 3*time.Second, 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)
- }
- })
- container.Wait()
- }
- func TestDiff(t *testing.T) {
- runtime := mkRuntime(t)
- defer nuke(runtime)
- // Create a container and remove a file
- container1, _, _ := mkContainer(runtime, []string{"_", "/bin/rm", "/etc/passwd"}, t)
- defer runtime.Destroy(container1)
- if err := container1.Run(); err != nil {
- t.Fatal(err)
- }
- // Check the changelog
- c, err := container1.Changes()
- if err != nil {
- t.Fatal(err)
- }
- success := false
- for _, elem := range c {
- if elem.Path == "/etc/passwd" && elem.Kind == 2 {
- success = true
- }
- }
- if !success {
- t.Fatalf("/etc/passwd as been removed but is not present in the diff")
- }
- // Commit the container
- rwTar, err := container1.ExportRw()
- if err != nil {
- t.Error(err)
- }
- img, err := runtime.graph.Create(rwTar, container1, "unit test commited image - diff", "", nil)
- if err != nil {
- t.Error(err)
- }
- // Create a new container from the commited image
- container2, _, _ := mkContainer(runtime, []string{img.ID, "cat", "/etc/passwd"}, t)
- defer runtime.Destroy(container2)
- if err := container2.Run(); err != nil {
- t.Fatal(err)
- }
- // Check the changelog
- c, err = container2.Changes()
- if err != nil {
- t.Fatal(err)
- }
- for _, elem := range c {
- if elem.Path == "/etc/passwd" {
- t.Fatalf("/etc/passwd should not be present in the diff after commit.")
- }
- }
- // Create a new containere
- container3, _, _ := mkContainer(runtime, []string{"_", "rm", "/bin/httpd"}, t)
- defer runtime.Destroy(container3)
- if err := container3.Run(); err != nil {
- t.Fatal(err)
- }
- // Check the changelog
- c, err = container3.Changes()
- if err != nil {
- t.Fatal(err)
- }
- success = false
- for _, elem := range c {
- if elem.Path == "/bin/httpd" && elem.Kind == 2 {
- success = true
- }
- }
- if !success {
- t.Fatalf("/bin/httpd should be present in the diff after commit.")
- }
- }
- func TestCommitAutoRun(t *testing.T) {
- runtime := mkRuntime(t)
- defer nuke(runtime)
- container1, _, _ := mkContainer(runtime, []string{"_", "/bin/sh", "-c", "echo hello > /world"}, t)
- 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", "", &Config{Cmd: []string{"cat", "/world"}})
- if err != nil {
- t.Error(err)
- }
- // FIXME: Make a TestCommit that stops here and check docker.root/layers/img.id/world
- container2, hostConfig, _ := mkContainer(runtime, []string{img.ID}, t)
- 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(hostConfig); 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 TestCommitRun(t *testing.T) {
- runtime := mkRuntime(t)
- defer nuke(runtime)
- container1, hostConfig, _ := mkContainer(runtime, []string{"_", "/bin/sh", "-c", "echo hello > /world"}, t)
- 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", "", nil)
- if err != nil {
- t.Error(err)
- }
- // FIXME: Make a TestCommit that stops here and check docker.root/layers/img.id/world
- container2, hostConfig, _ := mkContainer(runtime, []string{img.ID, "cat", "/world"}, t)
- 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(hostConfig); 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 := mkRuntime(t)
- defer nuke(runtime)
- container, hostConfig, _ := mkContainer(runtime, []string{"-m", "33554432", "-c", "1000", "-i", "_", "/bin/cat"}, t)
- defer runtime.Destroy(container)
- cStdin, err := container.StdinPipe()
- if err != nil {
- t.Fatal(err)
- }
- if err := container.Start(hostConfig); 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(hostConfig); 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.Close()
- container.WaitTimeout(2 * time.Second)
- }
- func TestRun(t *testing.T) {
- runtime := mkRuntime(t)
- defer nuke(runtime)
- container, _, _ := mkContainer(runtime, []string{"_", "ls", "-al"}, t)
- 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 := mkRuntime(t)
- defer nuke(runtime)
- container, err := NewBuilder(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 := mkRuntime(t)
- defer nuke(runtime)
- container, err := NewBuilder(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")
- }
- hostConfig := &HostConfig{}
- if err := container.Start(hostConfig); err != nil {
- t.Fatal(err)
- }
- setTimeout(t, "Waiting for the container to be started timed out", 2*time.Second, func() {
- for !container.State.Running {
- time.Sleep(10 * time.Millisecond)
- }
- })
- // Even if the state is running, lets give some time to lxc to spawn the process
- container.WaitTimeout(500 * time.Millisecond)
- 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)
- }
- }
- // Test that creating a container with a volume doesn't crash. Regression test for #995.
- func TestCreateVolume(t *testing.T) {
- runtime := mkRuntime(t)
- defer nuke(runtime)
- config, hc, _, err := ParseRun([]string{"-v", "/var/lib/data", GetTestImage(runtime).ID, "echo", "hello", "world"}, nil)
- if err != nil {
- t.Fatal(err)
- }
- c, err := NewBuilder(runtime).Create(config)
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(c)
- if err := c.Start(hc); err != nil {
- t.Fatal(err)
- }
- c.WaitTimeout(500 * time.Millisecond)
- c.Wait()
- }
- func TestKill(t *testing.T) {
- runtime := mkRuntime(t)
- defer nuke(runtime)
- container, err := NewBuilder(runtime).Create(&Config{
- Image: GetTestImage(runtime).ID,
- Cmd: []string{"sleep", "2"},
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container)
- if container.State.Running {
- t.Errorf("Container shouldn't be running")
- }
- hostConfig := &HostConfig{}
- if err := container.Start(hostConfig); 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 := mkRuntime(t)
- defer nuke(runtime)
- builder := NewBuilder(runtime)
- trueContainer, err := builder.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 := builder.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 := mkRuntime(t)
- defer nuke(runtime)
- container, err := NewBuilder(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 := mkRuntime(t)
- defer nuke(runtime)
- container, err := NewBuilder(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)
- }
- hostConfig := &HostConfig{}
- if err := container.Start(hostConfig); 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(hostConfig); 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 := mkRuntime(t)
- defer nuke(runtime)
- builder := NewBuilder(runtime)
- // Default user must be root
- container, err := builder.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 = builder.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 = builder.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 = builder.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 = builder.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))
- }
- // Test an wrong username
- container, err = builder.Create(&Config{
- Image: GetTestImage(runtime).ID,
- Cmd: []string{"id"},
- User: "unkownuser",
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container)
- output, err = container.Output()
- if container.State.ExitCode == 0 {
- t.Fatal("Starting container with wrong uid should fail but it passed.")
- }
- }
- func TestMultipleContainers(t *testing.T) {
- runtime := mkRuntime(t)
- defer nuke(runtime)
- builder := NewBuilder(runtime)
- container1, err := builder.Create(&Config{
- Image: GetTestImage(runtime).ID,
- Cmd: []string{"sleep", "2"},
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container1)
- container2, err := builder.Create(&Config{
- Image: GetTestImage(runtime).ID,
- Cmd: []string{"sleep", "2"},
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container2)
- // Start both containers
- hostConfig := &HostConfig{}
- if err := container1.Start(hostConfig); err != nil {
- t.Fatal(err)
- }
- if err := container2.Start(hostConfig); 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 := mkRuntime(t)
- defer nuke(runtime)
- container, err := NewBuilder(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)
- }
- hostConfig := &HostConfig{}
- if err := container.Start(hostConfig); 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 := mkRuntime(t)
- defer nuke(runtime)
- container, err := NewBuilder(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)
- }
- hostConfig := &HostConfig{}
- if err := container.Start(hostConfig); 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 := mkRuntime(t)
- defer nuke(runtime)
- container, err := NewBuilder(runtime).Create(&Config{
- Image: GetTestImage(runtime).ID,
- Cmd: []string{"env"},
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container)
- stdout, err := container.StdoutPipe()
- if err != nil {
- t.Fatal(err)
- }
- defer stdout.Close()
- hostConfig := &HostConfig{}
- if err := container.Start(hostConfig); 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=/",
- "container=lxc",
- }
- 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 TestEntrypoint(t *testing.T) {
- runtime := mkRuntime(t)
- defer nuke(runtime)
- container, err := NewBuilder(runtime).Create(
- &Config{
- Image: GetTestImage(runtime).ID,
- Entrypoint: []string{"/bin/echo"},
- Cmd: []string{"-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 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 := mkRuntime(t)
- 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)
- // CPU shares as well
- cpuMin := 100
- cpuMax := 10000
- cpu := cpuMin + rand.Intn(cpuMax-cpuMin)
- container, err := NewBuilder(runtime).Create(&Config{
- Image: GetTestImage(runtime).ID,
- Cmd: []string{"/bin/true"},
- Hostname: "foobar",
- Memory: int64(mem),
- CpuShares: int64(cpu),
- },
- )
- 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 := mkRuntime(b)
- defer nuke(runtime)
- for i := 0; i < b.N; i++ {
- container, err := NewBuilder(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 := mkRuntime(b)
- 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 := NewBuilder(runtime).Create(&Config{
- Image: GetTestImage(runtime).ID,
- Cmd: []string{"echo", "-n", "foo"},
- },
- )
- if err != nil {
- complete <- err
- return
- }
- defer runtime.Destroy(container)
- hostConfig := &HostConfig{}
- if err := container.Start(hostConfig); 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)
- }
- }
- func tempDir(t *testing.T) string {
- tmpDir, err := ioutil.TempDir("", "docker-test")
- if err != nil {
- t.Fatal(err)
- }
- return tmpDir
- }
- func TestBindMounts(t *testing.T) {
- r := mkRuntime(t)
- defer nuke(r)
- tmpDir := tempDir(t)
- defer os.RemoveAll(tmpDir)
- writeFile(path.Join(tmpDir, "touch-me"), "", t)
- // Test reading from a read-only bind mount
- stdout, _ := runContainer(r, []string{"-v", fmt.Sprintf("%s:/tmp:ro", tmpDir), "_", "ls", "/tmp"}, t)
- if !strings.Contains(stdout, "touch-me") {
- t.Fatal("Container failed to read from bind mount")
- }
- // test writing to bind mount
- runContainer(r, []string{"-v", fmt.Sprintf("%s:/tmp:rw", tmpDir), "_", "touch", "/tmp/holla"}, t)
- readFile(path.Join(tmpDir, "holla"), t) // Will fail if the file doesn't exist
- // test mounting to an illegal destination directory
- if _, err := runContainer(r, []string{"-v", fmt.Sprintf("%s:.", tmpDir), "ls", "."}, nil); err == nil {
- t.Fatal("Container bind mounted illegal directory")
- }
- }
- // Test that VolumesRW values are copied to the new container. Regression test for #1201
- func TestVolumesFromReadonlyMount(t *testing.T) {
- runtime := mkRuntime(t)
- defer nuke(runtime)
- container, err := NewBuilder(runtime).Create(
- &Config{
- Image: GetTestImage(runtime).ID,
- Cmd: []string{"/bin/echo", "-n", "foobar"},
- Volumes: map[string]struct{}{"/test": {}},
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container)
- _, err = container.Output()
- if err != nil {
- t.Fatal(err)
- }
- if !container.VolumesRW["/test"] {
- t.Fail()
- }
- container2, err := NewBuilder(runtime).Create(
- &Config{
- Image: GetTestImage(runtime).ID,
- Cmd: []string{"/bin/echo", "-n", "foobar"},
- VolumesFrom: container.ID,
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container2)
- _, err = container2.Output()
- if err != nil {
- t.Fatal(err)
- }
- if container.Volumes["/test"] != container2.Volumes["/test"] {
- t.Fail()
- }
- actual, exists := container2.VolumesRW["/test"]
- if !exists {
- t.Fail()
- }
- if container.VolumesRW["/test"] != actual {
- t.Fail()
- }
- }
- // Test that restarting a container with a volume does not create a new volume on restart. Regression test for #819.
- func TestRestartWithVolumes(t *testing.T) {
- runtime := mkRuntime(t)
- defer nuke(runtime)
- container, err := NewBuilder(runtime).Create(&Config{
- Image: GetTestImage(runtime).ID,
- Cmd: []string{"echo", "-n", "foobar"},
- Volumes: map[string]struct{}{"/test": {}},
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container)
- for key := range container.Config.Volumes {
- if key != "/test" {
- t.Fail()
- }
- }
- _, err = container.Output()
- if err != nil {
- t.Fatal(err)
- }
- expected := container.Volumes["/test"]
- if expected == "" {
- t.Fail()
- }
- // Run the container again to verify the volume path persists
- _, err = container.Output()
- if err != nil {
- t.Fatal(err)
- }
- actual := container.Volumes["/test"]
- if expected != actual {
- t.Fatalf("Expected volume path: %s Actual path: %s", expected, actual)
- }
- }
- func TestOnlyLoopbackExistsWhenUsingDisableNetworkOption(t *testing.T) {
- runtime := mkRuntime(t)
- defer nuke(runtime)
- config, hc, _, err := ParseRun([]string{"-n=false", GetTestImage(runtime).ID, "ip", "addr", "show"}, nil)
- if err != nil {
- t.Fatal(err)
- }
- c, err := NewBuilder(runtime).Create(config)
- if err != nil {
- t.Fatal(err)
- }
- stdout, err := c.StdoutPipe()
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(c)
- if err := c.Start(hc); err != nil {
- t.Fatal(err)
- }
- c.WaitTimeout(500 * time.Millisecond)
- c.Wait()
- output, err := ioutil.ReadAll(stdout)
- if err != nil {
- t.Fatal(err)
- }
- interfaces := regexp.MustCompile(`(?m)^[0-9]+: [a-zA-Z0-9]+`).FindAllString(string(output), -1)
- if len(interfaces) != 1 {
- t.Fatalf("Wrong interface count in test container: expected [1: lo], got [%s]", interfaces)
- }
- if interfaces[0] != "1: lo" {
- t.Fatalf("Wrong interface in test container: expected [1: lo], got [%s]", interfaces)
- }
- }
|