extra client info

cosmos · Jul 26, 2023 · e385250 · e385250
1 parent 993c21b
commit e385250
Show file tree

Hide file tree

Showing 3 changed files with 99 additions and 29 deletions.
diff --git a/cmd/query.go b/cmd/query.go
@@ -1079,18 +1079,17 @@ $ %s query clients-expiration demo-path`,
 			if err = c[dst].SetPath(path.Dst); err != nil {
 				return err
 			}
-
-			srcExpiration, err := relayer.QueryClientExpiration(cmd.Context(), c[src], c[dst])
+			srcExpiration, srcClientInfo, err := relayer.QueryClientExpiration(cmd.Context(), c[src], c[dst])
 			if err != nil {
 				return err
 			}
-			dstExpiration, err := relayer.QueryClientExpiration(cmd.Context(), c[dst], c[src])
+			dstExpiration, dstClientInfo, err := relayer.QueryClientExpiration(cmd.Context(), c[dst], c[src])
 			if err != nil {
 				return err
 			}
 
-			fmt.Fprintf(cmd.OutOrStdout(), relayer.SPrintClientExpiration(c[src], srcExpiration))
-			fmt.Fprintf(cmd.OutOrStdout(), relayer.SPrintClientExpiration(c[dst], dstExpiration))
+			fmt.Fprintf(cmd.OutOrStdout(), relayer.SPrintClientExpiration(c[src], srcExpiration, srcClientInfo))
+			fmt.Fprintf(cmd.OutOrStdout(), relayer.SPrintClientExpiration(c[dst], dstExpiration, dstClientInfo))
 
 			return nil
 		},

diff --git a/relayer/query.go b/relayer/query.go
@@ -272,39 +272,53 @@ func QueryBalance(ctx context.Context, chain *Chain, address string, showDenoms
 	return out, nil
 }
 
-func QueryClientExpiration(ctx context.Context, src, dst *Chain) (time.Time, error) {
+func QueryClientExpiration(ctx context.Context, src, dst *Chain) (time.Time, ClientStateInfo, error) {
 	latestHeight, err := src.ChainProvider.QueryLatestHeight(ctx)
 	if err != nil {
-		return time.Time{}, err
+		return time.Time{}, ClientStateInfo{}, err
 	}
 
 	clientStateRes, err := src.ChainProvider.QueryClientStateResponse(ctx, latestHeight, src.ClientID())
 	if err != nil {
-		return time.Time{}, err
+		return time.Time{}, ClientStateInfo{}, err
 	}
 
 	clientInfo, err := ClientInfoFromClientState(clientStateRes.ClientState)
 	if err != nil {
-		return time.Time{}, err
+		return time.Time{}, ClientStateInfo{}, err
 	}
 
 	clientTime, err := dst.ChainProvider.BlockTime(ctx, int64(clientInfo.LatestHeight.GetRevisionHeight()))
 	if err != nil {
-		return time.Time{}, err
+		return time.Time{}, ClientStateInfo{}, err
 	}
 
-	return clientTime.Add(clientInfo.TrustingPeriod), nil
+	return clientTime.Add(clientInfo.TrustingPeriod), clientInfo, nil
 }
 
-func SPrintClientExpiration(chain *Chain, expiration time.Time) string {
+func SPrintClientExpiration(chain *Chain, expiration time.Time, clientInfo ClientStateInfo) string {
 	now := time.Now()
 	remainingTime := expiration.Sub(now)
 	expirationFormatted := expiration.Format(time.RFC822)
 
+	var status string
 	if remainingTime < 0 {
-		return fmt.Sprintf("client %s (%s) is already expired (%s)\n",
-			chain.ClientID(), chain.ChainID(), expirationFormatted)
+		status = "EXPIRED"
+	} else {
+		status = "GOOD"
 	}
-	return fmt.Sprintf("client %s (%s) expires in %s (%s)\n",
-		chain.ClientID(), chain.ChainID(), remainingTime.Round(time.Second), expirationFormatted)
+
+	// if remainingTime < 0 {
+	// 	return fmt.Sprintf("client %s (%s) is already expired (%s)\n",
+	// 		chain.ClientID(), chain.ChainID(), expirationFormatted)
+	// }
+	return fmt.Sprintf(`
+	client: %s (%s)
+		HEALTH: %s
+		TIME: %s (%s)
+		LAST UPDATE HEIGHT: %d
+		TRUSTING PERIOD: %s
+
+	`,
+		chain.ClientID(), chain.ChainID(), status, expirationFormatted, remainingTime.Round(time.Second), clientInfo.LatestHeight.GetRevisionHeight(), clientInfo.TrustingPeriod.String())
 }
diff --git a/relayer/query_test.go b/relayer/query_test.go
@@ -1,76 +1,124 @@
 package relayer
 
 import (
-	"github.com/cosmos/relayer/v2/relayer/chains/cosmos"
 	"testing"
 	"time"
 
+	"github.com/cosmos/relayer/v2/relayer/chains/cosmos"
+
+	clienttypes "github.com/cosmos/ibc-go/v7/modules/core/02-client/types"
+	ibcexported "github.com/cosmos/ibc-go/v7/modules/core/exported"
 	"github.com/stretchr/testify/require"
 )
 
 func TestSPrintClientExpiration_PrintChainId(t *testing.T) {
 	previousTime := time.Now().Add(10 * time.Hour)
+	mockHeight := clienttypes.NewHeight(1, 100)
+	trustingPeriod := time.Duration(1 * time.Hour)
 
 	chain := mockChain("expected-chain-id", "test-client-id")
-	expiration := SPrintClientExpiration(chain, previousTime)
+	clientStateInfo := mockClientStateInfo("test-chain-id", trustingPeriod, mockHeight)
+	expiration := SPrintClientExpiration(chain, previousTime, *clientStateInfo)
 
 	require.Contains(t, expiration, "expected-chain-id")
 }
 
 func TestSPrintClientExpiration_PrintClientId(t *testing.T) {
 	previousTime := time.Now().Add(10 * time.Hour)
+	mockHeight := clienttypes.NewHeight(1, 100)
+	trustingPeriod := time.Duration(1 * time.Hour)
 
 	chain := mockChain("test-chain-id", "expected-client-id")
-	expiration := SPrintClientExpiration(chain, previousTime)
+	clientStateInfo := mockClientStateInfo("test-chain-id", trustingPeriod, mockHeight)
+	expiration := SPrintClientExpiration(chain, previousTime, *clientStateInfo)
 
 	require.Contains(t, expiration, "expected-client-id")
 }
 
-func TestSPrintClientExpiration_PrintIsAlreadyExpired_WhenTimeIsInPast(t *testing.T) {
+func TestSPrintClientExpiration_PrintExpired_WhenTimeIsInPast(t *testing.T) {
 	previousTime := time.Now().Add(-10 * time.Hour)
+	mockHeight := clienttypes.NewHeight(1, 100)
+	trustingPeriod := time.Duration(1 * time.Hour)
 
 	chain := mockChain("test-chain-id", "test-client-id")
-	expiration := SPrintClientExpiration(chain, previousTime)
+	clientStateInfo := mockClientStateInfo("test-chain-id", trustingPeriod, mockHeight)
+	expiration := SPrintClientExpiration(chain, previousTime, *clientStateInfo)
 
-	require.Contains(t, expiration, "is already expired")
+	require.Contains(t, expiration, "EXPIRED")
 }
 
 func TestSPrintClientExpiration_PrintRFC822FormattedTime_WhenTimeIsInPast(t *testing.T) {
 	pastTime := time.Now().Add(-10 * time.Hour)
+	mockHeight := clienttypes.NewHeight(1, 100)
+	trustingPeriod := time.Duration(1 * time.Hour)
 
-	chain := mockChain("test-chain-id", "test-client-id")
-	expiration := SPrintClientExpiration(chain, pastTime)
+	chain := mockChain("expected-chain-id", "test-client-id")
+	clientStateInfo := mockClientStateInfo("test-chain-id", trustingPeriod, mockHeight)
+	expiration := SPrintClientExpiration(chain, pastTime, *clientStateInfo)
 
 	require.Contains(t, expiration, pastTime.Format(time.RFC822))
 }
 
-func TestSPrintClientExpiration_PrintExpiresIn_WhenTimeIsInFuture(t *testing.T) {
+func TestSPrintClientExpiration_PrintGood_WhenTimeIsInFuture(t *testing.T) {
 	previousTime := time.Now().Add(10 * time.Hour)
+	mockHeight := clienttypes.NewHeight(1, 100)
+	trustingPeriod := time.Duration(1 * time.Hour)
 
 	chain := mockChain("test-chain-id", "test-client-id")
-	expiration := SPrintClientExpiration(chain, previousTime)
+	clientStateInfo := mockClientStateInfo("test-chain-id", trustingPeriod, mockHeight)
+	expiration := SPrintClientExpiration(chain, previousTime, *clientStateInfo)
 
-	require.Contains(t, expiration, "expires in")
+	require.Contains(t, expiration, "GOOD")
 }
 
 func TestSPrintClientExpiration_PrintRFC822FormattedTime_WhenTimeIsInFuture(t *testing.T) {
 	futureTime := time.Now().Add(10 * time.Hour)
+	mockHeight := clienttypes.NewHeight(1, 100)
+	trustingPeriod := time.Duration(1 * time.Hour)
 
 	chain := mockChain("test-chain-id", "test-client-id")
-	expiration := SPrintClientExpiration(chain, futureTime)
+	clientStateInfo := mockClientStateInfo("test-chain-id", trustingPeriod, mockHeight)
+	expiration := SPrintClientExpiration(chain, futureTime, *clientStateInfo)
 
 	require.Contains(t, expiration, futureTime.Format(time.RFC822))
 }
 
 func TestSPrintClientExpiration_PrintRemainingTime_WhenTimeIsInFuture(t *testing.T) {
 	futureTime := time.Now().Add(10 * time.Hour)
+	mockHeight := clienttypes.NewHeight(1, 100)
+	trustingPeriod := time.Duration(1 * time.Hour)
 
 	chain := mockChain("test-chain-id", "test-client-id")
-	expiration := SPrintClientExpiration(chain, futureTime)
+	clientStateInfo := mockClientStateInfo("test-chain-id", trustingPeriod, mockHeight)
+	expiration := SPrintClientExpiration(chain, futureTime, *clientStateInfo)
 
 	require.Contains(t, expiration, "10h0m0s")
 }
 
+func TestSPrintClientExpiration_TrustingPeriod(t *testing.T) {
+	previousTime := time.Now().Add(10 * time.Hour)
+	mockHeight := clienttypes.NewHeight(1, 100)
+	trustingPeriod := time.Duration(1 * time.Hour)
+
+	chain := mockChain("expected-chain-id", "test-client-id")
+	clientStateInfo := mockClientStateInfo("test-chain-id", trustingPeriod, mockHeight)
+	expiration := SPrintClientExpiration(chain, previousTime, *clientStateInfo)
+
+	require.Contains(t, expiration, "1h0m0s")
+}
+
+func TestSPrintClientExpiration_LastUpdateHeight(t *testing.T) {
+	previousTime := time.Now().Add(10 * time.Hour)
+	mockHeight := clienttypes.NewHeight(1, 100)
+	trustingPeriod := time.Duration(1 * time.Hour)
+
+	chain := mockChain("expected-chain-id", "test-client-id")
+	clientStateInfo := mockClientStateInfo("test-chain-id", trustingPeriod, mockHeight)
+	expiration := SPrintClientExpiration(chain, previousTime, *clientStateInfo)
+
+	require.Contains(t, expiration, "100")
+}
+
 func mockChain(chainId string, clientId string) *Chain {
 	return &Chain{
 		Chainid: chainId,
@@ -85,3 +133,12 @@ func mockChain(chainId string, clientId string) *Chain {
 		},
 	}
 }
+
+func mockClientStateInfo(chainID string, trustingPeriod time.Duration, latestHeight ibcexported.Height) *ClientStateInfo {
+	mockHeight := clienttypes.NewHeight(1, 100)
+	return &ClientStateInfo{
+		ChainID:        chainID,
+		TrustingPeriod: time.Duration(1 * time.Hour),
+		LatestHeight:   mockHeight,
+	}
+}