mirror of
https://github.com/ceph/ceph-csi.git
synced 2024-11-27 16:50:23 +00:00
118 lines
2.6 KiB
Go
118 lines
2.6 KiB
Go
package diskv
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"testing"
|
|
)
|
|
|
|
func TestBasicStreamCaching(t *testing.T) {
|
|
d := New(Options{
|
|
BasePath: "test-data",
|
|
CacheSizeMax: 1024,
|
|
})
|
|
defer d.EraseAll()
|
|
|
|
input := "a1b2c3"
|
|
key, writeBuf, sync := "a", bytes.NewBufferString(input), true
|
|
if err := d.WriteStream(key, writeBuf, sync); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if d.isCached(key) {
|
|
t.Fatalf("'%s' cached, but shouldn't be (yet)", key)
|
|
}
|
|
|
|
rc, err := d.ReadStream(key, false)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
readBuf, err := ioutil.ReadAll(rc)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !cmpBytes(readBuf, []byte(input)) {
|
|
t.Fatalf("'%s' != '%s'", string(readBuf), input)
|
|
}
|
|
|
|
if !d.isCached(key) {
|
|
t.Fatalf("'%s' isn't cached, but should be", key)
|
|
}
|
|
}
|
|
|
|
func TestReadStreamDirect(t *testing.T) {
|
|
var (
|
|
basePath = "test-data"
|
|
)
|
|
dWrite := New(Options{
|
|
BasePath: basePath,
|
|
CacheSizeMax: 0,
|
|
})
|
|
defer dWrite.EraseAll()
|
|
dRead := New(Options{
|
|
BasePath: basePath,
|
|
CacheSizeMax: 1024,
|
|
})
|
|
|
|
// Write
|
|
key, val1, val2 := "a", []byte(`1234567890`), []byte(`aaaaaaaaaa`)
|
|
if err := dWrite.Write(key, val1); err != nil {
|
|
t.Fatalf("during first write: %s", err)
|
|
}
|
|
|
|
// First, caching read.
|
|
val, err := dRead.Read(key)
|
|
if err != nil {
|
|
t.Fatalf("during initial read: %s", err)
|
|
}
|
|
t.Logf("read 1: %s => %s", key, string(val))
|
|
if !cmpBytes(val1, val) {
|
|
t.Errorf("expected %q, got %q", string(val1), string(val))
|
|
}
|
|
if !dRead.isCached(key) {
|
|
t.Errorf("%q should be cached, but isn't", key)
|
|
}
|
|
|
|
// Write a different value.
|
|
if err := dWrite.Write(key, val2); err != nil {
|
|
t.Fatalf("during second write: %s", err)
|
|
}
|
|
|
|
// Second read, should hit cache and get the old value.
|
|
val, err = dRead.Read(key)
|
|
if err != nil {
|
|
t.Fatalf("during second (cache-hit) read: %s", err)
|
|
}
|
|
t.Logf("read 2: %s => %s", key, string(val))
|
|
if !cmpBytes(val1, val) {
|
|
t.Errorf("expected %q, got %q", string(val1), string(val))
|
|
}
|
|
|
|
// Third, direct read, should get the updated value.
|
|
rc, err := dRead.ReadStream(key, true)
|
|
if err != nil {
|
|
t.Fatalf("during third (direct) read, ReadStream: %s", err)
|
|
}
|
|
defer rc.Close()
|
|
val, err = ioutil.ReadAll(rc)
|
|
if err != nil {
|
|
t.Fatalf("during third (direct) read, ReadAll: %s", err)
|
|
}
|
|
t.Logf("read 3: %s => %s", key, string(val))
|
|
if !cmpBytes(val2, val) {
|
|
t.Errorf("expected %q, got %q", string(val1), string(val))
|
|
}
|
|
|
|
// Fourth read, should hit cache and get the new value.
|
|
val, err = dRead.Read(key)
|
|
if err != nil {
|
|
t.Fatalf("during fourth (cache-hit) read: %s", err)
|
|
}
|
|
t.Logf("read 4: %s => %s", key, string(val))
|
|
if !cmpBytes(val2, val) {
|
|
t.Errorf("expected %q, got %q", string(val1), string(val))
|
|
}
|
|
}
|