github.com/bingoohuang/gg@v0.0.0-20240325092523-45da7dee9335/pkg/badgerdb/badger_test.go (about)

     1  package badgerdb
     2  
     3  import (
     4  	"fmt"
     5  	"testing"
     6  
     7  	"github.com/bingoohuang/gg/pkg/bytex"
     8  	"github.com/stretchr/testify/assert"
     9  )
    10  
    11  func TestNew(t *testing.T) {
    12  	db, err := Open(WithInMemory(true))
    13  	assert.Nil(t, err)
    14  
    15  	for i := uint64(0); i < 100; i++ {
    16  		err = db.Set(bytex.FromUint64(i), []byte(fmt.Sprintf("hello:%d", i)))
    17  		assert.Nil(t, err)
    18  	}
    19  
    20  	v, err := db.Get(bytex.FromUint64(99))
    21  	assert.Nil(t, err)
    22  	assert.Equal(t, "hello:99", string(v))
    23  
    24  	err = db.Walk(func(k, v []byte) error {
    25  		fmt.Printf("%d=%s\n", bytex.ToUint64(k), v)
    26  		return nil
    27  	})
    28  	assert.Nil(t, err)
    29  	assert.Nil(t, db.Close())
    30  }