Add test to check type narrowing query

transaction-ballerina/tests/retry_transaction_stmt.bal

@@ -312,3 +312,26 @@ function getPrevInfoInNested() returns string|error {
     }
     return str;
 }
+
+@test:Config
+function testTypeNarrowingQueryWithRetry() {
+    int? i = 3;
+    int|int[] result;
+
+    transaction {
+        if i is () {
+            any|error out = commit;
+        } else {
+            rollback;
+        }
+    }
+
+    retry {
+        result = i is int ?
+            from var _ in [1, 2]
+            where i + 2 == 5
+            select 2 : 2;
+    }
+
+    test:assertEquals([2, 2], result);
+}

transaction-ballerina/tests/transaction_stmt.bal

@@ -730,3 +730,23 @@ function testBreakWithinTransactionToOuterLoop() {
     test:assertEquals(str, "Loop continued with digit: 1 ->Loop continued with digit: 2 " +
     "->Loop continued with digit: 3 ->Loop broke with digit: 4");
 }
+
+@test:Config
+function testTypeNarrowingQueryWithTransaction() {
+    int? i = 3;
+    int|int[] result;
+
+    transaction {
+        if i is () {
+            result = 2;
+            any|error out = commit;
+        } else {
+            result = i is int ?
+                from var _ in [1, 2]
+                where i + 2 == 5
+                select 2 : 2;
+        }
+    }
+
+    test:assertEquals([2, 2], result);
+}