Function TestSQLAnalyzer_ScoreCalculation

cmd/gosqlx/cmd/sql_analyzer_test.go:341–411

(t *testing.T)

}

func TestSQLAnalyzer_ScoreCalculation(t *testing.T) {
	testCases := []struct {
		name        string
		sql         string
		expectRange func(score int) bool
	}{
		{
			name: "Perfect query",
			sql:  "SELECT name, email FROM users WHERE active = true",
			expectRange: func(score int) bool {
				return score >= 90 // Should be A grade
			},
		},
		{
			name: "Query with medium issues",
			sql:  "SELECT * FROM users",
			expectRange: func(score int) bool {
				return score >= 60 && score < 90 // B-C grade range
			},
		},
		{
			name: "Query with SELECT *",
			sql:  "SELECT * FROM users",
			expectRange: func(score int) bool {
				return score >= 50 && score < 90 // Should be B-C grade due to SELECT * issues
			},
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			// Parse and analyze
			tkz := tokenizer.GetTokenizer()
			defer tokenizer.PutTokenizer(tkz)

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

			p := parser.NewParser()
			astObj := ast.NewAST()
			defer ast.ReleaseAST(astObj)

			result, err := p.ParseFromModelTokens(tokens)
			if err != nil {
				t.Fatalf("Parsing failed: %v", err)
			}

			astObj.Statements = result.Statements

			analyzer := NewSQLAnalyzer()
			report, err := analyzer.Analyze(astObj)
			if err != nil {
				t.Fatalf("Analysis failed: %v", err)
			}

			// Verify score is in expected range