Prmbr/scripts/test-pricing-page-filtering.ts
2025-08-05 23:29:32 +08:00

123 lines
4.0 KiB
TypeScript

import { PrismaClient } from '@prisma/client'
import { SubscriptionService } from '../src/lib/subscription-service'
import { isPlanFree } from '../src/lib/subscription-utils'
const prisma = new PrismaClient()
async function testPricingPageFiltering() {
console.log('🧪 Testing pricing page filtering logic...')
try {
// 1. 获取所有套餐
console.log('\n1. Getting all available plans...')
const allPlans = await SubscriptionService.getAvailablePlans()
console.log(`📊 Found ${allPlans.length} active plans:`)
allPlans.forEach(plan => {
console.log(` - ${plan.displayName} (${plan.id}):`)
console.log(` * Price: $${plan.price}`)
console.log(` * Stripe Price ID: ${plan.stripePriceId || 'Not set'}`)
console.log(` * Is Free: ${isPlanFree(plan)}`)
})
// 2. 模拟定价页面的过滤逻辑
console.log('\n2. Testing pricing page filtering logic...')
// 模拟不同用户场景
const testScenarios = [
{ name: 'Anonymous User', userData: null },
{ name: 'Free User', userData: { subscriptionPlanId: 'free' } },
{ name: 'Pro User', userData: { subscriptionPlanId: 'pro' } }
]
for (const scenario of testScenarios) {
console.log(`\n📋 Scenario: ${scenario.name}`)
const filteredPlans = allPlans.filter(plan => {
// 免费套餐总是显示
if (isPlanFree(plan)) return true
// 用户当前套餐总是显示
if (scenario.userData && scenario.userData.subscriptionPlanId === plan.id) return true
// 其他套餐必须有 stripePriceId 才能显示(可订阅)
return plan.stripePriceId && plan.stripePriceId.trim() !== ''
})
console.log(` Visible plans: ${filteredPlans.length}`)
filteredPlans.forEach(plan => {
const isCurrent = scenario.userData?.subscriptionPlanId === plan.id
const canSubscribe = !isPlanFree(plan) && !!plan.stripePriceId && !isCurrent
console.log(` - ${plan.displayName}:`)
console.log(` * Current plan: ${isCurrent}`)
console.log(` * Can subscribe: ${canSubscribe}`)
console.log(` * Button action: ${getButtonAction(plan, isCurrent, canSubscribe)}`)
})
}
// 3. 检查套餐配置问题
console.log('\n3. Checking for potential issues...')
const plansWithoutPriceId = allPlans.filter(plan =>
!isPlanFree(plan) && (!plan.stripePriceId || plan.stripePriceId.trim() === '')
)
if (plansWithoutPriceId.length > 0) {
console.log(`⚠️ Found ${plansWithoutPriceId.length} paid plans without Stripe Price ID:`)
plansWithoutPriceId.forEach(plan => {
console.log(` - ${plan.displayName} ($${plan.price}) - will not be visible to non-subscribers`)
})
} else {
console.log('✅ All paid plans have valid Stripe Price IDs')
}
// 4. 验证免费套餐
const freePlans = allPlans.filter(isPlanFree)
console.log(`\n4. Free plans validation:`)
console.log(` Found ${freePlans.length} free plans`)
freePlans.forEach(plan => {
console.log(` - ${plan.displayName}: Always visible, no subscription button`)
})
console.log('\n🎉 Pricing page filtering test completed!')
} catch (error) {
console.error('❌ Test failed:', error)
throw error
} finally {
await prisma.$disconnect()
}
}
function getButtonAction(plan: any, isCurrent: boolean, canSubscribe: boolean): string {
if (isPlanFree(plan)) {
return isCurrent ? 'Current Plan (disabled)' : 'No button'
}
if (isCurrent) {
return 'Current Plan (disabled)'
}
if (canSubscribe) {
return 'Subscribe Now'
}
return 'No button (no price ID)'
}
// 运行测试
if (require.main === module) {
testPricingPageFiltering()
.then(() => {
console.log('✅ All tests completed!')
process.exit(0)
})
.catch((error) => {
console.error('❌ Tests failed:', error)
process.exit(1)
})
}
export { testPricingPageFiltering }