package main
 
 import (
 	"fmt"

 	"io/ioutil"

 	"os"
 	"os/exec"

 	"path/filepath"
 	"reflect"

 	"testing"
 )
 

 // save a repo and try to load it using stdout
 func TestSaveAndLoadRepoStdout(t *testing.T) {

 	runCmd := exec.Command(dockerBinary, "run", "-d", "busybox", "true")
 	out, _, err := runCommandWithOutput(runCmd)
 	errorOut(err, t, fmt.Sprintf("failed to create a container: %v %v", out, err))
 
 	cleanedContainerID := stripTrailingCharacters(out)
 
 	repoName := "foobar-save-load-test"
 
 	inspectCmd := exec.Command(dockerBinary, "inspect", cleanedContainerID)
 	out, _, err = runCommandWithOutput(inspectCmd)
 	errorOut(err, t, fmt.Sprintf("output should've been a container id: %v %v", cleanedContainerID, err))
 
 	commitCmd := exec.Command(dockerBinary, "commit", cleanedContainerID, repoName)
 	out, _, err = runCommandWithOutput(commitCmd)
 	errorOut(err, t, fmt.Sprintf("failed to commit container: %v %v", out, err))
 

 	inspectCmd = exec.Command(dockerBinary, "inspect", repoName)
 	before, _, err := runCommandWithOutput(inspectCmd)
 	errorOut(err, t, fmt.Sprintf("the repo should exist before saving it: %v %v", before, err))
 

 	saveCmdTemplate := `%v save %v > /tmp/foobar-save-load-test.tar`
 	saveCmdFinal := fmt.Sprintf(saveCmdTemplate, dockerBinary, repoName)
 	saveCmd := exec.Command("bash", "-c", saveCmdFinal)
 	out, _, err = runCommandWithOutput(saveCmd)
 	errorOut(err, t, fmt.Sprintf("failed to save repo: %v %v", out, err))
 
 	deleteImages(repoName)
 
 	loadCmdFinal := `cat /tmp/foobar-save-load-test.tar | docker load`
 	loadCmd := exec.Command("bash", "-c", loadCmdFinal)
 	out, _, err = runCommandWithOutput(loadCmd)
 	errorOut(err, t, fmt.Sprintf("failed to load repo: %v %v", out, err))
 
 	inspectCmd = exec.Command(dockerBinary, "inspect", repoName)

 	after, _, err := runCommandWithOutput(inspectCmd)
 	errorOut(err, t, fmt.Sprintf("the repo should exist after loading it: %v %v", after, err))
 
 	if before != after {
 		t.Fatalf("inspect is not the same after a save / load")
 	}

 	deleteContainer(cleanedContainerID)
 	deleteImages(repoName)

 
 	os.Remove("/tmp/foobar-save-load-test.tar")
 

 	logDone("save - save a repo using stdout")
 	logDone("load - load a repo using stdout")
 }
 

 func TestSaveSingleTag(t *testing.T) {
 	repoName := "foobar-save-single-tag-test"
 
 	tagCmdFinal := fmt.Sprintf("%v tag busybox:latest %v:latest", dockerBinary, repoName)
 	tagCmd := exec.Command("bash", "-c", tagCmdFinal)
 	out, _, err := runCommandWithOutput(tagCmd)
 	errorOut(err, t, fmt.Sprintf("failed to tag repo: %v %v", out, err))
 
 	idCmdFinal := fmt.Sprintf("%v images -q --no-trunc %v", dockerBinary, repoName)
 	idCmd := exec.Command("bash", "-c", idCmdFinal)
 	out, _, err = runCommandWithOutput(idCmd)
 	errorOut(err, t, fmt.Sprintf("failed to get repo ID: %v %v", out, err))
 
 	cleanedImageID := stripTrailingCharacters(out)
 
 	saveCmdFinal := fmt.Sprintf("%v save %v:latest | tar t | grep -E '(^repositories$|%v)'", dockerBinary, repoName, cleanedImageID)
 	saveCmd := exec.Command("bash", "-c", saveCmdFinal)
 	out, _, err = runCommandWithOutput(saveCmd)
 	errorOut(err, t, fmt.Sprintf("failed to save repo with image ID and 'repositories' file: %v %v", out, err))
 
 	deleteImages(repoName)
 
 	logDone("save - save a specific image:tag")
 }
 
 func TestSaveImageId(t *testing.T) {
 	repoName := "foobar-save-image-id-test"
 
 	tagCmdFinal := fmt.Sprintf("%v tag scratch:latest %v:latest", dockerBinary, repoName)
 	tagCmd := exec.Command("bash", "-c", tagCmdFinal)
 	out, _, err := runCommandWithOutput(tagCmd)
 	errorOut(err, t, fmt.Sprintf("failed to tag repo: %v %v", out, err))
 
 	idLongCmdFinal := fmt.Sprintf("%v images -q --no-trunc %v", dockerBinary, repoName)
 	idLongCmd := exec.Command("bash", "-c", idLongCmdFinal)
 	out, _, err = runCommandWithOutput(idLongCmd)
 	errorOut(err, t, fmt.Sprintf("failed to get repo ID: %v %v", out, err))
 
 	cleanedLongImageID := stripTrailingCharacters(out)
 
 	idShortCmdFinal := fmt.Sprintf("%v images -q %v", dockerBinary, repoName)
 	idShortCmd := exec.Command("bash", "-c", idShortCmdFinal)
 	out, _, err = runCommandWithOutput(idShortCmd)
 	errorOut(err, t, fmt.Sprintf("failed to get repo short ID: %v %v", out, err))
 
 	cleanedShortImageID := stripTrailingCharacters(out)
 
 	saveCmdFinal := fmt.Sprintf("%v save %v | tar t | grep %v", dockerBinary, cleanedShortImageID, cleanedLongImageID)
 	saveCmd := exec.Command("bash", "-c", saveCmdFinal)
 	out, _, err = runCommandWithOutput(saveCmd)
 	errorOut(err, t, fmt.Sprintf("failed to save repo with image ID: %v %v", out, err))
 
 	deleteImages(repoName)
 
 	logDone("save - save a image by ID")
 }
 

 // save a repo and try to load it using flags
 func TestSaveAndLoadRepoFlags(t *testing.T) {
 	runCmd := exec.Command(dockerBinary, "run", "-d", "busybox", "true")
 	out, _, err := runCommandWithOutput(runCmd)
 	errorOut(err, t, fmt.Sprintf("failed to create a container: %v %v", out, err))
 
 	cleanedContainerID := stripTrailingCharacters(out)
 
 	repoName := "foobar-save-load-test"
 
 	inspectCmd := exec.Command(dockerBinary, "inspect", cleanedContainerID)
 	out, _, err = runCommandWithOutput(inspectCmd)
 	errorOut(err, t, fmt.Sprintf("output should've been a container id: %v %v", cleanedContainerID, err))
 
 	commitCmd := exec.Command(dockerBinary, "commit", cleanedContainerID, repoName)
 	out, _, err = runCommandWithOutput(commitCmd)
 	errorOut(err, t, fmt.Sprintf("failed to commit container: %v %v", out, err))
 
 	inspectCmd = exec.Command(dockerBinary, "inspect", repoName)
 	before, _, err := runCommandWithOutput(inspectCmd)
 	errorOut(err, t, fmt.Sprintf("the repo should exist before saving it: %v %v", before, err))
 
 	saveCmdTemplate := `%v save -o /tmp/foobar-save-load-test.tar %v`
 	saveCmdFinal := fmt.Sprintf(saveCmdTemplate, dockerBinary, repoName)
 	saveCmd := exec.Command("bash", "-c", saveCmdFinal)
 	out, _, err = runCommandWithOutput(saveCmd)
 	errorOut(err, t, fmt.Sprintf("failed to save repo: %v %v", out, err))
 
 	deleteImages(repoName)
 
 	loadCmdFinal := `docker load -i /tmp/foobar-save-load-test.tar`
 	loadCmd := exec.Command("bash", "-c", loadCmdFinal)
 	out, _, err = runCommandWithOutput(loadCmd)
 	errorOut(err, t, fmt.Sprintf("failed to load repo: %v %v", out, err))
 
 	inspectCmd = exec.Command(dockerBinary, "inspect", repoName)
 	after, _, err := runCommandWithOutput(inspectCmd)
 	errorOut(err, t, fmt.Sprintf("the repo should exist after loading it: %v %v", after, err))
 
 	if before != after {
 		t.Fatalf("inspect is not the same after a save / load")
 	}
 
 	deleteContainer(cleanedContainerID)
 	deleteImages(repoName)
 
 	os.Remove("/tmp/foobar-save-load-test.tar")
 
 	logDone("save - save a repo using -o")
 	logDone("load - load a repo using -i")

 }

 
 func TestSaveMultipleNames(t *testing.T) {
 	repoName := "foobar-save-multi-name-test"
 
 	// Make one image
 	tagCmdFinal := fmt.Sprintf("%v tag scratch:latest %v-one:latest", dockerBinary, repoName)
 	tagCmd := exec.Command("bash", "-c", tagCmdFinal)
 	out, _, err := runCommandWithOutput(tagCmd)
 	errorOut(err, t, fmt.Sprintf("failed to tag repo: %v %v", out, err))
 	// Make two images
 	tagCmdFinal = fmt.Sprintf("%v tag scratch:latest %v-two:latest", dockerBinary, repoName)
 	tagCmd = exec.Command("bash", "-c", tagCmdFinal)
 	out, _, err = runCommandWithOutput(tagCmd)
 	errorOut(err, t, fmt.Sprintf("failed to tag repo: %v %v", out, err))
 
 	saveCmdFinal := fmt.Sprintf("%v save %v-one %v-two:latest | tar xO repositories | grep -q -E '(-one|-two)'", dockerBinary, repoName, repoName)
 	saveCmd := exec.Command("bash", "-c", saveCmdFinal)
 	out, _, err = runCommandWithOutput(saveCmd)
 	errorOut(err, t, fmt.Sprintf("failed to save multiple repos: %v %v", out, err))
 
 	deleteImages(repoName)
 
 	logDone("save - save by multiple names")
 }

 
 // Issue #6722 #5892 ensure directories are included in changes
 func TestSaveDirectoryPermissions(t *testing.T) {
 	layerEntries := []string{"opt/", "opt/a/", "opt/a/b/", "opt/a/b/c"}

 	layerEntriesAUFS := []string{"./", ".wh..wh.aufs", ".wh..wh.orph/", ".wh..wh.plnk/", "opt/", "opt/a/", "opt/a/b/", "opt/a/b/c"}

 
 	name := "save-directory-permissions"
 	tmpDir, err := ioutil.TempDir("", "save-layers-with-directories")
 	extractionDirectory := filepath.Join(tmpDir, "image-extraction-dir")
 	os.Mkdir(extractionDirectory, 0777)
 
 	if err != nil {
 		t.Errorf("failed to create temporary directory: %s", err)
 	}
 	defer os.RemoveAll(tmpDir)
 	defer deleteImages(name)
 	_, err = buildImage(name,
 		`FROM busybox
 	RUN adduser -D user && mkdir -p /opt/a/b && chown -R user:user /opt/a
 	RUN touch /opt/a/b/c && chown user:user /opt/a/b/c`,
 		true)
 	if err != nil {
 		t.Fatal(err)
 	}
 
 	saveCmdFinal := fmt.Sprintf("%s save %s | tar -xf - -C %s", dockerBinary, name, extractionDirectory)
 	saveCmd := exec.Command("bash", "-c", saveCmdFinal)
 	out, _, err := runCommandWithOutput(saveCmd)
 	if err != nil {
 		t.Errorf("failed to save and extract image: %s", out)
 	}
 
 	dirs, err := ioutil.ReadDir(extractionDirectory)
 	if err != nil {
 		t.Errorf("failed to get a listing of the layer directories: %s", err)
 	}
 
 	found := false
 	for _, entry := range dirs {
 		if entry.IsDir() {
 			layerPath := filepath.Join(extractionDirectory, entry.Name(), "layer.tar")
 
 			f, err := os.Open(layerPath)
 			if err != nil {
 				t.Fatalf("failed to open %s: %s", layerPath, err)
 			}
 
 			entries, err := ListTar(f)
 			if err != nil {
 				t.Fatalf("encountered error while listing tar entries: %s", err)
 			}
 

 			if reflect.DeepEqual(entries, layerEntries) || reflect.DeepEqual(entries, layerEntriesAUFS) {

 				found = true

 				break

 			}
 		}
 	}
 
 	if !found {
 		t.Fatalf("failed to find the layer with the right content listing")
 	}
 
 	logDone("save - ensure directories exist in exported layers")
 }