wangjia
e023fb6579
免费版此前形同虚设:ConnectNode 每次连接发固定 daily_minutes×1min TTL、
从不扣减已用,重连即崭新 10 分钟,日额度从未强制。改为账户级(全设备共享)
真卡控 + 累加式看广告加时:
- migration 000020: usage_daily 加 ad_bonus_minutes(sqlite+mysql)。
当日额度 = plans.daily_minutes + ad_bonus_minutes;剩余 = 额度 − minutes_used。
- usage.store.AddAdBonusMinutes: 事务内累加封顶(替代布尔 MarkAdUnlocked),
返回新总额+本次实际加时;GetDay/GetUsageRange 补读 ad_bonus_minutes。
- usage.service.UnlockAd: verify+nonce 后 +adBonusPerAd(10) 封顶 adDailyBonusCeiling(120),
返回 granted+remaining;TodaySummary 加 MinutesCap/AdBonusMinutes。
- usage.ads DevVerifier: 放行式占位校验(nonce 防重放仍生效),接真 AdMob 前
让看广告加时流程可端到端跑通;main.go 按 ADS_DEV_MODE/默认装配。
- ads/unlock: 204 → 200 返回 {granted_minutes, minutes_remaining}。
- ConnectNode 免费门: 读 AccountDayMinutes(used,bonus) → remaining≤0 拒 QUOTA_EXHAUSTED,
否则 TTL=remaining(凭证到点硬切断兜底)。nodes.store 加 AccountDayMinutes。
- /me: 补 ad_bonus_minutes,quota_today_min 分母改 daily+bonus,加 quota_cap_min。
- quota.CheckFreeConnect 同步累加语义(免费基础 10 分钟不再需先看广告)。
- openapi + 契约/迁移版本/集成测试同步;新增 SQLite AddAdBonusMinutes 单测。
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
94 lines
2.9 KiB
Go
94 lines
2.9 KiB
Go
package store_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/wangjia/pangolin/server/internal/config"
|
|
"github.com/wangjia/pangolin/server/internal/store"
|
|
)
|
|
|
|
// TestSQLiteMigrateUpDown verifies the SQLite migration set applies cleanly,
|
|
// is idempotent, seeds correctly, and rolls back — no container required, so it
|
|
// runs in normal CI (unlike the MySQL integration test behind //go:build integration).
|
|
func TestSQLiteMigrateUpDown(t *testing.T) {
|
|
cfg := &config.Config{Driver: "sqlite", DSN: ":memory:"}
|
|
db, err := store.Open(cfg)
|
|
if err != nil {
|
|
t.Fatalf("store.Open(sqlite): %v", err)
|
|
}
|
|
defer db.Close()
|
|
|
|
// 1. MigrateUp.
|
|
if err := store.MigrateUp(db, "sqlite"); err != nil {
|
|
t.Fatalf("MigrateUp: %v", err)
|
|
}
|
|
v, dirty, err := store.MigrateVersion(db, "sqlite")
|
|
if err != nil {
|
|
t.Fatalf("MigrateVersion: %v", err)
|
|
}
|
|
if dirty {
|
|
t.Fatalf("schema dirty after MigrateUp")
|
|
}
|
|
if v != 20 {
|
|
t.Errorf("version = %d, want 20", v)
|
|
}
|
|
|
|
// 2. Core tables exist.
|
|
for _, tbl := range []string{
|
|
"users", "devices", "plans", "subscriptions", "code_batches", "codes",
|
|
"usage_daily", "audit_log", "providers", "nodes", "node_events",
|
|
"directory_version", "provision_idempotency", "replacements", "admins",
|
|
"connect_credentials", "usage_device_daily", "sessions", "usage_hourly", "usage_device_hourly",
|
|
} {
|
|
var name string
|
|
err := db.QueryRow(
|
|
`SELECT name FROM sqlite_master WHERE type='table' AND name=?`, tbl,
|
|
).Scan(&name)
|
|
if err != nil {
|
|
t.Errorf("table %q missing: %v", tbl, err)
|
|
}
|
|
}
|
|
|
|
// 3. Seed: 3 plans + directory_version singleton.
|
|
var plans int
|
|
if err := db.QueryRow(`SELECT COUNT(*) FROM plans`).Scan(&plans); err != nil {
|
|
t.Fatalf("count plans: %v", err)
|
|
}
|
|
if plans != 3 {
|
|
t.Errorf("plans seeded = %d, want 3", plans)
|
|
}
|
|
var dv int
|
|
if err := db.QueryRow(`SELECT version FROM directory_version WHERE id=1`).Scan(&dv); err != nil {
|
|
t.Errorf("directory_version singleton missing: %v", err)
|
|
}
|
|
|
|
// 4. Added columns from later migrations are present (000011 / 000013).
|
|
if _, err := db.Exec(`SELECT reality_prk, reality_short_id, sub_token, totp_enabled FROM nodes
|
|
LEFT JOIN users ON 0=1 LIMIT 0`); err != nil {
|
|
// Separate queries — the join above is just a cheap column-existence probe.
|
|
if _, e := db.Exec(`SELECT reality_prk, reality_short_id FROM nodes LIMIT 0`); e != nil {
|
|
t.Errorf("nodes reality cols missing: %v", e)
|
|
}
|
|
if _, e := db.Exec(`SELECT sub_token, totp_secret_enc, totp_enabled FROM users LIMIT 0`); e != nil {
|
|
t.Errorf("users totp cols missing: %v", e)
|
|
}
|
|
}
|
|
|
|
// 5. Idempotent second MigrateUp.
|
|
if err := store.MigrateUp(db, "sqlite"); err != nil {
|
|
t.Fatalf("MigrateUp (idempotent): %v", err)
|
|
}
|
|
|
|
// 6. MigrateDown returns to baseline.
|
|
if err := store.MigrateDown(db, "sqlite"); err != nil {
|
|
t.Fatalf("MigrateDown: %v", err)
|
|
}
|
|
v2, _, err := store.MigrateVersion(db, "sqlite")
|
|
if err != nil {
|
|
t.Fatalf("MigrateVersion after down: %v", err)
|
|
}
|
|
if v2 != 0 {
|
|
t.Errorf("version after down = %d, want 0", v2)
|
|
}
|
|
}
|