mirror of
https://github.com/golang/go
synced 2024-10-01 15:58:33 -06:00
180 lines
4.5 KiB
Go
180 lines
4.5 KiB
Go
|
// Copyright 2015 The Go Authors. All rights reserved.
|
||
|
// Use of this source code is governed by a BSD-style
|
||
|
// license that can be found in the LICENSE file.package zipfs
|
||
|
package zipfs
|
||
|
|
||
|
import (
|
||
|
"archive/zip"
|
||
|
"bytes"
|
||
|
"fmt"
|
||
|
"io"
|
||
|
"io/ioutil"
|
||
|
"os"
|
||
|
"reflect"
|
||
|
"testing"
|
||
|
|
||
|
"golang.org/x/tools/godoc/vfs"
|
||
|
)
|
||
|
|
||
|
var (
|
||
|
|
||
|
// files to use to build zip used by zipfs in testing; maps path : contents
|
||
|
files = map[string]string{"foo": "foo", "bar/baz": "baz", "a/b/c": "c"}
|
||
|
|
||
|
// expected info for each entry in a file system described by files
|
||
|
tests = []struct {
|
||
|
Path string
|
||
|
IsDir bool
|
||
|
IsRegular bool
|
||
|
Name string
|
||
|
Contents string
|
||
|
Files map[string]bool
|
||
|
}{
|
||
|
{"/", true, false, "", "", map[string]bool{"foo": true, "bar": true, "a": true}},
|
||
|
{"//", true, false, "", "", map[string]bool{"foo": true, "bar": true, "a": true}},
|
||
|
{"/foo", false, true, "foo", "foo", nil},
|
||
|
{"/foo/", false, true, "foo", "foo", nil},
|
||
|
{"/foo//", false, true, "foo", "foo", nil},
|
||
|
{"/bar", true, false, "bar", "", map[string]bool{"baz": true}},
|
||
|
{"/bar/", true, false, "bar", "", map[string]bool{"baz": true}},
|
||
|
{"/bar/baz", false, true, "baz", "baz", nil},
|
||
|
{"//bar//baz", false, true, "baz", "baz", nil},
|
||
|
{"/a/b", true, false, "b", "", map[string]bool{"c": true}},
|
||
|
}
|
||
|
|
||
|
// to be initialized in setup()
|
||
|
fs vfs.FileSystem
|
||
|
statFuncs []statFunc
|
||
|
)
|
||
|
|
||
|
type statFunc struct {
|
||
|
Name string
|
||
|
Func func(string) (os.FileInfo, error)
|
||
|
}
|
||
|
|
||
|
func TestMain(t *testing.M) {
|
||
|
if err := setup(); err != nil {
|
||
|
fmt.Fprintf(os.Stderr, "Error setting up zipfs testing state: %v.\n", err)
|
||
|
os.Exit(1)
|
||
|
}
|
||
|
os.Exit(t.Run())
|
||
|
}
|
||
|
|
||
|
// setups state each of the tests uses
|
||
|
func setup() error {
|
||
|
// create zipfs
|
||
|
b := new(bytes.Buffer)
|
||
|
zw := zip.NewWriter(b)
|
||
|
for file, contents := range files {
|
||
|
w, err := zw.Create(file)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
_, err = io.WriteString(w, contents)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
}
|
||
|
zw.Close()
|
||
|
zr, err := zip.NewReader(bytes.NewReader(b.Bytes()), int64(b.Len()))
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
rc := &zip.ReadCloser{
|
||
|
Reader: *zr,
|
||
|
}
|
||
|
fs = New(rc, "foo")
|
||
|
|
||
|
// pull out different stat functions
|
||
|
statFuncs = []statFunc{
|
||
|
{"Stat", fs.Stat},
|
||
|
{"Lstat", fs.Lstat},
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func TestZipFSReadDir(t *testing.T) {
|
||
|
for _, test := range tests {
|
||
|
if test.IsDir {
|
||
|
infos, err := fs.ReadDir(test.Path)
|
||
|
if err != nil {
|
||
|
t.Errorf("Failed to read directory %v\n", test.Path)
|
||
|
continue
|
||
|
}
|
||
|
got := make(map[string]bool)
|
||
|
for _, info := range infos {
|
||
|
got[info.Name()] = true
|
||
|
}
|
||
|
if want := test.Files; !reflect.DeepEqual(got, want) {
|
||
|
t.Errorf("ReadDir %v got %v\nwanted %v\n", test.Path, got, want)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestZipFSStatFuncs(t *testing.T) {
|
||
|
for _, test := range tests {
|
||
|
for _, statFunc := range statFuncs {
|
||
|
|
||
|
// test can stat
|
||
|
info, err := statFunc.Func(test.Path)
|
||
|
if err != nil {
|
||
|
t.Errorf("Unexpected error using %v for %v: %v\n", statFunc.Name, test.Path, err)
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
// test info.Name()
|
||
|
if got, want := info.Name(), test.Name; got != want {
|
||
|
t.Errorf("Using %v for %v info.Name() got %v wanted %v\n", statFunc.Name, test.Path, got, want)
|
||
|
}
|
||
|
// test info.IsDir()
|
||
|
if got, want := info.IsDir(), test.IsDir; got != want {
|
||
|
t.Errorf("Using %v for %v info.IsDir() got %v wanted %v\n", statFunc.Name, test.Path, got, want)
|
||
|
}
|
||
|
// test info.Mode().IsDir()
|
||
|
if got, want := info.Mode().IsDir(), test.IsDir; got != want {
|
||
|
t.Errorf("Using %v for %v info.Mode().IsDir() got %v wanted %v\n", statFunc.Name, test.Path, got, want)
|
||
|
}
|
||
|
// test info.Mode().IsRegular()
|
||
|
if got, want := info.Mode().IsRegular(), test.IsRegular; got != want {
|
||
|
t.Errorf("Using %v for %v info.Mode().IsRegular() got %v wanted %v\n", statFunc.Name, test.Path, got, want)
|
||
|
}
|
||
|
// test info.Size()
|
||
|
if test.IsRegular {
|
||
|
if got, want := info.Size(), int64(len(test.Contents)); got != want {
|
||
|
t.Errorf("Using %v for %v inf.Size() got %v wanted %v", statFunc.Name, test.Path, got, want)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestZipFSOpenSeek(t *testing.T) {
|
||
|
for _, test := range tests {
|
||
|
if test.IsRegular {
|
||
|
|
||
|
// test Open()
|
||
|
f, err := fs.Open(test.Path)
|
||
|
if err != nil {
|
||
|
t.Error(err)
|
||
|
return
|
||
|
}
|
||
|
defer f.Close()
|
||
|
|
||
|
// test Seek() multiple times
|
||
|
for i := 0; i < 3; i++ {
|
||
|
all, err := ioutil.ReadAll(f)
|
||
|
if err != nil {
|
||
|
t.Error(err)
|
||
|
return
|
||
|
}
|
||
|
if got, want := string(all), test.Contents; got != want {
|
||
|
t.Errorf("File contents for %v got %v wanted %v\n", test.Path, got, want)
|
||
|
}
|
||
|
f.Seek(0, 0)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|