Test user defined functions too.

Author: dnwpark

tests/dbsetup/orm_qb.gel

@@ -618,4 +618,8 @@ type Link_Inh_A {
     };
 };
 
+function Read_Inh_A(x: Inh_A) -> int64 using (x.a ?? -1);
+function Read_Inh_A_Overload(x: Inh_A) -> int64 using (x.a ?? -1);
+function Read_Inh_A_Overload(x: Inh_AB) -> int64 using (x.ab ?? -1);
+
 }

tests/test_qb.py

@@ -2084,6 +2084,23 @@ def test_qb_is_type_as_function_arg_02(self):
             excluded_fields={'ab', 'ac', 'bc', 'abc', 'ab_ac'},
         )
 
+    def test_qb_is_type_as_function_arg_03(self):
+        # Test that exprs produced by is_ can be passed as function args to
+        # user defined function
+        from models.orm_qb import default
+
+        # Note, we do Inh_A[is Inh_B] since is_ currently pretends its return
+        # type is its argument type.
+        result = self.client.query(
+            default.Read_Inh_A(default.Inh_B.is_(default.Inh_A))
+        )
+        self.assertEqual(sorted(result), [4, 13, 17])
+
+        result = self.client.query(
+            default.Read_Inh_A_Overload(default.Inh_B.is_(default.Inh_A))
+        )
+        self.assertEqual(sorted(result), [6, 13, 20])
+
 
 class TestQueryBuilderModify(tb.ModelTestCase):
     """This test suite is for data manipulation using QB."""