fix: handle nested forloops codegeneration

EYBlockchain · Aug 12, 2022 · 87d9e8a · 87d9e8a
1 parent e56e050
commit 87d9e8a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/codeGenerators/orchestration/nodejs/toOrchestration.ts b/src/codeGenerators/orchestration/nodejs/toOrchestration.ts
@@ -169,7 +169,7 @@ export default function codeGenerator(node: any, options: any = {}): any {
           let condition = `${node.condition.leftExpression.name} ${node.condition.operator} ${node.condition.rightExpression.value};`
           let loopExpression = ` ${node.loopExpression.expression.rightHandSide.subExpression.name} ${node.loopExpression.expression.rightHandSide.operator}`;
           let loopBody;
-          if ( !!node.body.statements[0] && node.body.statements[0].nodeType == 'IfStatement')
+          if ( !!node.body.statements[0] && (node.body.statements[0].nodeType == 'IfStatement'  || node.body.statements[0].nodeType == 'ForStatement'))
           loopBody = `${codeGenerator(node.body)}`;
           else
           loopBody = `${codeGenerator(node.body.statements.statements)} 

diff --git a/test/contracts/for-for.zol b/test/contracts/for-for.zol
@@ -4,11 +4,11 @@ pragma solidity ^0.8.0;
 
 contract Test {
 
-secret uint private z;
+secret uint256 private z;
 
-function add(uint j) public {
-  uint i;
-  uint k;
+function add(uint256 j) public {
+  uint256 i;
+  uint256 k;
 for(i =0 ; i<5 ; i++) {
     for(k=0; k<2; k++) {
       z += j;