ADDED DROP/ALTER USER/GROUP

internal/engine/ydb/catalog_tests/alter_group_test.go

@@ -0,0 +1,122 @@
+package ydb_test
+
+import (
+	"strings"
+	"testing"
+
+	"github.com/google/go-cmp/cmp"
+	"github.com/google/go-cmp/cmp/cmpopts"
+	"github.com/sqlc-dev/sqlc/internal/engine/ydb"
+	"github.com/sqlc-dev/sqlc/internal/sql/ast"
+)
+
+func TestAlterGroup(t *testing.T) {
+	tests := []struct {
+		stmt     string
+		expected ast.Node
+	}{
+		{
+			stmt: `ALTER GROUP admins RENAME TO superusers`,
+			expected: &ast.Statement{
+				Raw: &ast.RawStmt{
+					Stmt: &ast.AlterRoleStmt{
+						Role: &ast.RoleSpec{
+							Rolename: strPtr("admins"),
+							Roletype: ast.RoleSpecType(1),
+						},
+						Action: 1,
+						Options: &ast.List{
+							Items: []ast.Node{
+								&ast.DefElem{
+									Defname:   strPtr("rename"),
+									Defaction: ast.DefElemAction(1),
+									Arg:       &ast.String{Str: "superusers"},
+								},
+							},
+						},
+					},
+				},
+			},
+		},
+		{
+			stmt: `ALTER GROUP devs ADD USER alice, bob, carol`,
+			expected: &ast.Statement{
+				Raw: &ast.RawStmt{
+					Stmt: &ast.AlterRoleStmt{
+						Role: &ast.RoleSpec{
+							Rolename: strPtr("devs"),
+							Roletype: ast.RoleSpecType(1),
+						},
+						Action: 1,
+						Options: &ast.List{
+							Items: []ast.Node{
+								&ast.DefElem{
+									Defname:   strPtr("rolemembers"),
+									Defaction: ast.DefElemAction(3),
+									Arg: &ast.List{
+										Items: []ast.Node{
+											&ast.RoleSpec{Rolename: strPtr("alice"), Roletype: ast.RoleSpecType(1)},
+											&ast.RoleSpec{Rolename: strPtr("bob"), Roletype: ast.RoleSpecType(1)},
+											&ast.RoleSpec{Rolename: strPtr("carol"), Roletype: ast.RoleSpecType(1)},
+										},
+									},
+								},
+							},
+						},
+					},
+				},
+			},
+		},
+		{
+			stmt: `ALTER GROUP ops DROP USER dan, erin`,
+			expected: &ast.Statement{
+				Raw: &ast.RawStmt{
+					Stmt: &ast.AlterRoleStmt{
+						Role: &ast.RoleSpec{
+							Rolename: strPtr("ops"),
+							Roletype: ast.RoleSpecType(1),
+						},
+						Action: 1,
+						Options: &ast.List{
+							Items: []ast.Node{
+								&ast.DefElem{
+									Defname:   strPtr("rolemembers"),
+									Defaction: ast.DefElemAction(4),
+									Arg: &ast.List{
+										Items: []ast.Node{
+											&ast.RoleSpec{Rolename: strPtr("dan"), Roletype: ast.RoleSpecType(1)},
+											&ast.RoleSpec{Rolename: strPtr("erin"), Roletype: ast.RoleSpecType(1)},
+										},
+									},
+								},
+							},
+						},
+					},
+				},
+			},
+		},
+	}
+
+	p := ydb.NewParser()
+	for _, tc := range tests {
+		t.Run(tc.stmt, func(t *testing.T) {
+			stmts, err := p.Parse(strings.NewReader(tc.stmt))
+			if err != nil {
+				t.Fatalf("Ошибка парсинга запроса %q: %v", tc.stmt, err)
+			}
+			if len(stmts) == 0 {
+				t.Fatalf("Запрос %q не распарсен", tc.stmt)
+			}
+
+			diff := cmp.Diff(tc.expected, &stmts[0],
+				cmpopts.IgnoreFields(ast.RawStmt{}, "StmtLocation", "StmtLen"),
+				cmpopts.IgnoreFields(ast.DefElem{}, "Location"),
+				cmpopts.IgnoreFields(ast.RoleSpec{}, "Location"),
+				cmpopts.IgnoreFields(ast.A_Const{}, "Location"),
+			)
+			if diff != "" {
+				t.Errorf("Несовпадение AST (-ожидалось +получено):\n%s", diff)
+			}
+		})
+	}
+}

internal/engine/ydb/catalog_tests/alter_user_test.go

@@ -0,0 +1,153 @@
+package ydb_test
+
+import (
+	"strings"
+	"testing"
+
+	"github.com/google/go-cmp/cmp"
+	"github.com/google/go-cmp/cmp/cmpopts"
+	"github.com/sqlc-dev/sqlc/internal/engine/ydb"
+	"github.com/sqlc-dev/sqlc/internal/sql/ast"
+)
+
+func TestAlterUser(t *testing.T) {
+	tests := []struct {
+		stmt     string
+		expected ast.Node
+	}{
+		{
+			stmt: `ALTER USER alice RENAME TO queen`,
+			expected: &ast.Statement{
+				Raw: &ast.RawStmt{
+					Stmt: &ast.AlterRoleStmt{
+						Role: &ast.RoleSpec{
+							Rolename: strPtr("alice"),
+							Roletype: ast.RoleSpecType(1),
+						},
+						Action: 1,
+						Options: &ast.List{
+							Items: []ast.Node{
+								&ast.DefElem{
+									Defname: strPtr("rename"),
+									Arg:     &ast.String{Str: "queen"},
+									Defaction: ast.DefElemAction(1),
+								},
+							},
+						},
+					},
+				},
+			},
+		},
+		{
+			stmt: `ALTER USER bob LOGIN`,
+			expected: &ast.Statement{
+				Raw: &ast.RawStmt{
+					Stmt: &ast.AlterRoleStmt{
+						Role: &ast.RoleSpec{
+							Rolename: strPtr("bob"),
+							Roletype: ast.RoleSpecType(1),
+						},
+						Action: 1,
+						Options: &ast.List{
+							Items: []ast.Node{
+								&ast.DefElem{
+									Defname: strPtr("login"),
+									Arg:     &ast.Boolean{Boolval: true},
+								},
+							},
+						},
+					},
+				},
+			},
+		},
+		{
+			stmt: `ALTER USER charlie NOLOGIN`,
+			expected: &ast.Statement{
+				Raw: &ast.RawStmt{
+					Stmt: &ast.AlterRoleStmt{
+						Role: &ast.RoleSpec{
+							Rolename: strPtr("charlie"),
+							Roletype: ast.RoleSpecType(1),
+						},
+						Action: 1,
+						Options: &ast.List{
+							Items: []ast.Node{
+								&ast.DefElem{
+									Defname: strPtr("nologin"),
+									Arg:     &ast.Boolean{Boolval: false},
+								},
+							},
+						},
+					},
+				},
+			},
+		},
+		{
+			stmt: `ALTER USER dave PASSWORD 'qwerty'`,
+			expected: &ast.Statement{
+				Raw: &ast.RawStmt{
+					Stmt: &ast.AlterRoleStmt{
+						Role: &ast.RoleSpec{
+							Rolename: strPtr("dave"),
+							Roletype: ast.RoleSpecType(1),
+						},
+						Action: 1,
+						Options: &ast.List{
+							Items: []ast.Node{
+								&ast.DefElem{
+									Defname: strPtr("password"),
+									Arg:     &ast.String{Str: "qwerty"},
+								},
+							},
+						},
+					},
+				},
+			},
+		},
+		{
+			stmt: `ALTER USER elena HASH 'abc123'`,
+			expected: &ast.Statement{
+				Raw: &ast.RawStmt{
+					Stmt: &ast.AlterRoleStmt{
+						Role: &ast.RoleSpec{
+							Rolename: strPtr("elena"),
+							Roletype: ast.RoleSpecType(1),
+						},
+						Action: 1,
+						Options: &ast.List{
+							Items: []ast.Node{
+								&ast.DefElem{
+									Defname: strPtr("hash"),
+									Arg:     &ast.String{Str: "abc123"},
+								},
+							},
+						},
+					},
+				},
+			},
+		},
+	}
+
+	p := ydb.NewParser()
+	for _, tc := range tests {
+		t.Run(tc.stmt, func(t *testing.T) {
+			stmts, err := p.Parse(strings.NewReader(tc.stmt))
+			if err != nil {
+				t.Fatalf("Ошибка парсинга запроса %q: %v", tc.stmt, err)
+			}
+			if len(stmts) == 0 {
+				t.Fatalf("Запрос %q не рас��арсен", tc.stmt)
+			}
+
+			diff := cmp.Diff(tc.expected, &stmts[0],
+				cmpopts.IgnoreFields(ast.RawStmt{}, "StmtLocation", "StmtLen"),
+				cmpopts.IgnoreFields(ast.DefElem{}, "Location"),
+				cmpopts.IgnoreFields(ast.RoleSpec{}, "Location"),
+				cmpopts.IgnoreFields(ast.A_Const{}, "Location"),
+			)
+			if diff != "" {
+				t.Errorf("Несовпадение AST (-ожидалось +получено):\n%s", diff)
+			}
+		})
+	}
+}

internal/engine/ydb/catalog_tests/drop_role_test.go

@@ -0,0 +1,87 @@
+package ydb_test
+
+import (
+	"strings"
+	"testing"
+
+	"github.com/google/go-cmp/cmp"
+	"github.com/google/go-cmp/cmp/cmpopts"
+	"github.com/sqlc-dev/sqlc/internal/engine/ydb"
+	"github.com/sqlc-dev/sqlc/internal/sql/ast"
+)
+
+func TestDropRole(t *testing.T) {
+	tests := []struct {
+		stmt     string
+		expected ast.Node
+	}{
+		{
+			stmt: `DROP USER user1;`,
+			expected: &ast.Statement{
+				Raw: &ast.RawStmt{
+					Stmt: &ast.DropRoleStmt{
+						MissingOk: false,
+						Roles: &ast.List{
+							Items: []ast.Node{
+								&ast.RoleSpec{Rolename: strPtr("user1"), Roletype: ast.RoleSpecType(1)},
+							},
+						},
+					},
+				},
+			},
+		},
+		{
+			stmt: "DROP USER IF EXISTS admin, user2",
+			expected: &ast.Statement{
+				Raw: &ast.RawStmt{
+					Stmt: &ast.DropRoleStmt{
+						MissingOk: true,
+						Roles: &ast.List{
+							Items: []ast.Node{
+								&ast.RoleSpec{Rolename: strPtr("admin"), Roletype: ast.RoleSpecType(1)},
+								&ast.RoleSpec{Rolename: strPtr("user2"), Roletype: ast.RoleSpecType(1)},
+							},
+						},
+					},
+				},
+			},
+		},
+		{
+			stmt: "DROP GROUP team1, team2",
+			expected: &ast.Statement{
+				Raw: &ast.RawStmt{
+					Stmt: &ast.DropRoleStmt{
+						MissingOk: false,
+						Roles: &ast.List{
+							Items: []ast.Node{
+								&ast.RoleSpec{Rolename: strPtr("team1"), Roletype: ast.RoleSpecType(1)},
+								&ast.RoleSpec{Rolename: strPtr("team2"), Roletype: ast.RoleSpecType(1)},
+							},
+						},
+					},
+				},
+			},
+		},
+	}
+
+	p := ydb.NewParser()
+	for _, tc := range tests {
+		t.Run(tc.stmt, func(t *testing.T) {
+			stmts, err := p.Parse(strings.NewReader(tc.stmt))
+			if err != nil {
+				t.Fatalf("Error parsing %q: %v", tc.stmt, err)
+			}
+			if len(stmts) == 0 {
+				t.Fatalf("Statement %q was not parsed", tc.stmt)
+			}
+
+			diff := cmp.Diff(tc.expected, &stmts[0],
+				cmpopts.IgnoreFields(ast.RawStmt{}, "StmtLocation", "StmtLen"),
+				cmpopts.IgnoreFields(ast.RoleSpec{}, "Location"),
+			)
+			if diff != "" {
+				t.Errorf("AST mismatch for %q (-expected +got):\n%s", tc.stmt, diff)
+			}
+		})
+	}
+}