fix: tests (#815)

icon-project · Jan 3, 2024 · af511b1 · af511b1
1 parent 354914e
commit af511b1
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 8 deletions.
diff --git a/...javascore/lightclients/tendermint/src/main/java/ibc/tendermint/TendermintLightClient.java b/...javascore/lightclients/tendermint/src/main/java/ibc/tendermint/TendermintLightClient.java
@@ -14,6 +14,7 @@
 import score.Context;
 import score.DictDB;
 import score.annotation.External;
+import score.annotation.Optional;
 
 import java.math.BigInteger;
 import java.util.Arrays;
@@ -39,14 +40,17 @@ public class TendermintLightClient extends Tendermint implements ILightClient {
     public static final BranchDB<String, DictDB<BigInteger, BigInteger>> processedHeights = Context.newBranchDB(
             PROCESSED_HEIGHTS, BigInteger.class);
 
-    public TendermintLightClient(Address ibcHandler) {
+    public TendermintLightClient(Address ibcHandler, @Optional boolean update) {
         this.ibcHandler = ibcHandler;
-        String neutronClient = "07-tendermint-2";
-        ClientState cs = ClientState.decode(getClientState(neutronClient));
-        Duration tp = cs.getTrustingPeriod();
-        tp.setSeconds(tp.getSeconds().multiply(BigInteger.TWO));
-        cs.setTrustingPeriod(tp);
-        clientStates.set(neutronClient, cs.encode());
+
+        if (update) {
+                String neutronClient = "07-tendermint-2";
+                ClientState cs = ClientState.decode(getClientState(neutronClient));
+                Duration tp = cs.getTrustingPeriod();
+                tp.setSeconds(tp.getSeconds().multiply(BigInteger.TWO));
+                cs.setTrustingPeriod(tp);
+                clientStates.set(neutronClient, cs.encode());
+        }
     }
 
     private void onlyHandler() {

diff --git a/...s/javascore/lightclients/tendermint/src/test/java/ibc/tendermint/LightClientTestBase.java b/...s/javascore/lightclients/tendermint/src/test/java/ibc/tendermint/LightClientTestBase.java
@@ -101,7 +101,7 @@ protected void setup() throws Exception {
         contextMock.when(() -> Context.getBlockTimestamp())
                 .thenReturn(System.currentTimeMillis() * 1000 + (sm.getBlock().getHeight() * 2_000_000));
 
-        client = sm.deploy(owner, TendermintLightClient.class, ibcHandler.getAddress());
+        client = sm.deploy(owner, TendermintLightClient.class, ibcHandler.getAddress(), false);
 
         clientSpy = (TendermintLightClient) spy(client.getInstance());
         client.setInstance(clientSpy);