Use fs.FS in MergeFS implementation

This commit is contained in:
Deluan
2021-07-20 19:50:00 -04:00
parent 7540881695
commit 774ad65155
4 changed files with 68 additions and 81 deletions
+14 -18
View File
@@ -1,9 +1,8 @@
package utils_test
import (
"io"
"io/fs"
"io/ioutil"
"net/http"
"os"
"path/filepath"
@@ -12,19 +11,19 @@ import (
. "github.com/onsi/gomega"
)
var _ = Describe("mergeFS", func() {
var _ = Describe("MergeFS", func() {
var baseName, overlayName string
var baseDir, overlayDir, mergedDir http.FileSystem
var baseDir, overlayDir, mergedDir fs.FS
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)
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() {
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())
@@ -47,13 +46,13 @@ var _ = Describe("mergeFS", func() {
Expect(string(content)).To(Equal("overridden"))
})
It("reads only files from base if overlay is empty", func() {
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)
list, err := dir.(fs.ReadDirFile).ReadDir(-1)
Expect(err).To(BeNil())
Expect(list).To(HaveLen(1))
@@ -67,7 +66,7 @@ var _ = Describe("mergeFS", func() {
dir, err := mergedDir.Open(".")
Expect(err).To(BeNil())
list, err := dir.Readdir(-1)
list, err := dir.(fs.ReadDirFile).ReadDir(-1)
Expect(err).To(BeNil())
Expect(list).To(HaveLen(2))
@@ -83,17 +82,14 @@ var _ = Describe("mergeFS", func() {
dir, err := mergedDir.Open(".")
Expect(err).To(BeNil())
list, _ := dir.Readdir(2)
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"))
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"))
list, _ = dir.(fs.ReadDirFile).ReadDir(2)
Expect(list).To(HaveLen(1))
Expect(list[0].Name()).To(Equal("3333"))
})
})