Fix Cached flag

This commit is contained in:
Deluan 2020-07-24 18:46:08 -04:00
parent 58b816c2ed
commit f8f16d676d
2 changed files with 5 additions and 1 deletions

View File

@ -105,7 +105,7 @@ func (fc *fileCache) Get(ctx context.Context, arg fmt.Stringer) (*CachedStream,
}
// All other cases, just return a Reader, without Seek capabilities
return &CachedStream{Reader: r, Cached: true}, nil
return &CachedStream{Reader: r, Cached: cached}, nil
}
func (fc *fileCache) Ready() bool {

View File

@ -60,6 +60,7 @@ var _ = Describe("File Caches", func() {
// First call is a MISS
s, err := fc.Get(context.TODO(), &testArg{"test"})
Expect(err).To(BeNil())
Expect(s.Cached).To(BeFalse())
Expect(ioutil.ReadAll(s)).To(Equal([]byte("test")))
// Second call is a HIT
@ -67,6 +68,7 @@ var _ = Describe("File Caches", func() {
s, err = fc.Get(context.TODO(), &testArg{"test"})
Expect(err).To(BeNil())
Expect(ioutil.ReadAll(s)).To(Equal([]byte("test")))
Expect(s.Cached).To(BeTrue())
Expect(called).To(BeFalse())
})
@ -79,6 +81,7 @@ var _ = Describe("File Caches", func() {
// First call is a MISS
s, err := fc.Get(context.TODO(), &testArg{"test"})
Expect(err).To(BeNil())
Expect(s.Cached).To(BeFalse())
Expect(ioutil.ReadAll(s)).To(Equal([]byte("test")))
// Second call is also a MISS
@ -86,6 +89,7 @@ var _ = Describe("File Caches", func() {
s, err = fc.Get(context.TODO(), &testArg{"test"})
Expect(err).To(BeNil())
Expect(ioutil.ReadAll(s)).To(Equal([]byte("test")))
Expect(s.Cached).To(BeFalse())
Expect(called).To(BeTrue())
})
})