BufDB

import (
  "bytes"
  "io/ioutil"
  "os"
  "testing"

  "github.com/djherbis/buffer"

  bolt "go.etcd.io/bbolt"
)

func TestStore(t *testing.T) {
  db, err := bolt.Open("my.db", 0600, nil)
  if err != nil {
    t.Error(err.Error())
  }
  defer os.Remove("my.db")
  defer db.Close()

  store := NewBufferStore(db, []byte("buckets"))

  key := []byte("hello")
  input := []byte("hello world")

  b := buffer.NewPartition(buffer.NewFilePool(10, "."))
  b.Write(input)

  store.Put(key, b)
  buf, err := store.Pull(key)
  if err != nil {
    t.Error(err.Error())
  }

  data, err := ioutil.ReadAll(buf)

  if !bytes.Equal(data, input) {
    t.Errorf("expected %s, got %s", input, data)
  }

  store.Wipe()
}