433 lines
9.4 KiB
Go
433 lines
9.4 KiB
Go
package docker
|
|
|
|
import (
|
|
"github.com/dotcloud/docker/utils"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestContainerTagImageDelete(t *testing.T) {
|
|
runtime := mkRuntime(t)
|
|
defer nuke(runtime)
|
|
|
|
srv := &Server{runtime: runtime}
|
|
|
|
initialImages, err := srv.Images(false, "")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := srv.runtime.repositories.Set("utest", "tag1", unitTestImageName, false); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := srv.runtime.repositories.Set("utest/docker", "tag2", unitTestImageName, false); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := srv.runtime.repositories.Set("utest:5000/docker", "tag3", unitTestImageName, false); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
images, err := srv.Images(false, "")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if len(images) != len(initialImages)+3 {
|
|
t.Errorf("Expected %d images, %d found", len(initialImages)+2, len(images))
|
|
}
|
|
|
|
if _, err := srv.ImageDelete("utest/docker:tag2", true); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
images, err = srv.Images(false, "")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if len(images) != len(initialImages)+2 {
|
|
t.Errorf("Expected %d images, %d found", len(initialImages)+2, len(images))
|
|
}
|
|
|
|
if _, err := srv.ImageDelete("utest:5000/docker:tag3", true); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
images, err = srv.Images(false, "")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if len(images) != len(initialImages)+1 {
|
|
t.Errorf("Expected %d images, %d found", len(initialImages)+1, len(images))
|
|
}
|
|
|
|
if _, err := srv.ImageDelete("utest:tag1", true); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
images, err = srv.Images(false, "")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if len(images) != len(initialImages) {
|
|
t.Errorf("Expected %d image, %d found", len(initialImages), len(images))
|
|
}
|
|
}
|
|
|
|
func TestCreateRm(t *testing.T) {
|
|
runtime := mkRuntime(t)
|
|
defer nuke(runtime)
|
|
|
|
srv := &Server{runtime: runtime}
|
|
|
|
config, _, _, err := ParseRun([]string{GetTestImage(runtime).ID, "echo test"}, nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
id, err := srv.ContainerCreate(config)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if len(runtime.List()) != 1 {
|
|
t.Errorf("Expected 1 container, %v found", len(runtime.List()))
|
|
}
|
|
|
|
if err = srv.ContainerDestroy(id, true); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if len(runtime.List()) != 0 {
|
|
t.Errorf("Expected 0 container, %v found", len(runtime.List()))
|
|
}
|
|
|
|
}
|
|
|
|
func TestCommit(t *testing.T) {
|
|
runtime := mkRuntime(t)
|
|
defer nuke(runtime)
|
|
|
|
srv := &Server{runtime: runtime}
|
|
|
|
config, _, _, err := ParseRun([]string{GetTestImage(runtime).ID, "/bin/cat"}, nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
id, err := srv.ContainerCreate(config)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if _, err := srv.ContainerCommit(id, "testrepo", "testtag", "", "", config); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestCreateStartRestartStopStartKillRm(t *testing.T) {
|
|
runtime := mkRuntime(t)
|
|
defer nuke(runtime)
|
|
|
|
srv := &Server{runtime: runtime}
|
|
|
|
config, hostConfig, _, err := ParseRun([]string{GetTestImage(runtime).ID, "/bin/cat"}, nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
id, err := srv.ContainerCreate(config)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if len(runtime.List()) != 1 {
|
|
t.Errorf("Expected 1 container, %v found", len(runtime.List()))
|
|
}
|
|
|
|
err = srv.ContainerStart(id, hostConfig)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = srv.ContainerRestart(id, 1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = srv.ContainerStop(id, 1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = srv.ContainerStart(id, hostConfig)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = srv.ContainerKill(id)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// FIXME: this failed once with a race condition ("Unable to remove filesystem for xxx: directory not empty")
|
|
if err = srv.ContainerDestroy(id, true); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if len(runtime.List()) != 0 {
|
|
t.Errorf("Expected 0 container, %v found", len(runtime.List()))
|
|
}
|
|
|
|
}
|
|
|
|
func TestRunWithTooLowMemoryLimit(t *testing.T) {
|
|
var err error
|
|
runtime := mkRuntime(t)
|
|
srv := &Server{runtime: runtime}
|
|
defer nuke(runtime)
|
|
// Try to create a container with a memory limit of 1 byte less than the minimum allowed limit.
|
|
_, err = srv.ContainerCreate(
|
|
&Config{
|
|
Image: GetTestImage(runtime).ID,
|
|
Memory: 524287,
|
|
CpuShares: 1000,
|
|
Cmd: []string{"/bin/cat"},
|
|
},
|
|
)
|
|
if err == nil {
|
|
t.Errorf("Memory limit is smaller than the allowed limit. Container creation should've failed!")
|
|
}
|
|
|
|
}
|
|
|
|
func TestContainerTop(t *testing.T) {
|
|
t.Skip("Fixme. Skipping test for now. Reported error: 'server_test.go:236: Expected 2 processes, found 1.'")
|
|
runtime := mkRuntime(t)
|
|
srv := &Server{runtime: runtime}
|
|
defer nuke(runtime)
|
|
|
|
c, hostConfig, _ := mkContainer(runtime, []string{"_", "/bin/sh", "-c", "sleep 2"}, t)
|
|
c, hostConfig, err := mkContainer(runtime, []string{"_", "/bin/sh", "-c", "sleep 2"}, t)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
defer runtime.Destroy(c)
|
|
if err := c.Start(hostConfig); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Give some time to the process to start
|
|
c.WaitTimeout(500 * time.Millisecond)
|
|
|
|
if !c.State.Running {
|
|
t.Errorf("Container should be running")
|
|
}
|
|
procs, err := srv.ContainerTop(c.ID, "")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if len(procs.Processes) != 2 {
|
|
t.Fatalf("Expected 2 processes, found %d.", len(procs.Processes))
|
|
}
|
|
|
|
pos := -1
|
|
for i := 0; i < len(procs.Titles); i++ {
|
|
if procs.Titles[i] == "CMD" {
|
|
pos = i
|
|
break
|
|
}
|
|
}
|
|
|
|
if pos == -1 {
|
|
t.Fatalf("Expected CMD, not found.")
|
|
}
|
|
|
|
if procs.Processes[0][pos] != "sh" && procs.Processes[0][pos] != "busybox" {
|
|
t.Fatalf("Expected `busybox` or `sh`, found %s.", procs.Processes[0][pos])
|
|
}
|
|
|
|
if procs.Processes[1][pos] != "sh" && procs.Processes[1][pos] != "busybox" {
|
|
t.Fatalf("Expected `busybox` or `sh`, found %s.", procs.Processes[1][pos])
|
|
}
|
|
}
|
|
|
|
func TestPools(t *testing.T) {
|
|
runtime := mkRuntime(t)
|
|
srv := &Server{
|
|
runtime: runtime,
|
|
pullingPool: make(map[string]struct{}),
|
|
pushingPool: make(map[string]struct{}),
|
|
}
|
|
defer nuke(runtime)
|
|
|
|
err := srv.poolAdd("pull", "test1")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
err = srv.poolAdd("pull", "test2")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
err = srv.poolAdd("push", "test1")
|
|
if err == nil || err.Error() != "pull test1 is already in progress" {
|
|
t.Fatalf("Expected `pull test1 is already in progress`")
|
|
}
|
|
err = srv.poolAdd("pull", "test1")
|
|
if err == nil || err.Error() != "pull test1 is already in progress" {
|
|
t.Fatalf("Expected `pull test1 is already in progress`")
|
|
}
|
|
err = srv.poolAdd("wait", "test3")
|
|
if err == nil || err.Error() != "Unknown pool type" {
|
|
t.Fatalf("Expected `Unknown pool type`")
|
|
}
|
|
|
|
err = srv.poolRemove("pull", "test2")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
err = srv.poolRemove("pull", "test2")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
err = srv.poolRemove("pull", "test1")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
err = srv.poolRemove("push", "test1")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
err = srv.poolRemove("wait", "test3")
|
|
if err == nil || err.Error() != "Unknown pool type" {
|
|
t.Fatalf("Expected `Unknown pool type`")
|
|
}
|
|
}
|
|
|
|
func TestLogEvent(t *testing.T) {
|
|
runtime := mkRuntime(t)
|
|
srv := &Server{
|
|
runtime: runtime,
|
|
events: make([]utils.JSONMessage, 0, 64),
|
|
listeners: make(map[string]chan utils.JSONMessage),
|
|
}
|
|
|
|
srv.LogEvent("fakeaction", "fakeid", "fakeimage")
|
|
|
|
listener := make(chan utils.JSONMessage)
|
|
srv.Lock()
|
|
srv.listeners["test"] = listener
|
|
srv.Unlock()
|
|
|
|
srv.LogEvent("fakeaction2", "fakeid", "fakeimage")
|
|
|
|
if len(srv.events) != 2 {
|
|
t.Fatalf("Expected 2 events, found %d", len(srv.events))
|
|
}
|
|
go func() {
|
|
time.Sleep(200 * time.Millisecond)
|
|
srv.LogEvent("fakeaction3", "fakeid", "fakeimage")
|
|
time.Sleep(200 * time.Millisecond)
|
|
srv.LogEvent("fakeaction4", "fakeid", "fakeimage")
|
|
}()
|
|
|
|
setTimeout(t, "Listening for events timed out", 2*time.Second, func() {
|
|
for i := 2; i < 4; i++ {
|
|
event := <-listener
|
|
if event != srv.events[i] {
|
|
t.Fatalf("Event received it different than expected")
|
|
}
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestRmi(t *testing.T) {
|
|
runtime := mkRuntime(t)
|
|
defer nuke(runtime)
|
|
srv := &Server{runtime: runtime}
|
|
|
|
initialImages, err := srv.Images(false, "")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
config, hostConfig, _, err := ParseRun([]string{GetTestImage(runtime).ID, "echo test"}, nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
containerID, err := srv.ContainerCreate(config)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
//To remove
|
|
err = srv.ContainerStart(containerID, hostConfig)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
imageID, err := srv.ContainerCommit(containerID, "test", "", "", "", nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = srv.ContainerTag(imageID, "test", "0.1", false)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
containerID, err = srv.ContainerCreate(config)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
//To remove
|
|
err = srv.ContainerStart(containerID, hostConfig)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
_, err = srv.ContainerCommit(containerID, "test", "", "", "", nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
images, err := srv.Images(false, "")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if len(images)-len(initialImages) != 2 {
|
|
t.Fatalf("Expected 2 new images, found %d.", len(images)-len(initialImages))
|
|
}
|
|
|
|
_, err = srv.ImageDelete(imageID, true)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
images, err = srv.Images(false, "")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if len(images)-len(initialImages) != 1 {
|
|
t.Fatalf("Expected 1 new image, found %d.", len(images)-len(initialImages))
|
|
}
|
|
|
|
for _, image := range images {
|
|
if strings.Contains(unitTestImageID, image.ID) {
|
|
continue
|
|
}
|
|
if image.Repository == "" {
|
|
t.Fatalf("Expected tagged image, got untagged one.")
|
|
}
|
|
}
|
|
}
|