forked from TrueCloudLab/distribution
6b73a9ab89
It's possible to run into a race condition in which the enumerator lists lots of repositories and then starts the long process of enumerating through them. In that time if someone deletes a repo, the enumerator may error out. Signed-off-by: Ryan Abrams <rdabrams@gmail.com>
47 lines
1 KiB
Go
47 lines
1 KiB
Go
package driver
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
)
|
|
|
|
type changingFileSystem struct {
|
|
StorageDriver
|
|
fileset []string
|
|
keptFiles map[string]bool
|
|
}
|
|
|
|
func (cfs *changingFileSystem) List(ctx context.Context, path string) ([]string, error) {
|
|
return cfs.fileset, nil
|
|
}
|
|
func (cfs *changingFileSystem) Stat(ctx context.Context, path string) (FileInfo, error) {
|
|
kept, ok := cfs.keptFiles[path]
|
|
if ok && kept {
|
|
return &FileInfoInternal{
|
|
FileInfoFields: FileInfoFields{
|
|
Path: path,
|
|
},
|
|
}, nil
|
|
}
|
|
return nil, PathNotFoundError{}
|
|
}
|
|
func TestWalkFileRemoved(t *testing.T) {
|
|
d := &changingFileSystem{
|
|
fileset: []string{"zoidberg", "bender"},
|
|
keptFiles: map[string]bool{
|
|
"zoidberg": true,
|
|
},
|
|
}
|
|
infos := []FileInfo{}
|
|
err := WalkFallback(context.Background(), d, "", func(fileInfo FileInfo) error {
|
|
infos = append(infos, fileInfo)
|
|
return nil
|
|
})
|
|
if len(infos) != 1 || infos[0].Path() != "zoidberg" {
|
|
t.Errorf(fmt.Sprintf("unexpected path set during walk: %s", infos))
|
|
}
|
|
if err != nil {
|
|
t.Fatalf(err.Error())
|
|
}
|
|
}
|