mirror of
https://github.com/dolthub/dolt.git
synced 2026-05-02 19:39:56 -05:00
Dan Willhite
93 lines
2.2 KiB
Go
93 lines
2.2 KiB
Go
package chunks
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"testing"
|
|
|
|
"github.com/attic-labs/noms/Godeps/_workspace/src/github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
func TestReadThroughStoreTestSuite(t *testing.T) {
|
|
suite.Run(t, &ReadThroughStoreTestSuite{})
|
|
}
|
|
|
|
type ReadThroughStoreTestSuite struct {
|
|
ChunkStoreTestSuite
|
|
}
|
|
|
|
func (suite *ReadThroughStoreTestSuite) SetupTest() {
|
|
suite.store = NewReadThroughStore(&MemoryStore{}, &TestStore{})
|
|
}
|
|
|
|
func (suite *ReadThroughStoreTestSuite) TearDownTest() {
|
|
}
|
|
|
|
func (suite *LevelDBStoreTestSuite) TestReadThroughStoreGet() {
|
|
bs := &TestStore{}
|
|
|
|
// Prepopulate the backing store with "abc".
|
|
input := "abc"
|
|
w := bs.Put()
|
|
_, err := w.Write([]byte(input))
|
|
suite.NoError(err)
|
|
ref := w.Ref()
|
|
|
|
// See http://www.di-mgt.com.au/sha_testvectors.html
|
|
suite.Equal("sha1-a9993e364706816aba3e25717850c26c9cd0d89d", ref.String())
|
|
|
|
suite.Equal(1, bs.Len())
|
|
suite.Equal(1, bs.Writes)
|
|
suite.Equal(0, bs.Reads)
|
|
|
|
cs := &TestStore{}
|
|
rts := NewReadThroughStore(cs, bs)
|
|
|
|
// Now read "abc". It is not yet in the cache so we hit the backing store.
|
|
reader := rts.Get(ref)
|
|
data, err := ioutil.ReadAll(reader)
|
|
suite.NoError(err)
|
|
suite.Equal(input, string(data))
|
|
reader.Close()
|
|
|
|
suite.Equal(1, bs.Len())
|
|
suite.Equal(1, cs.Len())
|
|
suite.Equal(1, cs.Writes)
|
|
suite.Equal(1, bs.Writes)
|
|
suite.Equal(1, cs.Reads)
|
|
suite.Equal(1, bs.Reads)
|
|
|
|
// Reading it again should not hit the backing store.
|
|
reader = rts.Get(ref)
|
|
data, err = ioutil.ReadAll(reader)
|
|
suite.NoError(err)
|
|
suite.Equal(input, string(data))
|
|
reader.Close()
|
|
|
|
suite.Equal(1, bs.Len())
|
|
suite.Equal(1, cs.Len())
|
|
suite.Equal(1, cs.Writes)
|
|
suite.Equal(1, bs.Writes)
|
|
suite.Equal(2, cs.Reads)
|
|
suite.Equal(1, bs.Reads)
|
|
}
|
|
|
|
func (suite *LevelDBStoreTestSuite) TestReadThroughStorePut() {
|
|
bs := &TestStore{}
|
|
cs := &TestStore{}
|
|
rts := NewReadThroughStore(cs, bs)
|
|
|
|
// Storing "abc" should store it to both backing and caching store.
|
|
input := "abc"
|
|
w := rts.Put()
|
|
_, err := w.Write([]byte(input))
|
|
suite.NoError(err)
|
|
ref := w.Ref()
|
|
|
|
// See http://www.di-mgt.com.au/sha_testvectors.html
|
|
suite.Equal("sha1-a9993e364706816aba3e25717850c26c9cd0d89d", ref.String())
|
|
|
|
assertInputInStore("abc", ref, bs, suite.Assert())
|
|
assertInputInStore("abc", ref, cs, suite.Assert())
|
|
assertInputInStore("abc", ref, rts, suite.Assert())
|
|
}
|