github.com/ewagmig/fabric@v2.1.1+incompatible/core/ledger/kvledger/txmgmt/privacyenabledstate/common_storage_db_test.go (about) 1 /* 2 Copyright IBM Corp. All Rights Reserved. 3 4 SPDX-License-Identifier: Apache-2.0 5 */ 6 7 package privacyenabledstate_test 8 9 import ( 10 "testing" 11 12 "github.com/hyperledger/fabric/core/ledger/kvledger/txmgmt/privacyenabledstate" 13 "github.com/hyperledger/fabric/core/ledger/kvledger/txmgmt/statedb/statecouchdb" 14 "github.com/hyperledger/fabric/core/ledger/kvledger/txmgmt/statedb/stateleveldb" 15 "github.com/hyperledger/fabric/core/ledger/mock" 16 . "github.com/onsi/gomega" 17 ) 18 19 func TestHealthCheckRegister(t *testing.T) { 20 gt := NewGomegaWithT(t) 21 fakeHealthCheckRegistry := &mock.HealthCheckRegistry{} 22 23 dbProvider := &privacyenabledstate.CommonStorageDBProvider{ 24 VersionedDBProvider: &stateleveldb.VersionedDBProvider{}, 25 HealthCheckRegistry: fakeHealthCheckRegistry, 26 } 27 28 err := dbProvider.RegisterHealthChecker() 29 gt.Expect(err).NotTo(HaveOccurred()) 30 gt.Expect(fakeHealthCheckRegistry.RegisterCheckerCallCount()).To(Equal(0)) 31 32 dbProvider.VersionedDBProvider = &statecouchdb.VersionedDBProvider{} 33 err = dbProvider.RegisterHealthChecker() 34 gt.Expect(err).NotTo(HaveOccurred()) 35 gt.Expect(fakeHealthCheckRegistry.RegisterCheckerCallCount()).To(Equal(1)) 36 37 arg1, arg2 := fakeHealthCheckRegistry.RegisterCheckerArgsForCall(0) 38 gt.Expect(arg1).To(Equal("couchdb")) 39 gt.Expect(arg2).NotTo(Equal(nil)) 40 }