forked from hyperledger/fabric
-
Notifications
You must be signed in to change notification settings - Fork 11
/
statecouchdb_test_export.go
53 lines (43 loc) · 1.47 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
/*
Copyright IBM Corp. All Rights Reserved.
SPDX-License-Identifier: Apache-2.0
*/
package statecouchdb
import (
"testing"
"github.com/hyperledger/fabric/common/metrics/disabled"
"github.com/hyperledger/fabric/core/ledger/kvledger/txmgmt/statedb"
"github.com/stretchr/testify/assert"
)
// TestVDBEnv provides a couch db backed versioned db for testing
type TestVDBEnv struct {
t testing.TB
DBProvider statedb.VersionedDBProvider
}
// NewTestVDBEnv instantiates and new couch db backed TestVDB
func NewTestVDBEnv(t testing.TB) *TestVDBEnv {
t.Logf("Creating new TestVDBEnv")
dbProvider, _ := NewVersionedDBProvider(&disabled.Provider{})
testVDBEnv := &TestVDBEnv{t, dbProvider}
// No cleanup for new test environment. Need to cleanup per test for each DB used in the test.
return testVDBEnv
}
// Cleanup drops the test couch databases and closes the db provider
func (env *TestVDBEnv) Cleanup() {
env.t.Logf("Cleaningup TestVDBEnv")
CleanupDB(env.t, env.DBProvider)
env.DBProvider.Close()
}
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)
}
}
}
}