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  }