4a91954532
- Fix fileChecksum for 061 migration: use TrimSpace hash (66207e7a) instead of raw sha256sum (97bdd9a3), matching the actual runtime computation - Add 222b4a09 as accepted DB checksum for 061 migration - Add missing GetAPIKeyRateLimit/SetAPIKeyRateLimit/UpdateAPIKeyRateLimitUsage/ InvalidateAPIKeyRateLimit methods to mock BillingCache in test stubs - Fix NewBillingCacheService call in singleflight test (add apiKeyRepo param)
55 lines
1.7 KiB
Go
55 lines
1.7 KiB
Go
package repository
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestIsMigrationChecksumCompatible(t *testing.T) {
|
|
t.Run("054历史checksum可兼容", func(t *testing.T) {
|
|
ok := isMigrationChecksumCompatible(
|
|
"054_drop_legacy_cache_columns.sql",
|
|
"182c193f3359946cf094090cd9e57d5c3fd9abaffbc1e8fc378646b8a6fa12b4",
|
|
"82de761156e03876653e7a6a4eee883cd927847036f779b0b9f34c42a8af7a7d",
|
|
)
|
|
require.True(t, ok)
|
|
})
|
|
|
|
t.Run("054在未知文件checksum下不兼容", func(t *testing.T) {
|
|
ok := isMigrationChecksumCompatible(
|
|
"054_drop_legacy_cache_columns.sql",
|
|
"182c193f3359946cf094090cd9e57d5c3fd9abaffbc1e8fc378646b8a6fa12b4",
|
|
"0000000000000000000000000000000000000000000000000000000000000000",
|
|
)
|
|
require.False(t, ok)
|
|
})
|
|
|
|
t.Run("061历史checksum可兼容", func(t *testing.T) {
|
|
ok := isMigrationChecksumCompatible(
|
|
"061_add_usage_log_request_type.sql",
|
|
"08a248652cbab7cfde147fc6ef8cda464f2477674e20b718312faa252e0481c0",
|
|
"66207e7aa5dd0429c2e2c0fabdaf79783ff157fa0af2e81adff2ee03790ec65c",
|
|
)
|
|
require.True(t, ok)
|
|
})
|
|
|
|
t.Run("061第二个历史checksum可兼容", func(t *testing.T) {
|
|
ok := isMigrationChecksumCompatible(
|
|
"061_add_usage_log_request_type.sql",
|
|
"222b4a09c797c22e5922b6b172327c824f5463aaa8760e4f621bc5c22e2be0f3",
|
|
"66207e7aa5dd0429c2e2c0fabdaf79783ff157fa0af2e81adff2ee03790ec65c",
|
|
)
|
|
require.True(t, ok)
|
|
})
|
|
|
|
t.Run("非白名单迁移不兼容", func(t *testing.T) {
|
|
ok := isMigrationChecksumCompatible(
|
|
"001_init.sql",
|
|
"182c193f3359946cf094090cd9e57d5c3fd9abaffbc1e8fc378646b8a6fa12b4",
|
|
"82de761156e03876653e7a6a4eee883cd927847036f779b0b9f34c42a8af7a7d",
|
|
)
|
|
require.False(t, ok)
|
|
})
|
|
}
|