bitxid/kvdb_test.go at master · meshplus/bitxid · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
package bitxid

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

	"github.com/meshplus/bitxhub-kit/storage/leveldb"
	"github.com/stretchr/testify/assert"
)

func TestDBCURD(t *testing.T) {
	dir, err := ioutil.TempDir("", "doc.db")
	assert.Nil(t, err)

	defer os.RemoveAll(dir)

	key := DID("did:bitxhub:appchain001:.")
	value := AccountDoc{
		BasicDoc: BasicDoc{ID: "did:bitxhub:appchain001:."},
	}
	valueUpdated := AccountDoc{
		BasicDoc: BasicDoc{ID: "did:bitxhub:appchain001:."},
		Service:  "test",
	}
	// dbPath := filepath.Join(dir, "docdb")
	s, err := leveldb.New(dir)
	assert.Nil(t, err)
	d, err := NewKVDocDB(s)
	assert.Nil(t, err)
	// test create:
	ret1, err := d.Create(&value)
	assert.Nil(t, err)
	assert.Equal(t, "./"+string(key), ret1)
	// test has:
	ret2 := d.Has(key)
	// assert.Nil(t, err)
	assert.Equal(t, true, ret2)
	// test get:
	ret3, err := d.Get(key, AccountDIDType)
	assert.Nil(t, err)
	assert.Equal(t, *ret3.(*AccountDoc), value)
	// test update:
	ret4, err := d.Update(&valueUpdated)
	assert.Nil(t, err)
	assert.Equal(t, "./"+string(key), ret4)
	ret5, err := d.Get(key, AccountDIDType)
	assert.Nil(t, err)
	assert.Equal(t, *ret5.(*AccountDoc), valueUpdated)
	// test delete:
	d.Delete(key)
	// assert.Nil(t, err)
	ret6 := d.Has(key)
	// assert.Nil(t, err)
	assert.Equal(t, false, ret6)
}