123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499 |
- package docker
- import (
- "fmt"
- "github.com/dotcloud/docker/utils"
- "io"
- "io/ioutil"
- "log"
- "net"
- "os"
- "strconv"
- "strings"
- "sync"
- "syscall"
- "testing"
- "time"
- )
- const (
- unitTestImageName = "docker-unit-tests"
- unitTestImageId = "e9aa60c60128cad1"
- unitTestStoreBase = "/var/lib/docker/unit-tests"
- testDaemonAddr = "127.0.0.1:4270"
- testDaemonProto = "tcp"
- )
- var globalRuntime *Runtime
- func nuke(runtime *Runtime) error {
- var wg sync.WaitGroup
- for _, container := range runtime.List() {
- wg.Add(1)
- go func(c *Container) {
- c.Kill()
- wg.Done()
- }(container)
- }
- wg.Wait()
- return os.RemoveAll(runtime.root)
- }
- func cleanup(runtime *Runtime) error {
- for _, container := range runtime.List() {
- container.Kill()
- runtime.Destroy(container)
- }
- images, err := runtime.graph.All()
- if err != nil {
- return err
- }
- for _, image := range images {
- if image.ID != unitTestImageId {
- runtime.graph.Delete(image.ID)
- }
- }
- return nil
- }
- func layerArchive(tarfile string) (io.Reader, error) {
- // FIXME: need to close f somewhere
- f, err := os.Open(tarfile)
- if err != nil {
- return nil, err
- }
- return f, nil
- }
- func init() {
- // Hack to run sys init during unit testing
- if utils.SelfPath() == "/sbin/init" {
- SysInit()
- return
- }
- if uid := syscall.Geteuid(); uid != 0 {
- log.Fatal("docker tests needs to be run as root")
- }
- NetworkBridgeIface = "testdockbr0"
- // Make it our Store root
- runtime, err := NewRuntimeFromDirectory(unitTestStoreBase, false)
- if err != nil {
- panic(err)
- }
- globalRuntime = runtime
- // Create the "Server"
- srv := &Server{
- runtime: runtime,
- enableCors: false,
- pullingPool: make(map[string]struct{}),
- pushingPool: make(map[string]struct{}),
- }
- // Retrieve the Image
- if err := srv.ImagePull(unitTestImageName, "", "", os.Stdout, utils.NewStreamFormatter(false), nil); err != nil {
- panic(err)
- }
- // Spawn a Daemon
- go func() {
- if err := ListenAndServe(testDaemonProto, testDaemonAddr, srv, os.Getenv("DEBUG") != ""); err != nil {
- panic(err)
- }
- }()
- // Give some time to ListenAndServer to actually start
- time.Sleep(time.Second)
- }
- // FIXME: test that ImagePull(json=true) send correct json output
- func newTestRuntime() (*Runtime, error) {
- root, err := ioutil.TempDir("", "docker-test")
- if err != nil {
- return nil, err
- }
- if err := os.Remove(root); err != nil {
- return nil, err
- }
- if err := utils.CopyDirectory(unitTestStoreBase, root); err != nil {
- return nil, err
- }
- runtime, err := NewRuntimeFromDirectory(root, false)
- if err != nil {
- return nil, err
- }
- runtime.UpdateCapabilities(true)
- return runtime, nil
- }
- func GetTestImage(runtime *Runtime) *Image {
- imgs, err := runtime.graph.All()
- if err != nil {
- panic(err)
- }
- for i := range imgs {
- if imgs[i].ID == unitTestImageId {
- return imgs[i]
- }
- }
- panic(fmt.Errorf("Test image %v not found", unitTestImageId))
- }
- func TestRuntimeCreate(t *testing.T) {
- runtime, err := newTestRuntime()
- if err != nil {
- t.Fatal(err)
- }
- defer nuke(runtime)
- // Make sure we start we 0 containers
- if len(runtime.List()) != 0 {
- t.Errorf("Expected 0 containers, %v found", len(runtime.List()))
- }
- builder := NewBuilder(runtime)
- container, err := builder.Create(&Config{
- Image: GetTestImage(runtime).ID,
- Cmd: []string{"ls", "-al"},
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer func() {
- if err := runtime.Destroy(container); err != nil {
- t.Error(err)
- }
- }()
- // Make sure we can find the newly created container with List()
- if len(runtime.List()) != 1 {
- t.Errorf("Expected 1 container, %v found", len(runtime.List()))
- }
- // Make sure the container List() returns is the right one
- if runtime.List()[0].ID != container.ID {
- t.Errorf("Unexpected container %v returned by List", runtime.List()[0])
- }
- // Make sure we can get the container with Get()
- if runtime.Get(container.ID) == nil {
- t.Errorf("Unable to get newly created container")
- }
- // Make sure it is the right container
- if runtime.Get(container.ID) != container {
- t.Errorf("Get() returned the wrong container")
- }
- // Make sure Exists returns it as existing
- if !runtime.Exists(container.ID) {
- t.Errorf("Exists() returned false for a newly created container")
- }
- // Make sure crete with bad parameters returns an error
- _, err = builder.Create(
- &Config{
- Image: GetTestImage(runtime).ID,
- },
- )
- if err == nil {
- t.Fatal("Builder.Create should throw an error when Cmd is missing")
- }
- _, err = builder.Create(
- &Config{
- Image: GetTestImage(runtime).ID,
- Cmd: []string{},
- },
- )
- if err == nil {
- t.Fatal("Builder.Create should throw an error when Cmd is empty")
- }
- }
- func TestDestroy(t *testing.T) {
- runtime, err := newTestRuntime()
- if err != nil {
- t.Fatal(err)
- }
- defer nuke(runtime)
- container, err := NewBuilder(runtime).Create(&Config{
- Image: GetTestImage(runtime).ID,
- Cmd: []string{"ls", "-al"},
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- // Destroy
- if err := runtime.Destroy(container); err != nil {
- t.Error(err)
- }
- // Make sure runtime.Exists() behaves correctly
- if runtime.Exists("test_destroy") {
- t.Errorf("Exists() returned true")
- }
- // Make sure runtime.List() doesn't list the destroyed container
- if len(runtime.List()) != 0 {
- t.Errorf("Expected 0 container, %v found", len(runtime.List()))
- }
- // Make sure runtime.Get() refuses to return the unexisting container
- if runtime.Get(container.ID) != nil {
- t.Errorf("Unable to get newly created container")
- }
- // Make sure the container root directory does not exist anymore
- _, err = os.Stat(container.root)
- if err == nil || !os.IsNotExist(err) {
- t.Errorf("Container root directory still exists after destroy")
- }
- // Test double destroy
- if err := runtime.Destroy(container); err == nil {
- // It should have failed
- t.Errorf("Double destroy did not fail")
- }
- }
- func TestGet(t *testing.T) {
- runtime, err := newTestRuntime()
- if err != nil {
- t.Fatal(err)
- }
- defer nuke(runtime)
- builder := NewBuilder(runtime)
- container1, err := builder.Create(&Config{
- Image: GetTestImage(runtime).ID,
- Cmd: []string{"ls", "-al"},
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container1)
- container2, err := builder.Create(&Config{
- Image: GetTestImage(runtime).ID,
- Cmd: []string{"ls", "-al"},
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container2)
- container3, err := builder.Create(&Config{
- Image: GetTestImage(runtime).ID,
- Cmd: []string{"ls", "-al"},
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime.Destroy(container3)
- if runtime.Get(container1.ID) != container1 {
- t.Errorf("Get(test1) returned %v while expecting %v", runtime.Get(container1.ID), container1)
- }
- if runtime.Get(container2.ID) != container2 {
- t.Errorf("Get(test2) returned %v while expecting %v", runtime.Get(container2.ID), container2)
- }
- if runtime.Get(container3.ID) != container3 {
- t.Errorf("Get(test3) returned %v while expecting %v", runtime.Get(container3.ID), container3)
- }
- }
- func findAvailalblePort(runtime *Runtime, port int) (*Container, error) {
- strPort := strconv.Itoa(port)
- container, err := NewBuilder(runtime).Create(&Config{
- Image: GetTestImage(runtime).ID,
- Cmd: []string{"sh", "-c", "echo well hello there | nc -l -p " + strPort},
- PortSpecs: []string{strPort},
- },
- )
- if err != nil {
- return nil, err
- }
- hostConfig := &HostConfig{}
- if err := container.Start(hostConfig); err != nil {
- if strings.Contains(err.Error(), "address already in use") {
- return nil, nil
- }
- return nil, err
- }
- return container, nil
- }
- // Run a container with a TCP port allocated, and test that it can receive connections on localhost
- func TestAllocatePortLocalhost(t *testing.T) {
- runtime, err := newTestRuntime()
- if err != nil {
- t.Fatal(err)
- }
- port := 5554
- var container *Container
- for {
- port += 1
- log.Println("Trying port", port)
- t.Log("Trying port", port)
- container, err = findAvailalblePort(runtime, port)
- if container != nil {
- break
- }
- if err != nil {
- t.Fatal(err)
- }
- log.Println("Port", port, "already in use")
- t.Log("Port", port, "already in use")
- }
- defer container.Kill()
- 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)
- conn, err := net.Dial("tcp",
- fmt.Sprintf(
- "localhost:%s", container.NetworkSettings.PortMapping[strconv.Itoa(port)],
- ),
- )
- if err != nil {
- t.Fatal(err)
- }
- defer conn.Close()
- output, err := ioutil.ReadAll(conn)
- if err != nil {
- t.Fatal(err)
- }
- if string(output) != "well hello there\n" {
- t.Fatalf("Received wrong output from network connection: should be '%s', not '%s'",
- "well hello there\n",
- string(output),
- )
- }
- container.Wait()
- }
- func TestRestore(t *testing.T) {
- root, err := ioutil.TempDir("", "docker-test")
- if err != nil {
- t.Fatal(err)
- }
- if err := os.Remove(root); err != nil {
- t.Fatal(err)
- }
- if err := utils.CopyDirectory(unitTestStoreBase, root); err != nil {
- t.Fatal(err)
- }
- runtime1, err := NewRuntimeFromDirectory(root, false)
- if err != nil {
- t.Fatal(err)
- }
- builder := NewBuilder(runtime1)
- // Create a container with one instance of docker
- container1, err := builder.Create(&Config{
- Image: GetTestImage(runtime1).ID,
- Cmd: []string{"ls", "-al"},
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime1.Destroy(container1)
- // Create a second container meant to be killed
- container2, err := builder.Create(&Config{
- Image: GetTestImage(runtime1).ID,
- Cmd: []string{"/bin/cat"},
- OpenStdin: true,
- },
- )
- if err != nil {
- t.Fatal(err)
- }
- defer runtime1.Destroy(container2)
- // Start the container non blocking
- hostConfig := &HostConfig{}
- if err := container2.Start(hostConfig); err != nil {
- t.Fatal(err)
- }
- if !container2.State.Running {
- t.Fatalf("Container %v should appear as running but isn't", container2.ID)
- }
- // Simulate a crash/manual quit of dockerd: process dies, states stays 'Running'
- cStdin, _ := container2.StdinPipe()
- cStdin.Close()
- if err := container2.WaitTimeout(2 * time.Second); err != nil {
- t.Fatal(err)
- }
- container2.State.Running = true
- container2.ToDisk()
- if len(runtime1.List()) != 2 {
- t.Errorf("Expected 2 container, %v found", len(runtime1.List()))
- }
- if err := container1.Run(); err != nil {
- t.Fatal(err)
- }
- if !container2.State.Running {
- t.Fatalf("Container %v should appear as running but isn't", container2.ID)
- }
- // Here are are simulating a docker restart - that is, reloading all containers
- // from scratch
- runtime2, err := NewRuntimeFromDirectory(root, false)
- if err != nil {
- t.Fatal(err)
- }
- defer nuke(runtime2)
- if len(runtime2.List()) != 2 {
- t.Errorf("Expected 2 container, %v found", len(runtime2.List()))
- }
- runningCount := 0
- for _, c := range runtime2.List() {
- if c.State.Running {
- t.Errorf("Running container found: %v (%v)", c.ID, c.Path)
- runningCount++
- }
- }
- if runningCount != 0 {
- t.Fatalf("Expected 0 container alive, %d found", runningCount)
- }
- container3 := runtime2.Get(container1.ID)
- if container3 == nil {
- t.Fatal("Unable to Get container")
- }
- if err := container3.Run(); err != nil {
- t.Fatal(err)
- }
- container2.State.Running = false
- }
|