forked from hyperledger/fabric
/
statecouchdb_test_export.go
48 lines (40 loc) · 1.42 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
/*
Copyright IBM Corp. All Rights Reserved.
SPDX-License-Identifier: Apache-2.0
*/
package statecouchdb
import (
"strings"
"testing"
"github.com/hyperledger/fabric/core/ledger/kvledger/txmgmt/statedb"
"github.com/hyperledger/fabric/core/ledger/util/couchdb"
)
// 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()
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(dbName string) {
env.t.Logf("Cleaningup TestVDBEnv")
env.DBProvider.Close()
CleanupDB(dbName)
}
// CleanupDB drops the test couch databases
func CleanupDB(dbName string) {
//create a new connection
couchDBDef := couchdb.GetCouchDBDefinition()
couchInstance, _ := couchdb.CreateCouchInstance(couchDBDef.URL, couchDBDef.Username, couchDBDef.Password,
couchDBDef.MaxRetries, couchDBDef.MaxRetriesOnStartup, couchDBDef.RequestTimeout)
db := couchdb.CouchDatabase{CouchInstance: couchInstance, DBName: strings.ToLower(dbName)}
//drop the test database
db.DropDatabase()
}