GetCertStore codecov

Author: hanabi1224

f3.go

@@ -113,20 +113,20 @@ func (m *F3) Broadcast(ctx context.Context, signatureBuilder *gpbft.SignatureBui
 }
 
 func (m *F3) GetLatestCert(context.Context) (*certs.FinalityCertificate, error) {
-	if state := m.state.Load(); state != nil {
-		return state.cs.Latest(), nil
+	if cs, err := m.GetCertStore(); err != nil {
+		return cs.Latest(), nil
 	}
 	return nil, ErrF3NotRunning
 }
 
 func (m *F3) GetCert(ctx context.Context, instance uint64) (*certs.FinalityCertificate, error) {
-	if state := m.state.Load(); state != nil {
-		return state.cs.Get(ctx, instance)
+	if cs, err := m.GetCertStore(); err != nil {
+		return cs.Get(ctx, instance)
 	}
 	return nil, ErrF3NotRunning
 }
 
-func (m *F3) GetCertStore(ctx context.Context) (*certstore.Store, error) {
+func (m *F3) GetCertStore() (*certstore.Store, error) {
 	if state := m.state.Load(); state != nil {
 		return state.cs, nil
 	}
@@ -135,8 +135,8 @@ func (m *F3) GetCertStore(ctx context.Context) (*certstore.Store, error) {
 
 // GetPowerTableByInstance returns the power table (committee) used to validate the specified instance.
 func (m *F3) GetPowerTableByInstance(ctx context.Context, instance uint64) (gpbft.PowerEntries, error) {
-	if state := m.state.Load(); state != nil {
-		return state.cs.GetPowerTable(ctx, instance)
+	if cs, err := m.GetCertStore(); err != nil {
+		return cs.GetPowerTable(ctx, instance)
 	}
 	return nil, ErrF3NotRunning
 }