| package sql |
|
|
| import ( |
| "strings" |
| "testing" |
|
|
| "ccLoad/internal/model" |
| ) |
|
|
| func TestBuildLatestChannelSuccessQuery_UsesIndexedSeek(t *testing.T) { |
| query, args := buildLatestChannelSuccessQuery(map[int][]int{ |
| 3: {0}, |
| 1: {1}, |
| }, &model.LogFilter{LogSource: model.LogSourceProxy}) |
|
|
| upperQuery := strings.ToUpper(query) |
| if strings.Contains(upperQuery, "ROW_NUMBER()") || strings.Contains(upperQuery, "OVER (") { |
| t.Fatalf("latest success query must not rank all historical rows:\n%s", query) |
| } |
| if !strings.Contains(upperQuery, "ORDER BY TIME DESC, ID DESC LIMIT 1") { |
| t.Fatalf("latest success query must seek the latest indexed row per channel:\n%s", query) |
| } |
| if !strings.Contains(query, "channel_id = scope.channel_id") { |
| t.Fatalf("latest success query must be correlated by channel scope:\n%s", query) |
| } |
| if len(args) != 3 { |
| t.Fatalf("args=%v, want two scope channel ids plus log_source", args) |
| } |
| } |
|
|
| func TestBuildLatestEntrySuccessQuery_UsesIndexedSeek(t *testing.T) { |
| query, args := buildLatestEntrySuccessQuery(map[statsRequestKey]int{ |
| {channelID: 3, model: "gpt-4o"}: 0, |
| {channelID: 1, model: "claude"}: 1, |
| }, &model.LogFilter{ModelLike: "gpt", LogSource: model.LogSourceProxy}) |
|
|
| upperQuery := strings.ToUpper(query) |
| if strings.Contains(upperQuery, "ROW_NUMBER()") || strings.Contains(upperQuery, "OVER (") { |
| t.Fatalf("latest entry success query must not rank all historical rows:\n%s", query) |
| } |
| if !strings.Contains(upperQuery, "ORDER BY TIME DESC, ID DESC LIMIT 1") { |
| t.Fatalf("latest entry success query must seek the latest indexed row per channel/model:\n%s", query) |
| } |
| if !strings.Contains(query, "COALESCE(model, '') = scope.model") { |
| t.Fatalf("latest entry success query must be correlated by channel/model scope:\n%s", query) |
| } |
| if len(args) != 6 { |
| t.Fatalf("args=%v, want two channel/model pairs plus model_like and log_source", args) |
| } |
| } |
|
|
