forked from mirror/afero
Fix UnionFile.Readdir return when c <= 0
Update Readdir to match behavior of stdlib os package. Fixes #197
This commit is contained in:
parent
f4711e4db9
commit
588a75ec4f
|
@ -404,7 +404,7 @@ func TestCacheOnReadFsNotInLayer(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
// #194
|
// #194
|
||||||
func TestUniontFileReaddirEmpty(t *testing.T) {
|
func TestUnionFileReaddirEmpty(t *testing.T) {
|
||||||
osFs := NewOsFs()
|
osFs := NewOsFs()
|
||||||
|
|
||||||
base := NewMemMapFs()
|
base := NewMemMapFs()
|
||||||
|
@ -439,7 +439,41 @@ func TestUniontFileReaddirEmpty(t *testing.T) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestUniontFileReaddirAskForTooMany(t *testing.T) {
|
// #197
|
||||||
|
func TestUnionFileReaddirDuplicateEmpty(t *testing.T) {
|
||||||
|
base := NewMemMapFs()
|
||||||
|
dir, err := TempDir(base, "", "empty-dir")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Overlay shares same empty directory as base
|
||||||
|
overlay := NewMemMapFs()
|
||||||
|
err = overlay.Mkdir(dir, 0700)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
ufs := &CopyOnWriteFs{base: base, layer: overlay}
|
||||||
|
|
||||||
|
f, err := ufs.Open(dir)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
|
||||||
|
names, err := f.Readdirnames(0)
|
||||||
|
|
||||||
|
if err == io.EOF {
|
||||||
|
t.Errorf("unexpected io.EOF error")
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(names) != 0 {
|
||||||
|
t.Fatal("should be empty")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUnionFileReaddirAskForTooMany(t *testing.T) {
|
||||||
base := &MemMapFs{}
|
base := &MemMapFs{}
|
||||||
overlay := &MemMapFs{}
|
overlay := &MemMapFs{}
|
||||||
|
|
||||||
|
|
10
unionFile.go
10
unionFile.go
|
@ -155,8 +155,8 @@ var defaultUnionMergeDirsFn = func(lofi, bofi []os.FileInfo) ([]os.FileInfo, err
|
||||||
}
|
}
|
||||||
|
|
||||||
// Readdir will weave the two directories together and
|
// Readdir will weave the two directories together and
|
||||||
// return a single view of the overlayed directories
|
// return a single view of the overlayed directories.
|
||||||
// At the end of the directory view, the error is io.EOF.
|
// At the end of the directory view, the error is io.EOF if c > 0.
|
||||||
func (f *UnionFile) Readdir(c int) (ofi []os.FileInfo, err error) {
|
func (f *UnionFile) Readdir(c int) (ofi []os.FileInfo, err error) {
|
||||||
var merge DirsMerger = f.Merger
|
var merge DirsMerger = f.Merger
|
||||||
if merge == nil {
|
if merge == nil {
|
||||||
|
@ -187,11 +187,15 @@ func (f *UnionFile) Readdir(c int) (ofi []os.FileInfo, err error) {
|
||||||
f.files = append(f.files, merged...)
|
f.files = append(f.files, merged...)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if c <= 0 && len(f.files) == 0 {
|
||||||
|
return f.files, nil
|
||||||
|
}
|
||||||
|
|
||||||
if f.off >= len(f.files) {
|
if f.off >= len(f.files) {
|
||||||
return nil, io.EOF
|
return nil, io.EOF
|
||||||
}
|
}
|
||||||
|
|
||||||
if c == -1 {
|
if c <= 0 {
|
||||||
return f.files[f.off:], nil
|
return f.files[f.off:], nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue