-
Notifications
You must be signed in to change notification settings - Fork 1
/
statecouchdb_test_export.go
87 lines (73 loc) · 2.1 KB
/
statecouchdb_test_export.go
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
/*
Copyright IBM Corp. All Rights Reserved.
SPDX-License-Identifier: Apache-2.0
*/
package statecouchdb
import (
"io/ioutil"
"os"
"testing"
"time"
"github.com/mcc-github/blockchain/common/metrics/disabled"
"github.com/mcc-github/blockchain/core/ledger/kvledger/txmgmt/statedb"
"github.com/mcc-github/blockchain/core/ledger/util/couchdb"
"github.com/stretchr/testify/assert"
)
var couchAddress string
type TestVDBEnv struct {
t testing.TB
DBProvider statedb.VersionedDBProvider
config *couchdb.Config
}
func NewTestVDBEnv(t testing.TB) *TestVDBEnv {
t.Logf("Creating new TestVDBEnv")
redoPath, err := ioutil.TempDir("", "cvdbenv")
if err != nil {
t.Fatalf("Failed to create redo log directory: %s", err)
}
config := &couchdb.Config{
Address: couchAddress,
Username: "",
Password: "",
InternalQueryLimit: 1000,
MaxBatchUpdateSize: 1000,
MaxRetries: 3,
MaxRetriesOnStartup: 20,
RequestTimeout: 35 * time.Second,
RedoLogPath: redoPath,
}
dbProvider, err := NewVersionedDBProvider(config, &disabled.Provider{})
if err != nil {
t.Fatalf("Error creating CouchDB Provider: %s", err)
}
testVDBEnv := &TestVDBEnv{
t: t,
DBProvider: dbProvider,
config: config,
}
return testVDBEnv
}
func (env *TestVDBEnv) CloseAndReopen() {
env.DBProvider.Close()
dbProvider, _ := NewVersionedDBProvider(env.config, &disabled.Provider{})
env.DBProvider = dbProvider
}
func (env *TestVDBEnv) Cleanup() {
env.t.Logf("Cleaningup TestVDBEnv")
CleanupDB(env.t, env.DBProvider)
env.DBProvider.Close()
os.RemoveAll(env.config.RedoLogPath)
}
func CleanupDB(t testing.TB, dbProvider statedb.VersionedDBProvider) {
couchdbProvider, _ := dbProvider.(*VersionedDBProvider)
for _, v := range couchdbProvider.databases {
if _, err := v.metadataDB.DropDatabase(); err != nil {
assert.Failf(t, "DropDatabase %s fails. err: %v", v.metadataDB.DBName, err)
}
for _, db := range v.namespaceDBs {
if _, err := db.DropDatabase(); err != nil {
assert.Failf(t, "DropDatabase %s fails. err: %v", db.DBName, err)
}
}
}
}