integration-cli/docker_cli_import_test.go

   1 package main
   2
   3 import (
   4         "bufio"
   5         "compress/gzip"
   6         "io/ioutil"
   7         "os"
   8         "os/exec"
   9         "regexp"
  10         "strings"
  11
  12         "github.com/docker/docker/integration-cli/checker"
  13         "github.com/docker/docker/integration-cli/cli"
  14         "github.com/docker/docker/pkg/testutil"
  15         icmd "github.com/docker/docker/pkg/testutil/cmd"
  16         "github.com/go-check/check"
  17 )
  18
  19 func (s *DockerSuite) TestImportDisplay(c *check.C) {
  20         testRequires(c, DaemonIsLinux)
  21         out, _ := dockerCmd(c, "run", "-d", "busybox", "true")
  22         cleanedContainerID := strings.TrimSpace(out)
  23
  24         out, _, err := testutil.RunCommandPipelineWithOutput(
  25                 exec.Command(dockerBinary, "export", cleanedContainerID),
  26                 exec.Command(dockerBinary, "import", "-"),
  27         )
  28         c.Assert(err, checker.IsNil)
  29
  30         c.Assert(out, checker.Count, "\n", 1, check.Commentf("display is expected 1 '\\n' but didn't"))
  31
  32         image := strings.TrimSpace(out)
  33         out, _ = dockerCmd(c, "run", "--rm", image, "true")
  34         c.Assert(out, checker.Equals, "", check.Commentf("command output should've been nothing."))
  35 }
  36
  37 func (s *DockerSuite) TestImportBadURL(c *check.C) {
  38         out, _, err := dockerCmdWithError("import", "http://nourl/bad")
  39         c.Assert(err, checker.NotNil, check.Commentf("import was supposed to fail but didn't"))
  40         // Depending on your system you can get either of these errors
  41         if !strings.Contains(out, "dial tcp") &&
  42                 !strings.Contains(out, "ApplyLayer exit status 1 stdout:  stderr: archive/tar: invalid tar header") &&
  43                 !strings.Contains(out, "Error processing tar file") {
  44                 c.Fatalf("expected an error msg but didn't get one.\nErr: %v\nOut: %v", err, out)
  45         }
  46 }
  47
  48 func (s *DockerSuite) TestImportFile(c *check.C) {
  49         testRequires(c, DaemonIsLinux)
  50         dockerCmd(c, "run", "--name", "test-import", "busybox", "true")
  51
  52         temporaryFile, err := ioutil.TempFile("", "exportImportTest")
  53         c.Assert(err, checker.IsNil, check.Commentf("failed to create temporary file"))
  54         defer os.Remove(temporaryFile.Name())
  55
  56         icmd.RunCmd(icmd.Cmd{
  57                 Command: []string{dockerBinary, "export", "test-import"},
  58                 Stdout:  bufio.NewWriter(temporaryFile),
  59         }).Assert(c, icmd.Success)
  60
  61         out, _ := dockerCmd(c, "import", temporaryFile.Name())
  62         c.Assert(out, checker.Count, "\n", 1, check.Commentf("display is expected 1 '\\n' but didn't"))
  63         image := strings.TrimSpace(out)
  64
  65         out, _ = dockerCmd(c, "run", "--rm", image, "true")
  66         c.Assert(out, checker.Equals, "", check.Commentf("command output should've been nothing."))
  67 }
  68
  69 func (s *DockerSuite) TestImportGzipped(c *check.C) {
  70         testRequires(c, DaemonIsLinux)
  71         dockerCmd(c, "run", "--name", "test-import", "busybox", "true")
  72
  73         temporaryFile, err := ioutil.TempFile("", "exportImportTest")
  74         c.Assert(err, checker.IsNil, check.Commentf("failed to create temporary file"))
  75         defer os.Remove(temporaryFile.Name())
  76
  77         w := gzip.NewWriter(temporaryFile)
  78         icmd.RunCmd(icmd.Cmd{
  79                 Command: []string{dockerBinary, "export", "test-import"},
  80                 Stdout:  w,
  81         }).Assert(c, icmd.Success)
  82         c.Assert(w.Close(), checker.IsNil, check.Commentf("failed to close gzip writer"))
  83         temporaryFile.Close()
  84         out, _ := dockerCmd(c, "import", temporaryFile.Name())
  85         c.Assert(out, checker.Count, "\n", 1, check.Commentf("display is expected 1 '\\n' but didn't"))
  86         image := strings.TrimSpace(out)
  87
  88         out, _ = dockerCmd(c, "run", "--rm", image, "true")
  89         c.Assert(out, checker.Equals, "", check.Commentf("command output should've been nothing."))
  90 }
  91
  92 func (s *DockerSuite) TestImportFileWithMessage(c *check.C) {
  93         testRequires(c, DaemonIsLinux)
  94         dockerCmd(c, "run", "--name", "test-import", "busybox", "true")
  95
  96         temporaryFile, err := ioutil.TempFile("", "exportImportTest")
  97         c.Assert(err, checker.IsNil, check.Commentf("failed to create temporary file"))
  98         defer os.Remove(temporaryFile.Name())
  99
 100         icmd.RunCmd(icmd.Cmd{
 101                 Command: []string{dockerBinary, "export", "test-import"},
 102                 Stdout:  bufio.NewWriter(temporaryFile),
 103         }).Assert(c, icmd.Success)
 104
 105         message := "Testing commit message"
 106         out, _ := dockerCmd(c, "import", "-m", message, temporaryFile.Name())
 107         c.Assert(out, checker.Count, "\n", 1, check.Commentf("display is expected 1 '\\n' but didn't"))
 108         image := strings.TrimSpace(out)
 109
 110         out, _ = dockerCmd(c, "history", image)
 111         split := strings.Split(out, "\n")
 112
 113         c.Assert(split, checker.HasLen, 3, check.Commentf("expected 3 lines from image history"))
 114         r := regexp.MustCompile("[\\s]{2,}")
 115         split = r.Split(split[1], -1)
 116
 117         c.Assert(message, checker.Equals, split[3], check.Commentf("didn't get expected value in commit message"))
 118
 119         out, _ = dockerCmd(c, "run", "--rm", image, "true")
 120         c.Assert(out, checker.Equals, "", check.Commentf("command output should've been nothing"))
 121 }
 122
 123 func (s *DockerSuite) TestImportFileNonExistentFile(c *check.C) {
 124         _, _, err := dockerCmdWithError("import", "example.com/myImage.tar")
 125         c.Assert(err, checker.NotNil, check.Commentf("import non-existing file must failed"))
 126 }
 127
 128 func (s *DockerSuite) TestImportWithQuotedChanges(c *check.C) {
 129         testRequires(c, DaemonIsLinux)
 130         cli.DockerCmd(c, "run", "--name", "test-import", "busybox", "true")
 131
 132         temporaryFile, err := ioutil.TempFile("", "exportImportTest")
 133         c.Assert(err, checker.IsNil, check.Commentf("failed to create temporary file"))
 134         defer os.Remove(temporaryFile.Name())
 135
 136         cli.Docker(cli.Args("export", "test-import"), cli.WithStdout(bufio.NewWriter(temporaryFile))).Assert(c, icmd.Success)
 137
 138         result := cli.DockerCmd(c, "import", "-c", `ENTRYPOINT ["/bin/sh", "-c"]`, temporaryFile.Name())
 139         image := strings.TrimSpace(result.Stdout())
 140
 141         result = cli.DockerCmd(c, "run", "--rm", image, "true")
 142         c.Assert(result, icmd.Matches, icmd.Expected{Out: icmd.None})
 143 }