// src/lib/credit-utils.ts
export function calculateCreditsForPlan(planId: 'free' | 'pro' | 'lifetime'): number {
  if (planId === 'pro') return 1000;
  if (planId === 'lifetime') return 50000;
  return 10;
}

// src/lib/credit-utils.test.ts
import { describe, it, expect } from 'vitest';
import { calculateCreditsForPlan } from './credit-utils';

// 'describe' 定义了一个测试套件
describe('Credit Calculation Utilities', () => {

  // 'it' 或 'test' 定义了一个单独的测试用例
  it('should return 1000 credits for pro plan', () => {
    // 'expect' 是断言
    expect(calculateCreditsForPlan('pro')).toBe(1000);
  });

  it('should return 10 credits for free plan', () => {
    expect(calculateCreditsForPlan('free')).toBe(10);
  });

  it('should return 50000 credits for lifetime plan', () => {
    expect(calculateCreditsForPlan('lifetime')).toBe(50000);
  });

});

// src/components/forms/NewsletterForm.test.tsx
import { render, screen, fireEvent } from '@testing-library/react';
import { describe, it, expect, vi } from 'vitest';
import { NewsletterForm } from './NewsletterForm';

// 模拟 Server Action
// 我们不希望在测试中真的去调用 Resend API
vi.mock('@/actions/newsletter.actions', () => ({
  subscribeToNewsletter: vi.fn(),
}));
import { subscribeToNewsletter } from '@/actions/newsletter.actions';

describe('NewsletterForm', () => {
  it('should render the form', () => {
    render(<NewsletterForm />);

    // 检查表单和按钮是否已渲染
    expect(screen.getByPlaceholderText('your@email.com')).toBeInTheDocument();
    expect(screen.getByRole('button', { name: /Subscribe/i })).toBeInTheDocument();
  });

  it('should show an error message for an invalid email', async () => {
    // 模拟 Server Action 返回一个错误
    (subscribeToNewsletter as vi.Mock).mockResolvedValueOnce({
      error: 'Invalid email.',
    });

    render(<NewsletterForm />);

    const input = screen.getByPlaceholderText('your@email.com');
    const button = screen.getByRole('button', { name: /Subscribe/i });

    // 模拟用户输入
    await fireEvent.change(input, { target: { value: 'not-an-email' } });
    // 模拟用户点击
    await fireEvent.click(button);

    // 等待并断言错误消息出现
    // 'findByText' 是 'getByText' 的异步版本，它会等待 UI 更新
    expect(await screen.findByText('Invalid email.')).toBeInTheDocument();
  });
});

// tests/e2e/subscription.spec.ts
import { test, expect } from '@playwright/test';
import { db } from '@/db'; // 假设 E2E 测试可以访问数据库
import { users } from '@/db/schema';
import { eq } from 'drizzle-orm';

const TEST_USER_EMAIL = `test-user-${Date.now()}@example.com`;
const TEST_USER_PASSWORD = 'Password123';

test.describe('Subscription Flow', () => {

  // 在每个测试用例运行前，重置数据库状态
  test.beforeEach(async () => {
    // 1. 清理：确保测试用户不存在
    await db.delete(users).where(eq(users.email, TEST_USER_EMAIL));
  });

  test('A new user can register, purchase a Pro plan, and see their credits', async ({ page }) => {

    // 步骤 1: 注册
    await page.goto('/register');
    await page.getByLabel('Email').fill(TEST_USER_EMAIL);
    await page.getByLabel('Password').fill(TEST_USER_PASSWORD);
    await page.getByRole('button', { name: 'Register' }).click();

    // 应该跳转到了仪表盘
    await expect(page.getByText('Welcome')).toBeVisible();

    // 步骤 2: 导航到定价页
    await page.getByRole('link', { name: 'Pricing' }).click();
    await expect(page.getByText('Pro Plan')).toBeVisible();

    // 步骤 3: 拦截 Stripe Checkout
    await page.route('**/api/stripe/checkout', (route) => {
      // (这是简化的 Mock, 实际上你可能会拦截 createCheckoutSession 的 Server Action)
      // 假设点击按钮会调用一个 API 或 Action，我们拦截它
      // 更简单的方法是拦截对 Stripe 的 *外部* 调用
      console.log('Intercepted Stripe checkout request');
    });

    // 步骤 4: 点击升级按钮 (来自 Pro Plan 卡片)
    // 假设卡片在 `[data-testid="plan-pro"]` 内
    await page.locator('[data-testid="plan-pro"] button').click();

    // --- 关键的 Mock 流程 ---
    // 在一个真实的 E2E 测试中，点击后会跳转到 Stripe。
    // 我们将在这里“作弊”，模拟支付成功。

    // 模拟 1: Webhook 处理器被触发
    // E2E 测试直接更新数据库，假装 Webhook 已经完成
    const testUser = await db.query.users.findFirst({ where: eq(users.email, TEST_USER_EMAIL) });

    // 假装 Stripe Webhook 成功了，并授予了积分
    await db.update(users)
      .set({
        subscriptionStatus: 'pro',
        totalCredits: 1000 // (来自 price-config)
      })
      .where(eq(users.id, testUser!.id));

    // 模拟 2: 用户被重定向回成功页面
    await page.goto('/dashboard?payment=success');

    // 步骤 5: 验证结果
    // 刷新页面以获取最新的服务器渲染数据
    await page.reload();

    // 断言：用户现在是 Pro 计划
    await expect(page.getByText('Plan: Pro')).toBeVisible();
    // 断言：用户获得了积分
    await expect(page.getByText(/Credits: 1000/i)).toBeVisible();
  });
});

-- db/tests/test_schema.sql
BEGIN;
-- 载入 pgTAP 扩展
CREATE EXTENSION IF NOT EXISTS pgtap;

-- 开始测试
SELECT plan(3); -- 声明我们有 3 个测试

-- 测试 1: 检查 'users' 表是否存在
SELECT has_table('public', 'users', 'Table "users" should exist.');

-- 测试 2: 检查 'users' 表是否有 'total_credits' 列
SELECT has_col('public', 'users', 'total_credits', 'Column "users.total_credits" should exist.');

-- 测试 3: 检查 'total_credits' 的 CHECK 约束是否有效
-- 尝试插入一个非法数据 (积分 < 0)
-- 'throws_ok' 期望这个操作失败
SELECT throws_ok(
  $$
    INSERT INTO users (id, email, total_credits)
    VALUES ('test-id', 'check-test@example.com', -50)
  $$,
  '23514', -- 23514 是 PostgreSQL 中 check_violation 的错误码
  'CHECK constraint for total_credits >= 0 should be active.'
);

-- 完成测试
SELECT * FROM finish();
ROLLBACK;