74 lines
2.3 KiB
Go
74 lines
2.3 KiB
Go
|
package filecount
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"io"
|
||
|
|
"os"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
/*
|
||
|
|
The code below is lifted from numerous articles and originates from Andrew Gerrand's 10 things you (probably) don't know about Go.
|
||
|
|
it allows for mocking a filesystem; this allows for consistent testing of this code across platforms (directory sizes reported
|
||
|
|
differently by different platforms, for example), while preserving the rest of the functionality as-is, without modification.
|
||
|
|
*/
|
||
|
|
|
||
|
|
type fileSystem interface {
|
||
|
|
Open(name string) (file, error)
|
||
|
|
Stat(name string) (os.FileInfo, error)
|
||
|
|
}
|
||
|
|
|
||
|
|
type file interface {
|
||
|
|
io.Closer
|
||
|
|
io.Reader
|
||
|
|
io.ReaderAt
|
||
|
|
io.Seeker
|
||
|
|
Stat() (os.FileInfo, error)
|
||
|
|
}
|
||
|
|
|
||
|
|
// osFS implements fileSystem using the local disk
|
||
|
|
type osFS struct{}
|
||
|
|
|
||
|
|
func (osFS) Open(name string) (file, error) { return os.Open(name) }
|
||
|
|
func (osFS) Stat(name string) (os.FileInfo, error) { return os.Stat(name) }
|
||
|
|
|
||
|
|
/*
|
||
|
|
The following are for mocking the filesystem - this allows us to mock Stat() files. This means that we can set file attributes, and know that they
|
||
|
|
will be the same regardless of the platform sitting underneath our tests (directory sizes vary see https://github.com/influxdata/telegraf/issues/6011)
|
||
|
|
|
||
|
|
NOTE: still need the on-disk file structure to mirror this because the 3rd party library ("github.com/karrick/godirwalk") uses its own
|
||
|
|
walk functions, that we cannot mock from here.
|
||
|
|
*/
|
||
|
|
|
||
|
|
type fakeFileSystem struct {
|
||
|
|
files map[string]fakeFileInfo
|
||
|
|
}
|
||
|
|
|
||
|
|
type fakeFileInfo struct {
|
||
|
|
name string
|
||
|
|
size int64
|
||
|
|
filemode uint32
|
||
|
|
modtime time.Time
|
||
|
|
isdir bool
|
||
|
|
sys interface{}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f fakeFileInfo) Name() string { return f.name }
|
||
|
|
func (f fakeFileInfo) Size() int64 { return f.size }
|
||
|
|
func (f fakeFileInfo) Mode() os.FileMode { return os.FileMode(f.filemode) }
|
||
|
|
func (f fakeFileInfo) ModTime() time.Time { return f.modtime }
|
||
|
|
func (f fakeFileInfo) IsDir() bool { return f.isdir }
|
||
|
|
func (f fakeFileInfo) Sys() interface{} { return f.sys }
|
||
|
|
|
||
|
|
func (f fakeFileSystem) Open(name string) (file, error) {
|
||
|
|
return nil, &os.PathError{Op: "Open", Path: name, Err: errors.New("Not implemented by fake filesystem")}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f fakeFileSystem) Stat(name string) (os.FileInfo, error) {
|
||
|
|
if fakeInfo, found := f.files[name]; found {
|
||
|
|
return fakeInfo, nil
|
||
|
|
}
|
||
|
|
return nil, &os.PathError{Op: "Stat", Path: name, Err: errors.New("No such file or directory")}
|
||
|
|
|
||
|
|
}
|