mirror of https://github.com/octoleo/restic.git
130 lines
2.9 KiB
Go
130 lines
2.9 KiB
Go
package dump
|
|
|
|
import (
|
|
"archive/zip"
|
|
"bytes"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/restic/restic/internal/fs"
|
|
)
|
|
|
|
func TestWriteZip(t *testing.T) {
|
|
WriteTest(t, WriteZip, checkZip)
|
|
}
|
|
|
|
func readZipFile(f *zip.File) ([]byte, error) {
|
|
rc, err := f.Open()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
b := &bytes.Buffer{}
|
|
_, err = b.ReadFrom(rc)
|
|
if err != nil {
|
|
// ignore subsequent errors
|
|
_ = rc.Close()
|
|
return nil, err
|
|
}
|
|
|
|
err = rc.Close()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return b.Bytes(), nil
|
|
}
|
|
|
|
func checkZip(t *testing.T, testDir string, srcZip *bytes.Buffer) error {
|
|
z, err := zip.NewReader(bytes.NewReader(srcZip.Bytes()), int64(srcZip.Len()))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fileNumber := 0
|
|
zipFiles := len(z.File)
|
|
|
|
err = filepath.Walk(testDir, func(path string, info os.FileInfo, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if info.Name() != filepath.Base(testDir) {
|
|
fileNumber++
|
|
}
|
|
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, f := range z.File {
|
|
matchPath := filepath.Join(testDir, f.Name)
|
|
match, err := os.Lstat(matchPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// check metadata, zip header contains time rounded to seconds
|
|
fileTime := match.ModTime().Truncate(time.Second)
|
|
zipTime := f.Modified
|
|
if !fileTime.Equal(zipTime) {
|
|
return fmt.Errorf("modTime does not match, got: %s, want: %s", zipTime, fileTime)
|
|
}
|
|
if f.Mode() != match.Mode() {
|
|
return fmt.Errorf("mode does not match, got: %v [%08x], want: %v [%08x]",
|
|
f.Mode(), uint32(f.Mode()), match.Mode(), uint32(match.Mode()))
|
|
}
|
|
t.Logf("Mode is %v [%08x] for %s", f.Mode(), uint32(f.Mode()), f.Name)
|
|
|
|
switch {
|
|
case f.FileInfo().IsDir():
|
|
filebase := filepath.ToSlash(match.Name())
|
|
if filepath.Base(f.Name) != filebase {
|
|
return fmt.Errorf("foldernames don't match got %v want %v", filepath.Base(f.Name), filebase)
|
|
}
|
|
if !strings.HasSuffix(f.Name, "/") {
|
|
return fmt.Errorf("foldernames must end with separator got %v", f.Name)
|
|
}
|
|
case f.Mode()&os.ModeSymlink != 0:
|
|
target, err := fs.Readlink(matchPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
linkName, err := readZipFile(f)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if target != string(linkName) {
|
|
return fmt.Errorf("symlink target does not match, got %s want %s", string(linkName), target)
|
|
}
|
|
default:
|
|
if uint64(match.Size()) != f.UncompressedSize64 {
|
|
return fmt.Errorf("size does not match got %v want %v", f.UncompressedSize64, match.Size())
|
|
}
|
|
contentsFile, err := ioutil.ReadFile(matchPath)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
contentsZip, err := readZipFile(f)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if string(contentsZip) != string(contentsFile) {
|
|
return fmt.Errorf("contents does not match, got %s want %s", contentsZip, contentsFile)
|
|
}
|
|
}
|
|
}
|
|
|
|
if zipFiles != fileNumber {
|
|
return fmt.Errorf("not the same amount of files got %v want %v", zipFiles, fileNumber)
|
|
}
|
|
|
|
return nil
|
|
}
|