Function TestDMLUpdateReturning

tests/integration/dml_test.go:244–272

TestDMLUpdateReturning tests UPDATE ... RETURNING

(t *testing.T)

// TestDMLUpdateReturning tests UPDATE ... RETURNING
func TestDMLUpdateReturning(t *testing.T) {
	mustExec(t, testHarness.DuckgresDB, "CREATE TABLE dml_update_returning (id INTEGER, name TEXT, version INTEGER)")
	mustExec(t, testHarness.DuckgresDB, "INSERT INTO dml_update_returning VALUES (1, 'Test', 1)")

	t.Run("update_returning_star", func(t *testing.T) {
		skipIfKnown(t)
		rows, err := testHarness.DuckgresDB.Query("UPDATE dml_update_returning SET version = version + 1 WHERE id = 1 RETURNING *")
		if err != nil {
			t.Fatalf("Update RETURNING failed: %v", err)
		}
		defer func() { _ = rows.Close() }()

		if !rows.Next() {
			t.Error("Expected 1 row returned")
		}
		var id int
		var name string
		var version int
		if err := rows.Scan(&id, &name, &version); err != nil {
			t.Fatalf("Scan failed: %v", err)
		}
		if version != 2 {
			t.Errorf("Expected version=2, got %d", version)
		}
	})

	// Cleanup
	mustExec(t, testHarness.DuckgresDB, "DROP TABLE IF EXISTS dml_update_returning")
}

// TestDMLUpdateFromJoin tests UPDATE with FROM clause (join)
func TestDMLUpdateFromJoin(t *testing.T) {