Function loadSQLTestFiles

pkg/sql/integration_test.go:49–98

loadSQLTestFiles loads all SQL files from testdata directory

(t *testing.T, rootPath string)

// loadSQLTestFiles loads all SQL files from testdata directory
func loadSQLTestFiles(t *testing.T, rootPath string) []SQLTestFile {
	var files []SQLTestFile

	dialects := []string{"postgresql", "mysql", "mssql", "oracle", "real_world"}

	for _, dialect := range dialects {
		dialectPath := filepath.Join(rootPath, "testdata", dialect)

		err := filepath.WalkDir(dialectPath, func(path string, d fs.DirEntry, err error) error {
			if err != nil {
				return err
			}

			if d.IsDir() || !strings.HasSuffix(path, ".sql") {
				return nil
			}

			content, err := os.ReadFile(path)
			if err != nil {
				t.Logf("Warning: Could not read file %s: %v", path, err)
				return nil
			}

			// Extract complexity from comment
			complexity := "Medium"
			contentStr := string(content)
			if strings.Contains(contentStr, "Complexity: Simple") {
				complexity = "Simple"
			} else if strings.Contains(contentStr, "Complexity: Complex") {
				complexity = "Complex"
			}

			files = append(files, SQLTestFile{
				Path:       path,
				Dialect:    dialect,
				Name:       filepath.Base(path),
				Complexity: complexity,
				Content:    contentStr,
			})

			return nil
		})

		if err != nil {
			t.Logf("Warning: Could not walk directory %s: %v", dialectPath, err)
		}
	}

	return files
}

// extractSQLStatement extracts the main SQL statement from a file (ignoring comments)
func extractSQLStatement(content string) string {