| package main |
| |
| import ( |
| "bufio" |
| "compress/gzip" |
| "context" |
| "os" |
| "os/exec" |
| "regexp" |
| "strings" |
| "testing" |
| |
| "github.com/docker/docker/integration-cli/cli" |
| "gotest.tools/v3/assert" |
| "gotest.tools/v3/icmd" |
| ) |
| |
| type DockerCLIImportSuite struct { |
| ds *DockerSuite |
| } |
| |
| func (s *DockerCLIImportSuite) TearDownTest(ctx context.Context, c *testing.T) { |
| s.ds.TearDownTest(ctx, c) |
| } |
| |
| func (s *DockerCLIImportSuite) OnTimeout(c *testing.T) { |
| s.ds.OnTimeout(c) |
| } |
| |
| func (s *DockerCLIImportSuite) TestImportDisplay(c *testing.T) { |
| testRequires(c, DaemonIsLinux) |
| cID := cli.DockerCmd(c, "run", "-d", "busybox", "true").Stdout() |
| cID = strings.TrimSpace(cID) |
| |
| out, err := RunCommandPipelineWithOutput( |
| exec.Command(dockerBinary, "export", cID), |
| exec.Command(dockerBinary, "import", "-"), |
| ) |
| assert.NilError(c, err) |
| |
| assert.Assert(c, strings.Count(out, "\n") == 1, "display is expected 1 '\\n' but didn't") |
| |
| imgRef := strings.TrimSpace(out) |
| out = cli.DockerCmd(c, "run", "--rm", imgRef, "true").Combined() |
| assert.Equal(c, out, "", "command output should've been nothing.") |
| } |
| |
| func (s *DockerCLIImportSuite) TestImportBadURL(c *testing.T) { |
| out, _, err := dockerCmdWithError("import", "https://nosuchdomain.invalid/bad") |
| assert.Assert(c, err != nil, "import was supposed to fail but didn't") |
| // Depending on your system you can get either of these errors |
| if !strings.Contains(out, "dial tcp") && |
| !strings.Contains(out, "ApplyLayer exit status 1 stdout: stderr: archive/tar: invalid tar header") && |
| !strings.Contains(out, "Error processing tar file") { |
| c.Fatalf("expected an error msg but didn't get one.\nErr: %v\nOut: %v", err, out) |
| } |
| } |
| |
| func (s *DockerCLIImportSuite) TestImportFile(c *testing.T) { |
| testRequires(c, DaemonIsLinux) |
| cli.DockerCmd(c, "run", "--name", "test-import", "busybox", "true") |
| |
| temporaryFile, err := os.CreateTemp("", "exportImportTest") |
| assert.Assert(c, err == nil, "failed to create temporary file") |
| defer os.Remove(temporaryFile.Name()) |
| |
| icmd.RunCmd(icmd.Cmd{ |
| Command: []string{dockerBinary, "export", "test-import"}, |
| Stdout: bufio.NewWriter(temporaryFile), |
| }).Assert(c, icmd.Success) |
| |
| out := cli.DockerCmd(c, "import", temporaryFile.Name()).Combined() |
| assert.Assert(c, strings.Count(out, "\n") == 1, "display is expected 1 '\\n' but didn't") |
| imgRef := strings.TrimSpace(out) |
| |
| out = cli.DockerCmd(c, "run", "--rm", imgRef, "true").Combined() |
| assert.Equal(c, out, "", "command output should've been nothing.") |
| } |
| |
| func (s *DockerCLIImportSuite) TestImportGzipped(c *testing.T) { |
| testRequires(c, DaemonIsLinux) |
| cli.DockerCmd(c, "run", "--name", "test-import", "busybox", "true") |
| |
| temporaryFile, err := os.CreateTemp("", "exportImportTest") |
| assert.Assert(c, err == nil, "failed to create temporary file") |
| defer os.Remove(temporaryFile.Name()) |
| |
| w := gzip.NewWriter(temporaryFile) |
| icmd.RunCmd(icmd.Cmd{ |
| Command: []string{dockerBinary, "export", "test-import"}, |
| Stdout: w, |
| }).Assert(c, icmd.Success) |
| assert.Assert(c, w.Close() == nil, "failed to close gzip writer") |
| temporaryFile.Close() |
| out := cli.DockerCmd(c, "import", temporaryFile.Name()).Combined() |
| assert.Assert(c, strings.Count(out, "\n") == 1, "display is expected 1 '\\n' but didn't") |
| imgRef := strings.TrimSpace(out) |
| |
| out = cli.DockerCmd(c, "run", "--rm", imgRef, "true").Combined() |
| assert.Equal(c, out, "", "command output should've been nothing.") |
| } |
| |
| func (s *DockerCLIImportSuite) TestImportFileWithMessage(c *testing.T) { |
| testRequires(c, DaemonIsLinux) |
| cli.DockerCmd(c, "run", "--name", "test-import", "busybox", "true") |
| |
| temporaryFile, err := os.CreateTemp("", "exportImportTest") |
| assert.Assert(c, err == nil, "failed to create temporary file") |
| defer os.Remove(temporaryFile.Name()) |
| |
| icmd.RunCmd(icmd.Cmd{ |
| Command: []string{dockerBinary, "export", "test-import"}, |
| Stdout: bufio.NewWriter(temporaryFile), |
| }).Assert(c, icmd.Success) |
| |
| message := "Testing commit message" |
| out := cli.DockerCmd(c, "import", "-m", message, temporaryFile.Name()).Combined() |
| assert.Assert(c, strings.Count(out, "\n") == 1, "display is expected 1 '\\n' but didn't") |
| imgRef := strings.TrimSpace(out) |
| |
| out = cli.DockerCmd(c, "history", imgRef).Combined() |
| split := strings.Split(out, "\n") |
| |
| assert.Equal(c, len(split), 3, "expected 3 lines from image history") |
| r := regexp.MustCompile(`[\s]{2,}`) |
| split = r.Split(split[1], -1) |
| |
| assert.Equal(c, message, split[3], "didn't get expected value in commit message") |
| |
| out = cli.DockerCmd(c, "run", "--rm", imgRef, "true").Combined() |
| assert.Equal(c, out, "", "command output should've been nothing") |
| } |
| |
| func (s *DockerCLIImportSuite) TestImportFileNonExistentFile(c *testing.T) { |
| _, _, err := dockerCmdWithError("import", "example.com/myImage.tar") |
| assert.Assert(c, err != nil, "import non-existing file must failed") |
| } |
| |
| func (s *DockerCLIImportSuite) TestImportWithQuotedChanges(c *testing.T) { |
| testRequires(c, DaemonIsLinux) |
| cli.DockerCmd(c, "run", "--name", "test-import", "busybox", "true") |
| |
| temporaryFile, err := os.CreateTemp("", "exportImportTest") |
| assert.Assert(c, err == nil, "failed to create temporary file") |
| defer os.Remove(temporaryFile.Name()) |
| |
| cli.Docker(cli.Args("export", "test-import"), cli.WithStdout(bufio.NewWriter(temporaryFile))).Assert(c, icmd.Success) |
| |
| result := cli.DockerCmd(c, "import", "-c", `ENTRYPOINT ["/bin/sh", "-c"]`, temporaryFile.Name()) |
| imgRef := strings.TrimSpace(result.Stdout()) |
| |
| result = cli.DockerCmd(c, "run", "--rm", imgRef, "true") |
| result.Assert(c, icmd.Expected{Out: icmd.None}) |
| } |