add TestParseTychoShardAccount

Author: callmedenchick

tychoclient/client_test.go

@@ -430,3 +430,76 @@ func TestParseTychoBlockErrorCases(t *testing.T) {
 		})
 	}
 }
+
+// TestParseTychoShardAccount tests parsing of shard account data fetched from Tycho.
+func TestParseTychoShardAccount(t *testing.T) {
+	client, err := NewClient()
+	if err != nil {
+		t.Fatalf("Failed to create client: %v", err)
+	}
+	defer client.Close()
+
+	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
+	defer cancel()
+
+	// Test with a zero address (most likely to return some data)
+	testAddress := make([]byte, 32)
+
+	accountInfo, err := client.GetShardAccount(ctx, 0, testAddress, false)
+	if err != nil {
+		t.Fatalf("GetShardAccount failed, account might not exists: %v", err)
+	}
+
+	if len(accountInfo.AccountState) == 0 {
+		t.Skip("No account data to test parsing")
+	}
+
+	// Verify that ParseShardAccount works
+	if accountInfo.ParsedAccountState == nil {
+		t.Error("ParsedAccountState should not be nil when AccountState exists")
+	}
+
+	t.Logf("Successfully parsed account data: %d bytes -> cell with %d bits, %d refs",
+		len(accountInfo.AccountState),
+		accountInfo.ParsedAccountState.BitsAvailableForRead(),
+		accountInfo.ParsedAccountState.RefsAvailableForRead())
+}
+
+// TestParseTychoShardAccountAtSeqno tests fetching and parsing account data at a specific seqno
+func TestParseTychoShardAccountAtSeqno(t *testing.T) {
+	client, err := NewClient()
+	if err != nil {
+		t.Fatalf("Failed to create client: %v", err)
+	}
+	defer client.Close()
+
+	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
+	defer cancel()
+
+	// Get status to find a valid seqno
+	status, err := client.GetStatus(ctx)
+	if err != nil {
+		t.Fatalf("Failed to get status: %v", err)
+	}
+
+	if status.McStateInfo == nil || status.McStateInfo.McSeqno < 2 {
+		t.Skip("Need at least seqno 2 for historical testing")
+	}
+
+	// Test with a historical seqno
+	targetSeqno := status.McStateInfo.McSeqno - 1
+	testAddress := make([]byte, 32) // Zero address
+
+	accountInfo, err := client.GetShardAccountAtSeqno(ctx, 0, testAddress, false, masterchainWorkchain, masterchainShard, targetSeqno)
+	if err != nil {
+		t.Logf("GetShardAccountAtSeqno failed: %v", err)
+		return // Expected for non-existent accounts
+	}
+
+	// If we got data, verify parsing works
+	if len(accountInfo.AccountState) > 0 && accountInfo.ParsedAccountState == nil {
+		t.Error("ParsedAccountState should not be nil when AccountState exists")
+	}
+
+	t.Logf("Historical account parsing test completed")
+}