feat: support CreateEnumStmt

Author: cvng

crates/codegen/src/get_node_properties.rs

@@ -559,6 +559,12 @@ fn custom_handlers(node: &Node) -> TokenStream {
  tokens.push(TokenProperty::from(Token::Authorization));
  }
  },
+ "CreateEnumStmt" => quote! {
+ tokens.push(TokenProperty::from(Token::Create));
+ tokens.push(TokenProperty::from(Token::TypeP));
+ tokens.push(TokenProperty::from(Token::As));
+ tokens.push(TokenProperty::from(Token::EnumP));
+ },
  _ => quote! {},
  }
 }

crates/parser/src/codegen.rs

@@ -66,10 +66,7 @@ mod tests {
 
  debug!("selected node: {:#?}", node_graph[node_index]);
 
- assert!(node_graph[node_index]
- .properties
- .iter()
- .all(|p| { expected.contains(p) }));
+ assert_eq!(node_graph[node_index].properties, expected);
  assert_eq!(node_graph[node_index].properties.len(), expected.len());
  }
 
@@ -140,4 +137,21 @@ mod tests {
  ],
  )
  }
+
+ #[test]
+ fn test_create_enum() {
+ test_get_node_properties(
+ "create type status as enum ('open', 'closed');",
+ SyntaxKind::CreateEnumStmt,
+ vec![
+ TokenProperty::from(SyntaxKind::Create),
+ TokenProperty::from(SyntaxKind::TypeP),
+ TokenProperty::from(SyntaxKind::As),
+ TokenProperty::from(SyntaxKind::EnumP),
+ TokenProperty::from("status".to_string()),
+ TokenProperty::from("open".to_string()),
+ TokenProperty::from("closed".to_string()),
+ ],
+ )
+ }
 }

crates/parser/tests/data/statements/valid/0037.sql

@@ -0,0 +1 @@
+CREATE TYPE bug_status AS ENUM ('new', 'open', 'closed');