split tests

Author: togami2864

tests/sqlparser_bigquery.rs

@@ -51,7 +51,7 @@ fn parse_byte_literal() {
 }
 
 #[test]
-fn parse_struct_literal() {
+fn parse_tuple_struct_literal() {
     // tuple syntax: https://cloud.google.com/bigquery/docs/reference/standard-sql/data-types#tuple_syntax
     // syntax: (expr1, expr2 [, ... ])
     let sql = "SELECT (1, 2, 3), (1, 1.0, '123', true)";
@@ -74,9 +74,12 @@ fn parse_struct_literal() {
         ]),
         expr_from_projection(&select.projection[1])
     );
+}
 
-    // // typeless struct syntax https://cloud.google.com/bigquery/docs/reference/standard-sql/data-types#typeless_struct_syntax
-    // // syntax: STRUCT( expr1 [AS field_name] [, ... ])
+#[test]
+fn parse_typeless_struct_syntax() {
+    // typeless struct syntax https://cloud.google.com/bigquery/docs/reference/standard-sql/data-types#typeless_struct_syntax
+    // syntax: STRUCT( expr1 [AS field_name] [, ... ])
     let sql = "SELECT STRUCT(1, 2, 3), STRUCT('abc'), STRUCT(1, t.str_col), STRUCT(1 AS a, 'abc' AS b), STRUCT(str_col AS abc)";
     let select = bigquery().verified_only_select(sql);
     assert_eq!(5, select.projection.len());
@@ -156,12 +159,43 @@ fn parse_struct_literal() {
         },
         expr_from_projection(&select.projection[4])
     );
-    // // typed struct syntax https://cloud.google.com/bigquery/docs/reference/standard-sql/data-types#typed_struct_syntax
-    // // syntax: STRUCT<[field_name] field_type, ...>( expr1 [, ... ])
+}
+
+#[test]
+fn parse_typed_struct_syntax() {
+    // typed struct syntax https://cloud.google.com/bigquery/docs/reference/standard-sql/data-types#typed_struct_syntax
+    // syntax: STRUCT<[field_name] field_type, ...>( expr1 [, ... ])
     // let sql = r#"SELECT STRUCT<int64>(5), STRUCT<x int64, y string>(1, t.str_col), STRUCT<int64>(int_col)"#;
     let sql = r#"SELECT STRUCT<BOOL>(true), STRUCT<BYTES>(B'abc')"#;
     let select = bigquery().verified_only_select(sql);
     assert_eq!(2, select.projection.len());
+    assert_eq!(
+        &Expr::Struct {
+            expr: Box::new(StructExpr(vec![ExprWithFieldName {
+                expr: Expr::Value(Value::Boolean(true)),
+                field_name: None
+            }])),
+            type_ann: Some(vec![StructTypeAnn {
+                field_name: None,
+                field_type: StructFieldType::Bool
+            }])
+        },
+        expr_from_projection(&select.projection[0])
+    );
+
+    assert_eq!(
+        &Expr::Struct {
+            expr: Box::new(StructExpr(vec![ExprWithFieldName {
+                expr: Expr::Value(Value::SingleQuotedByteStringLiteral("abc".to_string())),
+                field_name: None
+            }])),
+            type_ann: Some(vec![StructTypeAnn {
+                field_name: None,
+                field_type: StructFieldType::Bytes
+            }])
+        },
+        expr_from_projection(&select.projection[1])
+    );
 
     let sql = r#"SELECT STRUCT<DATE>("2011-05-05"), STRUCT<DATETIME>(DATETIME(2008, 12, 25, 05, 30, 00)), STRUCT<FLOAT64>(5.0), STRUCT<INT64>(1)"#;
     let select = bigquery().verified_only_select(sql);
@@ -175,6 +209,7 @@ fn parse_struct_literal() {
     let select = bigquery().verified_only_select(sql);
     assert_eq!(3, select.projection.len());
 
+    // TODO: support
     // let sql = r#"SELECT STRUCT<NUMERIC>(NUMERIC '1'), STRUCT<BIGNUMERIC>(BIGNUMERIC '1')"#;
     // let select = bigquery().verified_only_select(sql);
     // assert_eq!(1, select.projection.len());