mirror of
https://github.com/navidrome/navidrome.git
synced 2024-09-11 20:12:54 +02:00
108 lines
2.4 KiB
Go
108 lines
2.4 KiB
Go
package utils_test
|
|
|
|
import (
|
|
"io"
|
|
"io/fs"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/navidrome/navidrome/utils"
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
var _ = Describe("MergeFS", func() {
|
|
var baseName, overlayName string
|
|
var baseDir, overlayDir, mergedDir fs.FS
|
|
|
|
BeforeEach(func() {
|
|
baseName, _ = os.MkdirTemp("", "merge_fs_base_test")
|
|
overlayName, _ = os.MkdirTemp("", "merge_fs_overlay_test")
|
|
baseDir = os.DirFS(baseName)
|
|
overlayDir = os.DirFS(overlayName)
|
|
mergedDir = utils.MergeFS{Base: baseDir, Overlay: overlayDir}
|
|
})
|
|
|
|
It("reads from Base dir if not found in Overlay", func() {
|
|
_f(baseName, "a.json")
|
|
file, err := mergedDir.Open("a.json")
|
|
Expect(err).To(BeNil())
|
|
|
|
stat, err := file.Stat()
|
|
Expect(err).To(BeNil())
|
|
|
|
Expect(stat.Name()).To(Equal("a.json"))
|
|
})
|
|
|
|
It("reads overridden file", func() {
|
|
_f(baseName, "b.json", "original")
|
|
_f(baseName, "b.json", "overridden")
|
|
|
|
file, err := mergedDir.Open("b.json")
|
|
Expect(err).To(BeNil())
|
|
|
|
content, err := io.ReadAll(file)
|
|
Expect(err).To(BeNil())
|
|
Expect(string(content)).To(Equal("overridden"))
|
|
})
|
|
|
|
It("reads only files from Base if Overlay is empty", func() {
|
|
_f(baseName, "test.txt")
|
|
|
|
dir, err := mergedDir.Open(".")
|
|
Expect(err).To(BeNil())
|
|
|
|
list, err := dir.(fs.ReadDirFile).ReadDir(-1)
|
|
Expect(err).To(BeNil())
|
|
|
|
Expect(list).To(HaveLen(1))
|
|
Expect(list[0].Name()).To(Equal("test.txt"))
|
|
})
|
|
|
|
It("reads merged dirs", func() {
|
|
_f(baseName, "1111.txt")
|
|
_f(overlayName, "2222.json")
|
|
|
|
dir, err := mergedDir.Open(".")
|
|
Expect(err).To(BeNil())
|
|
|
|
list, err := dir.(fs.ReadDirFile).ReadDir(-1)
|
|
Expect(err).To(BeNil())
|
|
|
|
Expect(list).To(HaveLen(2))
|
|
Expect(list[0].Name()).To(Equal("1111.txt"))
|
|
Expect(list[1].Name()).To(Equal("2222.json"))
|
|
})
|
|
|
|
It("allows to seek to the beginning of the directory", func() {
|
|
_f(baseName, "1111")
|
|
_f(baseName, "2222")
|
|
_f(baseName, "3333")
|
|
|
|
dir, err := mergedDir.Open(".")
|
|
Expect(err).To(BeNil())
|
|
|
|
list, _ := dir.(fs.ReadDirFile).ReadDir(2)
|
|
Expect(list).To(HaveLen(2))
|
|
Expect(list[0].Name()).To(Equal("1111"))
|
|
Expect(list[1].Name()).To(Equal("2222"))
|
|
|
|
list, _ = dir.(fs.ReadDirFile).ReadDir(2)
|
|
Expect(list).To(HaveLen(1))
|
|
Expect(list[0].Name()).To(Equal("3333"))
|
|
})
|
|
})
|
|
|
|
func _f(dir, name string, content ...string) string {
|
|
path := filepath.Join(dir, name)
|
|
file, err := os.Create(path)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
if len(content) > 0 {
|
|
_, _ = file.WriteString(content[0])
|
|
}
|
|
_ = file.Close()
|
|
return path
|
|
}
|