Function TestInsertStatementSQL

pkg/sql/ast/sql_test.go:114–156

(t *testing.T)

}

func TestInsertStatementSQL(t *testing.T) {
	tests := []struct {
		name string
		stmt *InsertStatement
		want string
	}{
		{
			name: "simple insert",
			stmt: &InsertStatement{
				TableName: "users",
				Columns:   []Expression{&Identifier{Name: "name"}, &Identifier{Name: "email"}},
				Values:    [][]Expression{{&LiteralValue{Value: "Alice", Type: "STRING"}, &LiteralValue{Value: "alice@example.com", Type: "STRING"}}},
			},
			want: "INSERT INTO users (name, email) VALUES ('Alice', 'alice@example.com')",
		},
		{
			name: "multi-row insert",
			stmt: &InsertStatement{
				TableName: "users",
				Columns:   []Expression{&Identifier{Name: "name"}},
				Values:    [][]Expression{{&LiteralValue{Value: "Alice", Type: "STRING"}}, {&LiteralValue{Value: "Bob", Type: "STRING"}}},
			},
			want: "INSERT INTO users (name) VALUES ('Alice'), ('Bob')",
		},
		{
			name: "insert with on conflict do nothing",
			stmt: &InsertStatement{
				TableName:  "users",
				Columns:    []Expression{&Identifier{Name: "email"}},
				Values:     [][]Expression{{&LiteralValue{Value: "a@b.com", Type: "STRING"}}},
				OnConflict: &OnConflict{Target: []Expression{&Identifier{Name: "email"}}, Action: OnConflictAction{DoNothing: true}},
			},
			want: "INSERT INTO users (email) VALUES ('a@b.com') ON CONFLICT (email) DO NOTHING",
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := tt.stmt.SQL(); got != tt.want {
				t.Errorf("SQL() =\n  %s\nwant:\n  %s", got, tt.want)
			}
		})
	}
}

func TestUpdateStatementSQL(t *testing.T) {
	stmt := &UpdateStatement{