From fd56278f992020593d8498ffad001855efcb25eb Mon Sep 17 00:00:00 2001
From: Jun Kimura <jun.kimura@datachain.jp>
Date: Wed, 14 Feb 2024 15:28:32 +0900
Subject: [PATCH] fix non-existence proof generation for mock-client

Signed-off-by: Jun Kimura <jun.kimura@datachain.jp>
---
 pkg/testing/chains.go | 34 +++++++++++++++-------------------
 1 file changed, 15 insertions(+), 19 deletions(-)

diff --git a/pkg/testing/chains.go b/pkg/testing/chains.go
index 0d9614ca..d25606ae 100644
--- a/pkg/testing/chains.go
+++ b/pkg/testing/chains.go
@@ -910,25 +910,21 @@ func (chain *Chain) TimeoutOnClose(
 		}
 		proofUnreceived = proof.Data
 	} else {
-		if chain.ClientType() == ibcclient.MockClient {
-			proofUnreceived = []byte{}
-		} else {
-			rc, err := counterparty.IBCHandler.GetPacketReceipt(
-				counterparty.CallOpts(ctx, RelayerKeyIndex),
-				packet.DestinationPort, packet.DestinationChannel, packet.Sequence,
-			)
-			if err != nil {
-				return err
-			}
-			if rc == 1 {
-				return fmt.Errorf("packet receipt exists: port=%v channel=%v sequence=%v", packet.DestinationPort, packet.DestinationChannel, packet.Sequence)
-			}
-			p, err := counterparty.QueryPacketReceiptProof(chain, channel.ClientID, packet, proofHeight)
-			if err != nil {
-				return err
-			}
-			proofUnreceived = p.Data
+		rc, err := counterparty.IBCHandler.GetPacketReceipt(
+			counterparty.CallOpts(ctx, RelayerKeyIndex),
+			packet.DestinationPort, packet.DestinationChannel, packet.Sequence,
+		)
+		if err != nil {
+			return err
+		}
+		if rc == 1 {
+			return fmt.Errorf("packet receipt exists: port=%v channel=%v sequence=%v", packet.DestinationPort, packet.DestinationChannel, packet.Sequence)
+		}
+		p, err := counterparty.QueryPacketReceiptProof(chain, channel.ClientID, packet, proofHeight)
+		if err != nil {
+			return err
 		}
+		proofUnreceived = p.Data
 	}
 
 	nextSequenceRecv, err := counterparty.IBCHandler.GetNextSequenceRecv(
@@ -1326,7 +1322,7 @@ func (chain *Chain) QueryPacketReceiptProof(counterparty *Chain, counterpartyCli
 		if rc == 1 {
 			return nil, errors.New("an existence proof of packet receipt not supported")
 		} else {
-			proof.Data = []byte{}
+			proof.Data = chain.generateMockClientProof(proof.Height, host.PacketReceiptPath(packetFromCounterparty.DestinationPort, packetFromCounterparty.DestinationChannel, packetFromCounterparty.Sequence), []byte{})
 		}
 	}
 	return proof, nil