🧪 Add state var to test number of contract decls

ChmielewskiKamil · Dec 17, 2024 · 7c6c887 · 7c6c887
1 parent 2226ca2
commit 7c6c887
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/parser/parser_test.go b/parser/parser_test.go
@@ -186,9 +186,13 @@ func Test_ParseContractDeclaration(t *testing.T) {
 	src := `
     contract MyContract is BaseContract {
         uint256 public myVar;
+
         function myFunction() public view returns (uint256) {
+            uint256 balance;
             return myVar;
         }
+
+        function test() public {}
     }
     `
 
@@ -224,8 +228,8 @@ func Test_ParseContractDeclaration(t *testing.T) {
 		t.Fatalf("Expected ContractBody, got nil")
 	}
 
-	if len(body.Declarations) != 2 {
-		t.Fatalf("Expected 2 declarations in the contract body, got %d", len(body.Declarations))
+	if len(body.Declarations) != 3 {
+		t.Fatalf("Expected 3 declarations in the contract body, got %d", len(body.Declarations))
 	}
 
 	// Verify state variable