Fix gas in timeout tests

strangelove-ventures · Apr 14, 2022 · db3496c · db3496c
1 parent 4db4409
commit db3496c
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/ibc/test_relay.go b/ibc/test_relay.go
@@ -333,8 +333,10 @@ func (ibc IBCTestCase) RelayPacketTestHeightTimeout(testName string, srcChain Ch
 		return err
 	}
 
-	if srcFinalBalance != srcInitialBalance {
-		return fmt.Errorf("source balances do not match. expected: %d, actual: %d", srcInitialBalance, srcFinalBalance)
+	totalFees := srcChain.GetGasFeesInNativeDenom(srcTx.GasWanted)
+
+	if srcFinalBalance != srcInitialBalance-totalFees {
+		return fmt.Errorf("source balances do not match. expected: %d, actual: %d", srcInitialBalance-totalFees, srcFinalBalance)
 	}
 
 	if dstFinalBalance != dstInitialBalance {
@@ -423,8 +425,10 @@ func (ibc IBCTestCase) RelayPacketTestTimestampTimeout(testName string, srcChain
 		return err
 	}
 
-	if srcFinalBalance != srcInitialBalance {
-		return fmt.Errorf("source balances do not match. expected: %d, actual: %d", srcInitialBalance, srcFinalBalance)
+	totalFees := srcChain.GetGasFeesInNativeDenom(srcTx.GasWanted)
+
+	if srcFinalBalance != srcInitialBalance-totalFees {
+		return fmt.Errorf("source balances do not match. expected: %d, actual: %d", srcInitialBalance-totalFees, srcFinalBalance)
 	}
 
 	if dstFinalBalance != dstInitialBalance {