Updated codegen.rs, get_node_properties.rs

Author: mobley-trent

crates/codegen/src/get_node_properties.rs

@@ -181,6 +181,10 @@ fn custom_handlers(node: &Node) -> TokenStream {
                 tokens.push(TokenProperty::from(Token::GroupP));
                 tokens.push(TokenProperty::from(Token::By));
             }
+            if n.sort_clause.len() > 0 {
+                tokens.push(TokenProperty::from(Token::Order));
+                tokens.push(TokenProperty::from(Token::By));
+            }
             match n.op() {
                 protobuf::SetOperation::Undefined => {},
                 protobuf::SetOperation::SetopNone => {},

crates/parser/src/codegen.rs

@@ -106,6 +106,20 @@ mod tests {
         )
     }
 
+    #[test]
+    fn test_select_with_order_by() {
+        test_get_node_properties(
+            "SELECT a, b, c FROM table1 ORDER BY c;",
+            SyntaxKind::SelectStmt,
+            vec![
+                TokenProperty::from(SyntaxKind::Select),
+                TokenProperty::from(SyntaxKind::From),
+                TokenProperty::from(SyntaxKind::Order),
+                TokenProperty::from(SyntaxKind::By),
+            ],
+        )
+    }
+
     #[test]
     fn test_create_domain() {
         test_get_node_properties(