Function TestBug2_MySQLValuesHelperBasic

pkg/sql/parser/uat_bugs_test.go:151–196

--------------------------------------------------------------------------- Bug 2 - MySQL VALUES() helper (MEDIUM) INSERT INTO t (id, name) VALUES (1,'Alice') ON DUPLICATE KEY UPDATE name=VALUES(name) used to fail with E2001. VALUES(col) is MySQL's way of referencing the attempted-to-insert value in

(t *testing.T)

// ---------------------------------------------------------------------------

func TestBug2_MySQLValuesHelperBasic(t *testing.T) {
	sql := "INSERT INTO users (id, name) VALUES (1, 'Alice') ON DUPLICATE KEY UPDATE name=VALUES(name)"
	tkz := tokenizer.GetTokenizer()
	defer tokenizer.PutTokenizer(tkz)

	tokens, err := tkz.Tokenize([]byte(sql))
	if err != nil {
		t.Fatalf("tokenize failed: %v", err)
	}

	p := parser.GetParser()
	defer parser.PutParser(p)

	tree, err := p.ParseFromModelTokens(tokens)
	if err != nil {
		t.Fatalf("Bug 2 regression: ON DUPLICATE KEY UPDATE VALUES() failed: %v", err)
	}

	if len(tree.Statements) != 1 {
		t.Fatalf("expected 1 statement, got %d", len(tree.Statements))
	}

	ins, ok := tree.Statements[0].(*ast.InsertStatement)
	if !ok {
		t.Fatalf("expected *ast.InsertStatement, got %T", tree.Statements[0])
	}
	if ins.OnDuplicateKey == nil {
		t.Fatal("expected OnDuplicateKey to be set")
	}
	if len(ins.OnDuplicateKey.Updates) != 1 {
		t.Fatalf("expected 1 update expression, got %d", len(ins.OnDuplicateKey.Updates))
	}

	// The RHS of the assignment must be a FunctionCall named "VALUES"
	fn, ok := ins.OnDuplicateKey.Updates[0].Value.(*ast.FunctionCall)
	if !ok {
		t.Fatalf("expected *ast.FunctionCall for VALUES(), got %T",
			ins.OnDuplicateKey.Updates[0].Value)
	}
	if !strings.EqualFold(fn.Name, "VALUES") {
		t.Errorf("expected function name VALUES, got %q", fn.Name)
	}
	if len(fn.Arguments) != 1 {
		t.Fatalf("expected VALUES() to have 1 argument, got %d", len(fn.Arguments))
	}
}

func TestBug2_MySQLValuesHelperMultipleColumns(t *testing.T) {
	sql := "INSERT INTO users (id, name, email) VALUES (1, 'Alice', 'a@b.com') " +