491 lines
12 KiB
Go
491 lines
12 KiB
Go
/*
|
|
Copyright 2018 Google LLC
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
package snapshot
|
|
|
|
import (
|
|
"archive/tar"
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"sort"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/GoogleContainerTools/kaniko/pkg/config"
|
|
"github.com/GoogleContainerTools/kaniko/pkg/util"
|
|
"github.com/GoogleContainerTools/kaniko/testutil"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
func TestSnapshotFSFileChange(t *testing.T) {
|
|
testDir, snapshotter, cleanup, err := setUpTest()
|
|
testDirWithoutLeadingSlash := strings.TrimLeft(testDir, "/")
|
|
defer cleanup()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
// Make some changes to the filesystem
|
|
newFiles := map[string]string{
|
|
"foo": "newbaz1",
|
|
"bar/bat": "baz",
|
|
}
|
|
if err := testutil.SetupFiles(testDir, newFiles); err != nil {
|
|
t.Fatalf("Error setting up fs: %s", err)
|
|
}
|
|
// Take another snapshot
|
|
tarPath, err := snapshotter.TakeSnapshotFS()
|
|
if err != nil {
|
|
t.Fatalf("Error taking snapshot of fs: %s", err)
|
|
}
|
|
|
|
f, err := os.Open(tarPath)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
// Check contents of the snapshot, make sure contents is equivalent to snapshotFiles
|
|
tr := tar.NewReader(f)
|
|
fooPath := filepath.Join(testDirWithoutLeadingSlash, "foo")
|
|
batPath := filepath.Join(testDirWithoutLeadingSlash, "bar/bat")
|
|
snapshotFiles := map[string]string{
|
|
fooPath: "newbaz1",
|
|
batPath: "baz",
|
|
}
|
|
for _, path := range util.ParentDirectoriesWithoutLeadingSlash(batPath) {
|
|
if path == "/" {
|
|
continue
|
|
}
|
|
snapshotFiles[path+"/"] = ""
|
|
}
|
|
|
|
actualFiles := []string{}
|
|
for {
|
|
hdr, err := tr.Next()
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
|
|
actualFiles = append(actualFiles, hdr.Name)
|
|
|
|
if _, isFile := snapshotFiles[hdr.Name]; !isFile {
|
|
t.Fatalf("File %s unexpectedly in tar", hdr.Name)
|
|
}
|
|
if hdr.Typeflag == tar.TypeDir {
|
|
continue
|
|
}
|
|
contents, _ := ioutil.ReadAll(tr)
|
|
if string(contents) != snapshotFiles[hdr.Name] {
|
|
t.Fatalf("Contents of %s incorrect, expected: %s, actual: %s", hdr.Name, snapshotFiles[hdr.Name], string(contents))
|
|
}
|
|
}
|
|
if len(actualFiles) != len(snapshotFiles) {
|
|
t.Fatalf("Incorrect number of files were added, expected: %d, actual: %d", len(snapshotFiles), len(actualFiles))
|
|
}
|
|
}
|
|
|
|
func TestSnapshotFSIsReproducible(t *testing.T) {
|
|
testDir, snapshotter, cleanup, err := setUpTest()
|
|
defer cleanup()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
// Make some changes to the filesystem
|
|
newFiles := map[string]string{
|
|
"foo": "newbaz1",
|
|
"bar/bat": "baz",
|
|
}
|
|
if err := testutil.SetupFiles(testDir, newFiles); err != nil {
|
|
t.Fatalf("Error setting up fs: %s", err)
|
|
}
|
|
// Take another snapshot
|
|
tarPath, err := snapshotter.TakeSnapshotFS()
|
|
if err != nil {
|
|
t.Fatalf("Error taking snapshot of fs: %s", err)
|
|
}
|
|
|
|
f, err := os.Open(tarPath)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
// Check contents of the snapshot, make sure contents are sorted by name
|
|
tr := tar.NewReader(f)
|
|
var filesInTar []string
|
|
for {
|
|
hdr, err := tr.Next()
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
filesInTar = append(filesInTar, hdr.Name)
|
|
}
|
|
if !sort.StringsAreSorted(filesInTar) {
|
|
t.Fatalf("Expected the file in the tar archive were sorted, actual list was not sorted: %v", filesInTar)
|
|
}
|
|
}
|
|
|
|
func TestSnapshotFSChangePermissions(t *testing.T) {
|
|
testDir, snapshotter, cleanup, err := setUpTest()
|
|
testDirWithoutLeadingSlash := strings.TrimLeft(testDir, "/")
|
|
defer cleanup()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
// Change permissions on a file
|
|
batPath := filepath.Join(testDir, "bar/bat")
|
|
batPathWithoutLeadingSlash := filepath.Join(testDirWithoutLeadingSlash, "bar/bat")
|
|
if err := os.Chmod(batPath, 0600); err != nil {
|
|
t.Fatalf("Error changing permissions on %s: %v", batPath, err)
|
|
}
|
|
// Take another snapshot
|
|
tarPath, err := snapshotter.TakeSnapshotFS()
|
|
if err != nil {
|
|
t.Fatalf("Error taking snapshot of fs: %s", err)
|
|
}
|
|
f, err := os.Open(tarPath)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
// Check contents of the snapshot, make sure contents is equivalent to snapshotFiles
|
|
tr := tar.NewReader(f)
|
|
snapshotFiles := map[string]string{
|
|
batPathWithoutLeadingSlash: "baz2",
|
|
}
|
|
for _, path := range util.ParentDirectoriesWithoutLeadingSlash(batPathWithoutLeadingSlash) {
|
|
if path == "/" {
|
|
continue
|
|
}
|
|
snapshotFiles[path+"/"] = ""
|
|
}
|
|
|
|
foundFiles := []string{}
|
|
for {
|
|
hdr, err := tr.Next()
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
foundFiles = append(foundFiles, hdr.Name)
|
|
if _, isFile := snapshotFiles[hdr.Name]; !isFile {
|
|
t.Fatalf("File %s unexpectedly in tar", hdr.Name)
|
|
}
|
|
if hdr.Typeflag == tar.TypeDir {
|
|
continue
|
|
}
|
|
contents, _ := ioutil.ReadAll(tr)
|
|
if string(contents) != snapshotFiles[hdr.Name] {
|
|
t.Fatalf("Contents of %s incorrect, expected: %s, actual: %s", hdr.Name, snapshotFiles[hdr.Name], string(contents))
|
|
}
|
|
}
|
|
if len(foundFiles) != len(snapshotFiles) {
|
|
t.Logf("expected\n%v\not equal\n%v", foundFiles, snapshotFiles)
|
|
t.Fatalf("Incorrect number of files were added, expected: %d, got: %d",
|
|
len(snapshotFiles),
|
|
len(foundFiles))
|
|
}
|
|
}
|
|
|
|
func TestSnapshotFiles(t *testing.T) {
|
|
testDir, snapshotter, cleanup, err := setUpTest()
|
|
testDirWithoutLeadingSlash := strings.TrimLeft(testDir, "/")
|
|
defer cleanup()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
// Make some changes to the filesystem
|
|
newFiles := map[string]string{
|
|
"foo": "newbaz1",
|
|
}
|
|
if err := testutil.SetupFiles(testDir, newFiles); err != nil {
|
|
t.Fatalf("Error setting up fs: %s", err)
|
|
}
|
|
filesToSnapshot := []string{
|
|
filepath.Join(testDir, "foo"),
|
|
}
|
|
tarPath, err := snapshotter.TakeSnapshot(filesToSnapshot)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer os.Remove(tarPath)
|
|
|
|
expectedFiles := []string{
|
|
filepath.Join(testDirWithoutLeadingSlash, "foo"),
|
|
}
|
|
for _, path := range util.ParentDirectoriesWithoutLeadingSlash(filepath.Join(testDir, "foo")) {
|
|
expectedFiles = append(expectedFiles, strings.TrimRight(path, "/")+"/")
|
|
}
|
|
|
|
f, err := os.Open(tarPath)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
// Check contents of the snapshot, make sure contents is equivalent to snapshotFiles
|
|
tr := tar.NewReader(f)
|
|
var actualFiles []string
|
|
for {
|
|
hdr, err := tr.Next()
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
actualFiles = append(actualFiles, hdr.Name)
|
|
}
|
|
sort.Strings(expectedFiles)
|
|
sort.Strings(actualFiles)
|
|
testutil.CheckErrorAndDeepEqual(t, false, nil, expectedFiles, actualFiles)
|
|
}
|
|
|
|
func TestEmptySnapshotFS(t *testing.T) {
|
|
_, snapshotter, cleanup, err := setUpTest()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer cleanup()
|
|
|
|
// Take snapshot with no changes
|
|
tarPath, err := snapshotter.TakeSnapshotFS()
|
|
if err != nil {
|
|
t.Fatalf("Error taking snapshot of fs: %s", err)
|
|
}
|
|
|
|
f, err := os.Open(tarPath)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
tr := tar.NewReader(f)
|
|
|
|
if _, err := tr.Next(); err != io.EOF {
|
|
t.Fatal("no files expected in tar, found files.")
|
|
}
|
|
}
|
|
|
|
func TestFileWithLinks(t *testing.T) {
|
|
|
|
link := "baz/link"
|
|
tcs := []struct {
|
|
name string
|
|
path string
|
|
linkFileTarget string
|
|
expected []string
|
|
shouldErr bool
|
|
}{
|
|
{
|
|
name: "given path is a symlink that points to a valid target",
|
|
path: link,
|
|
linkFileTarget: "file",
|
|
expected: []string{link, "baz/file"},
|
|
},
|
|
{
|
|
name: "given path is a symlink points to non existing path",
|
|
path: link,
|
|
linkFileTarget: "does-not-exists",
|
|
expected: []string{link},
|
|
},
|
|
{
|
|
name: "given path is a regular file",
|
|
path: "kaniko/file",
|
|
linkFileTarget: "file",
|
|
expected: []string{"kaniko/file"},
|
|
},
|
|
}
|
|
|
|
for _, tt := range tcs {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
testDir, cleanup, err := setUpTestDir()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer cleanup()
|
|
if err := setupSymlink(testDir, link, tt.linkFileTarget); err != nil {
|
|
t.Fatalf("could not set up symlink due to %s", err)
|
|
}
|
|
actual, err := filesWithLinks(filepath.Join(testDir, tt.path))
|
|
if err != nil {
|
|
t.Fatalf("unexpected error %s", err)
|
|
}
|
|
sortAndCompareFilepaths(t, testDir, tt.expected, actual)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestSnasphotPreservesFileOrder(t *testing.T) {
|
|
newFiles := map[string]string{
|
|
"foo": "newbaz1",
|
|
"bar/bat": "baz",
|
|
"bar/qux": "quuz",
|
|
"qux": "quuz",
|
|
"corge": "grault",
|
|
"garply": "waldo",
|
|
"fred": "plugh",
|
|
"xyzzy": "thud",
|
|
}
|
|
|
|
newFileNames := []string{}
|
|
|
|
for fileName := range newFiles {
|
|
newFileNames = append(newFileNames, fileName)
|
|
}
|
|
|
|
filesInTars := [][]string{}
|
|
|
|
for i := 0; i <= 2; i++ {
|
|
testDir, snapshotter, cleanup, err := setUpTest()
|
|
testDirWithoutLeadingSlash := strings.TrimLeft(testDir, "/")
|
|
defer cleanup()
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
// Make some changes to the filesystem
|
|
if err := testutil.SetupFiles(testDir, newFiles); err != nil {
|
|
t.Fatalf("Error setting up fs: %s", err)
|
|
}
|
|
|
|
filesToSnapshot := []string{}
|
|
for _, file := range newFileNames {
|
|
filesToSnapshot = append(filesToSnapshot, filepath.Join(testDir, file))
|
|
}
|
|
|
|
// Take a snapshot
|
|
tarPath, err := snapshotter.TakeSnapshot(filesToSnapshot)
|
|
|
|
if err != nil {
|
|
t.Fatalf("Error taking snapshot of fs: %s", err)
|
|
}
|
|
|
|
f, err := os.Open(tarPath)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
tr := tar.NewReader(f)
|
|
filesInTars = append(filesInTars, []string{})
|
|
for {
|
|
hdr, err := tr.Next()
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
filesInTars[i] = append(filesInTars[i], strings.TrimPrefix(hdr.Name, testDirWithoutLeadingSlash))
|
|
}
|
|
}
|
|
|
|
// Check contents of all snapshots, make sure files appear in consistent order
|
|
for i := 1; i < len(filesInTars); i++ {
|
|
testutil.CheckErrorAndDeepEqual(t, false, nil, filesInTars[0], filesInTars[i])
|
|
}
|
|
}
|
|
|
|
func TestSnapshotOmitsUnameGname(t *testing.T) {
|
|
_, snapshotter, cleanup, err := setUpTest()
|
|
|
|
defer cleanup()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
tarPath, err := snapshotter.TakeSnapshotFS()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
f, err := os.Open(tarPath)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
tr := tar.NewReader(f)
|
|
for {
|
|
hdr, err := tr.Next()
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if hdr.Uname != "" || hdr.Gname != "" {
|
|
t.Fatalf("Expected Uname/Gname for %s to be empty: Uname = '%s', Gname = '%s'", hdr.Name, hdr.Uname, hdr.Gname)
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
func setupSymlink(dir string, link string, target string) error {
|
|
return os.Symlink(target, filepath.Join(dir, link))
|
|
}
|
|
|
|
func sortAndCompareFilepaths(t *testing.T, testDir string, expected []string, actual []string) {
|
|
expectedFullPaths := make([]string, len(expected))
|
|
for i, file := range expected {
|
|
expectedFullPaths[i] = filepath.Join(testDir, file)
|
|
}
|
|
sort.Strings(expectedFullPaths)
|
|
sort.Strings(actual)
|
|
testutil.CheckDeepEqual(t, expectedFullPaths, actual)
|
|
}
|
|
|
|
func setUpTestDir() (string, func(), error) {
|
|
testDir, err := ioutil.TempDir("", "")
|
|
if err != nil {
|
|
return "", nil, errors.Wrap(err, "setting up temp dir")
|
|
}
|
|
files := map[string]string{
|
|
"foo": "baz1",
|
|
"bar/bat": "baz2",
|
|
"kaniko/file": "file",
|
|
"baz/file": "testfile",
|
|
}
|
|
// Set up initial files
|
|
if err := testutil.SetupFiles(testDir, files); err != nil {
|
|
return "", nil, errors.Wrap(err, "setting up file system")
|
|
}
|
|
|
|
cleanup := func() {
|
|
os.RemoveAll(testDir)
|
|
}
|
|
|
|
return testDir, cleanup, nil
|
|
}
|
|
|
|
func setUpTest() (string, *Snapshotter, func(), error) {
|
|
testDir, dirCleanUp, err := setUpTestDir()
|
|
if err != nil {
|
|
return "", nil, nil, err
|
|
}
|
|
snapshotPath, err := ioutil.TempDir("", "")
|
|
if err != nil {
|
|
return "", nil, nil, errors.Wrap(err, "setting up temp dir")
|
|
}
|
|
|
|
snapshotPathPrefix = snapshotPath
|
|
|
|
// Take the initial snapshot
|
|
l := NewLayeredMap(util.Hasher(), util.CacheHasher())
|
|
snapshotter := NewSnapshotter(l, testDir)
|
|
if err := snapshotter.Init(); err != nil {
|
|
return "", nil, nil, errors.Wrap(err, "initializing snapshotter")
|
|
}
|
|
|
|
original := config.KanikoDir
|
|
config.KanikoDir = testDir
|
|
cleanup := func() {
|
|
os.RemoveAll(snapshotPath)
|
|
config.KanikoDir = original
|
|
dirCleanUp()
|
|
}
|
|
|
|
return testDir, snapshotter, cleanup, nil
|
|
}
|