Function AssertColumns

pkg/gosqlx/testing/testing.go:181–205

AssertColumns asserts that the SQL selects the expected columns. This extracts column names from SELECT statements and compares them (order-independent). Example: testing.AssertColumns(t, "SELECT id, name, email FROM users", []string{"id", "name", "email"})

(t TestingT, sql string, expectedColumns []string)

//	    "SELECT id, name, email FROM users",
//	    []string{"id", "name", "email"})
func AssertColumns(t TestingT, sql string, expectedColumns []string) bool {
	t.Helper()

	astNode, err := gosqlx.Parse(sql)
	if err != nil {
		t.Errorf("Failed to parse SQL for column extraction:\n  SQL: %s\n  Error: %v",
			truncateSQL(sql), err)
		return false
	}

	columns := extractColumns(astNode)

	// Sort both slices for comparison
	sort.Strings(columns)
	expectedSorted := make([]string, len(expectedColumns))
	copy(expectedSorted, expectedColumns)
	sort.Strings(expectedSorted)

	if !stringSlicesEqual(columns, expectedSorted) {
		t.Errorf("SQL column references do not match expected:\n  SQL: %s\n  Expected: %v\n  Got: %v",
			truncateSQL(sql), expectedColumns, columns)
		return false
	}
	return true
}

// AssertParsesTo asserts that SQL parses to a specific AST statement type.
// This is useful for verifying that SQL is interpreted as the expected statement type.