use existing ConnectToCanonicalValidators to set up for the test

signature-aggregator/aggregator/aggregator.go

@@ -44,7 +44,7 @@ const (
 var (
 	// Errors
 	errNotEnoughSignatures     = errors.New("failed to collect a threshold of signatures")
-	errNotEnoughConnectedStake = errors.New("failed to connect to a threshold of stake")
+	ErrNotEnoughConnectedStake = errors.New("failed to connect to a threshold of stake")
 )
 
 type SignatureAggregator struct {
@@ -149,7 +149,7 @@ func (s *SignatureAggregator) CreateSignedMessage(
 			zap.Uint64("quorumPercentage", quorumPercentage),
 		)
 		s.metrics.FailuresToConnectToSufficientStake.Inc()
-		return nil, errNotEnoughConnectedStake
+		return nil, ErrNotEnoughConnectedStake
 	}
 
 	accumulatedSignatureWeight := big.NewInt(0)

signature-aggregator/healthcheck/healthcheck.go

@@ -2,28 +2,38 @@ package healthcheck
 
 import (
 	"context"
+	"fmt"
+	"math/big"
 	"net/http"
 
 	"github.com/alexliesenfeld/health"
+	"github.com/ava-labs/avalanchego/utils/constants"
+	"github.com/ava-labs/coreth/precompile/contracts/warp"
+	"github.com/ava-labs/icm-services/peers"
+	"github.com/ava-labs/icm-services/signature-aggregator/aggregator"
+	"github.com/ava-labs/icm-services/utils"
 )
 
-func HandleHealthCheckRequest(uint16 apiPort) {
+func HandleHealthCheckRequest(network peers.AppRequestNetwork) {
 	healthChecker := health.NewChecker(
 		health.WithCheck(health.Check{
 			Name: "signature-aggregator-health",
 			Check: func(context.Context) error {
+				connectedValidators, err := network.ConnectToCanonicalValidators(constants.PrimaryNetworkID)
+				if err != nil {
+					return fmt.Errorf("Failed to connect to primary network validators: %w", err)
+				}
+				if !utils.CheckStakeWeightExceedsThreshold(
+					big.NewInt(0).SetUint64(connectedValidators.ConnectedWeight),
+					connectedValidators.TotalValidatorWeight,
+					warp.WarpDefaultQuorumNumerator,
+				) {
+					return aggregator.ErrNotEnoughConnectedStake
+				}
+				return nil
 			},
 		}),
 	)
 
-	readinessCheck := health.NewChecker(
-		health.WithCheck(health.Check{
-			Name: "signature-aggregator-readiness",
-			Check: func(context.Context) error {
-
-			},
-		}),
-	)
 	http.Handle("/health", health.NewHandler(healthChecker))
-	http.Handle("/ready", health.NewHandler(readinessCheck))
 }

signature-aggregator/main/main.go

@@ -139,7 +139,7 @@ func main() {
 		metricsInstance,
 		signatureAggregator,
 	)
-	healthcheck.HandleHealthCheckRequest(cfg.APIPort)
+	healthcheck.HandleHealthCheckRequest(network)
 
 	logger.Info("Initialization complete")
 	err = http.ListenAndServe(fmt.Sprintf(":%d", cfg.APIPort), nil)