112 lines
2.5 KiB
Go
112 lines
2.5 KiB
Go
package utils_test
|
|
|
|
import (
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"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 http.FileSystem
|
|
|
|
BeforeEach(func() {
|
|
baseName, _ = ioutil.TempDir("", "merge_fs_base_test")
|
|
overlayName, _ = ioutil.TempDir("", "merge_fs_overlay_test")
|
|
baseDir = http.Dir(baseName)
|
|
overlayDir = http.Dir(overlayName)
|
|
mergedDir = utils.NewMergeFS(baseDir, 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 := ioutil.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.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.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.Readdir(2)
|
|
Expect(list).To(HaveLen(2))
|
|
Expect(list[0].Name()).To(Equal("1111"))
|
|
Expect(list[1].Name()).To(Equal("2222"))
|
|
|
|
Expect(dir.Seek(0, io.SeekStart)).To(Equal(int64(0)))
|
|
|
|
list, _ = dir.Readdir(2)
|
|
Expect(list).To(HaveLen(2))
|
|
Expect(list[0].Name()).To(Equal("1111"))
|
|
Expect(list[1].Name()).To(Equal("2222"))
|
|
})
|
|
})
|
|
|
|
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
|
|
}
|