import { DeepSeekService } from '../../../src/modules/ai/service/deepseek';
import { CoolCommException } from '@cool-midway/core';

describe('智能服务模块: 多租户 BYOK 配置隔离与 Token 成本风控', () => {
  let service: DeepSeekService;

  beforeEach(() => {
    delete process.env.DEEPSEEK_API_KEY;
    delete process.env.DEEPSEEK_BASE_URL;
    service = new DeepSeekService();
  });

  describe('1. 租户专属配置隔离 (Tenant BYOK)', () => {
    it('未单独配置的租户，获取配置时应优雅回退至平台公共默认配置', () => {
      const tenantConfig = service.getTenantConfig(88);
      expect(tenantConfig.baseUrl).toBe('https://api.deepseek.com');
      expect(tenantConfig.model).toBe('deepseek-chat');
      expect(service.hasTenantConfig(88)).toBe(false);
    });

    it('特定租户配置专属 API Key 与定制模型后，应严格与公共配置及其他租户隔离', () => {
      // 租户 101 配置独立 Key (BYOK)
      service.updateTenantConfig(101, {
        apiKey: 'sk-tenant-101-custom-key',
        model: 'deepseek-reasoner',
        baseUrl: 'https://gateway.company101.com',
      });

      expect(service.hasTenantConfig(101)).toBe(true);
      const config101 = service.getTenantConfig(101);
      expect(config101.apiKey).toBe('sk-tenant-101-custom-key');
      expect(config101.model).toBe('deepseek-reasoner');
      expect(config101.baseUrl).toBe('https://gateway.company101.com');

      // 租户 202 保持未受影响
      expect(service.hasTenantConfig(202)).toBe(false);
      const config202 = service.getTenantConfig(202);
      expect(config202.apiKey).toBeUndefined();
      expect(config202.baseUrl).toBe('https://api.deepseek.com');
    });
  });

  describe('2. 租户 Token 消耗度量与限流风控 (Token Metering & Quota)', () => {
    it('调用 recordTokenUsage 应准确累计指定租户的 Token 消耗流水', () => {
      service.recordTokenUsage(101, {
        promptTokens: 120,
        completionTokens: 80,
        totalTokens: 200,
      });

      service.recordTokenUsage(101, {
        promptTokens: 50,
        completionTokens: 50,
        totalTokens: 100,
      });

      const usage = service.getTenantTokenUsage(101);
      expect(usage.totalTokens).toBe(300);
      expect(usage.promptTokens).toBe(170);
      expect(usage.completionTokens).toBe(130);
      expect(usage.callCount).toBe(2);
    });

    it('当租户 Token 用量超出配额上限时，assertTenantTokenQuota 必须阻断抛错', () => {
      // 设置租户 101 最大配额 500 Token
      service.setTenantTokenQuota(101, 500);

      service.recordTokenUsage(101, { totalTokens: 480 });

      // 预估 10 Token，尚未超限 (490 <= 500)
      expect(() => service.assertTenantTokenQuota(101, 10)).not.toThrow();

      // 预估 30 Token，超出配额 (480 + 30 = 510 > 500)
      expect(() => service.assertTenantTokenQuota(101, 30)).toThrow(CoolCommException);
    });
  });
});