usage_log_effective_model_index_test.go 780 B

12345678910111213141516171819
  1. package migrations
  2. import (
  3. "strings"
  4. "testing"
  5. "github.com/stretchr/testify/require"
  6. )
  7. func TestUsageLogEffectiveModelIndexesMigration(t *testing.T) {
  8. content, err := FS.ReadFile("226_add_usage_log_effective_model_indexes_notx.sql")
  9. require.NoError(t, err)
  10. sql := strings.Join(strings.Fields(string(content)), " ")
  11. require.Contains(t, sql, "CREATE INDEX CONCURRENTLY IF NOT EXISTS idx_usage_logs_effective_requested_model_created")
  12. require.Contains(t, sql, "(COALESCE(NULLIF(BTRIM(requested_model), ''), model)), created_at DESC, id DESC")
  13. require.Contains(t, sql, "CREATE INDEX CONCURRENTLY IF NOT EXISTS idx_usage_logs_effective_upstream_model_created")
  14. require.Contains(t, sql, "(COALESCE(NULLIF(BTRIM(upstream_model), ''), model)), created_at DESC, id DESC")
  15. }