[YQL-17700] More tests for TryMember (#4402)

Author: nepal

ydb/library/yql/tests/sql/sql2yql/canondata/result.json

@@ -6152,6 +6152,13 @@
             "uri": "https://{canondata_backend}/1784117/d56ae82ad9d30397a41490647be1bd2124718f98/resource.tar.gz#test_sql2yql.test_expr-to_sorted_set_vartuple_key_/sql.yql"
         }
     ],
+    "test_sql2yql.test[expr-try_member]": [
+        {
+            "checksum": "11bd9037e4c2efde4307441e96c6d3a2",
+            "size": 9904,
+            "uri": "https://{canondata_backend}/1946324/d9bbf763189fff6203bd54ad96c51e1ebd23dffc/resource.tar.gz#test_sql2yql.test_expr-try_member_/sql.yql"
+        }
+    ],
     "test_sql2yql.test[expr-type_as_key]": [
         {
             "checksum": "33f4e933fbd02cb6bcb6d0c1fec847f5",
@@ -24639,6 +24646,13 @@
             "uri": "https://{canondata_backend}/1936842/9dd3254ebbd843324a90765405d3999550dc8d63/resource.tar.gz#test_sql_format.test_expr-to_sorted_set_vartuple_key_/formatted.sql"
         }
     ],
+    "test_sql_format.test[expr-try_member]": [
+        {
+            "checksum": "3bedd6cf1a866afb6bb2f7acaafe5505",
+            "size": 1414,
+            "uri": "https://{canondata_backend}/995452/62dcdc716d63823bd4c68a1e5002cc6c38164d34/resource.tar.gz#test_sql_format.test_expr-try_member_/formatted.sql"
+        }
+    ],
     "test_sql_format.test[expr-type_as_key]": [
         {
             "checksum": "06e10aa05a1864c2016fe5fcc32671bd",

ydb/library/yql/tests/sql/suites/expr/try_member.cfg

@@ -0,0 +1 @@
+providers yt

ydb/library/yql/tests/sql/suites/expr/try_member.sql

@@ -0,0 +1,53 @@
+/* postgres can not */
+/* syntax version 1 */
+/* yt can not */
+
+$s = <|a:1, b:2u/1u, c:Just(Just(1))|>;
+$js = Just($s);
+$es = Nothing(Struct<a:Int32, b:Uint32?, c:Int32??>?);
+
+-- result of TryMember() when member is not present is third argument
+-- (optional third arg if struct is optional, but third arg is not)
+select
+  TryMember($s, "z", 'qqq'),
+  TryMember($js, "z", null),
+  TryMember($es, "z", Just(Just('qqq'))),
+  TryMember($js, "z", 'zzz'),
+;
+
+
+-- fully equivalent to <struct>.<name>
+select
+  TryMember($s, "a", null),
+  TryMember($s, "b", null),
+  TryMember($s, "c", null),
+
+  TryMember($js, "a", null),
+  TryMember($js, "b", null),
+  TryMember($js, "c", null),
+
+  TryMember($es, "a", null),
+  TryMember($es, "b", null),
+  TryMember($es, "c", null),
+;
+
+-- TypeOf TryMember is type of third argument
+-- field type should either match third type exactly, or (if the third type is optional) 
+-- Optional(field) should be equal to third type
+select
+  TryMember($s, "a", 999),
+  TryMember($s, "a", Just(999)),
+  TryMember($s, "b", Just(999u)),
+  TryMember($s, "c", Just(Just(999))),
+  
+  TryMember($js, "a", 999),
+  TryMember($js, "a", Just(999)),
+  TryMember($js, "b", Just(999u)),
+  TryMember($js, "c", Just(Just(999))),
+  
+  TryMember($es, "a", 999),
+  TryMember($es, "a", Just(999)),
+  TryMember($es, "b", Just(999u)),
+  TryMember($es, "c", Just(Just(999))),
+;
+  

ydb/library/yql/tests/sql/yt_native_file/part6/canondata/result.json

@@ -1087,6 +1087,27 @@
             "uri": "https://{canondata_backend}/1923547/9db3e11e4addc3a65ee82684bfbe078b6795a57a/resource.tar.gz#test.test_expr-to_hashed_set_tuple_key-default.txt-Results_/results.txt"
         }
     ],
+    "test.test[expr-try_member--Debug]": [
+        {
+            "checksum": "b46b58ff6e8b11197426aca99cdc8cca",
+            "size": 1555,
+            "uri": "https://{canondata_backend}/1775319/a2ce5580a0be182068abee641ca2dc356687c009/resource.tar.gz#test.test_expr-try_member--Debug_/opt.yql"
+        }
+    ],
+    "test.test[expr-try_member--Plan]": [
+        {
+            "checksum": "b2a2eb5d6b0a138ee924c128fc7738ef",
+            "size": 1331,
+            "uri": "https://{canondata_backend}/1775319/a2ce5580a0be182068abee641ca2dc356687c009/resource.tar.gz#test.test_expr-try_member--Plan_/plan.txt"
+        }
+    ],
+    "test.test[expr-try_member--Results]": [
+        {
+            "checksum": "97747cdc452ea4f288e72ef9aeaf2c31",
+            "size": 13783,
+            "uri": "https://{canondata_backend}/1775319/a2ce5580a0be182068abee641ca2dc356687c009/resource.tar.gz#test.test_expr-try_member--Results_/results.txt"
+        }
+    ],
     "test.test[file-parse_file_in_select_as_int--Debug]": [
         {
             "checksum": "8b1263ef7639d56cbd615bee42452d1d",