galaxy-opc-plugin 0.2.0 → 0.2.2

package/src/tools/media-tool.ts

@@ -46,6 +46,28 @@ const MediaSchema = Type.Union([
     action: Type.Literal("delete_content"),
     content_id: Type.String({ description: "内容 ID" }),
   }),
+  Type.Object({
+    action: Type.Literal("generate_content_brief"),
+    company_id: Type.String({ description: "公司 ID" }),
+    platform: Type.String({ description: "目标平台: 微信公众号/小红书/抖音/微博/知乎/B站" }),
+    topic: Type.String({ description: "主题关键词" }),
+  }),
+  Type.Object({
+    action: Type.Literal("submit_for_review"),
+    content_id: Type.String({ description: "内容 ID" }),
+  }),
+  Type.Object({
+    action: Type.Literal("approve_content"),
+    content_id: Type.String({ description: "内容 ID" }),
+    approved: Type.Boolean({ description: "是否通过: true=通过, false=需修改" }),
+    review_notes: Type.Optional(Type.String({ description: "审批备注/修改意见" })),
+    reviewer: Type.Optional(Type.String({ description: "审批人" })),
+  }),
+  Type.Object({
+    action: Type.Literal("content_analytics"),
+    company_id: Type.String({ description: "公司 ID" }),
+    month: Type.Optional(Type.String({ description: "月份 (YYYY-MM)，默认当月" })),
+  }),
 ]);
 
 type MediaParams = Static<typeof MediaSchema>;
@@ -90,7 +112,9 @@ export function registerMediaTool(api: OpenClawPluginApi, db: OpcDatabase): void
       label: "OPC 新媒体运营",
       description:
         "新媒体运营工具。操作: create_content(创建内容), list_content(内容列表), " +
-        "update_content(更新内容), content_calendar(发布日历), platform_guide(平台指南), delete_content(删除内容)",
+        "update_content(更新内容), content_calendar(发布日历), platform_guide(平台指南), " +
+        "delete_content(删除内容), generate_content_brief(AI内容策划摘要), " +
+        "submit_for_review(提交审批), approve_content(审批内容), content_analytics(内容分析)",
       parameters: MediaSchema,
       async execute(_toolCallId, params) {
         const p = params as MediaParams;
@@ -161,6 +185,137 @@ export function registerMediaTool(api: OpenClawPluginApi, db: OpcDatabase): void
               return json({ ok: true });
             }
 
+            case "generate_content_brief": {
+              // 获取公司信息
+              const company = db.queryOne("SELECT * FROM opc_companies WHERE id = ?", p.company_id) as Record<string, unknown> | null;
+              if (!company) return toolError(`公司 ${p.company_id} 不存在`, "COMPANY_NOT_FOUND");
+
+              // 获取 OPB Canvas
+              const canvas = db.queryOne("SELECT * FROM opc_opb_canvas WHERE company_id = ?", p.company_id) as Record<string, unknown> | null;
+
+              // 获取平台指南
+              const guide = PLATFORM_GUIDES[p.platform];
+
+              const brief = {
+                ok: true,
+                platform: p.platform,
+                topic: p.topic,
+                company_info: {
+                  name: company.name,
+                  industry: company.industry,
+                  description: company.description,
+                },
+                canvas_highlights: canvas ? {
+                  value_proposition: canvas.solution || canvas.unique_value,
+                  target_customer: canvas.target_customer,
+                  pain_point: canvas.pain_point,
+                } : null,
+                platform_guide: guide ?? null,
+                content_strategy: {
+                  title_suggestions: [
+                    `${p.topic} — 一人企业的实战经验分享`,
+                    `关于${p.topic}，你需要知道的3件事`,
+                    `${company.industry}行业 ${p.topic} 深度解析`,
+                  ],
+                  outline: [
+                    "1. 引入：痛点/现象描述（引起共鸣）",
+                    "2. 分析：为什么会这样（专业视角）",
+                    "3. 解决方案：具体可操作的方法",
+                    "4. 案例/数据：增加可信度",
+                    "5. 总结 + 行动号召（引导互动）",
+                  ],
+                  keywords: [p.topic, company.industry as string, "一人企业", "创业"],
+                  hashtags: guide ? [`#${p.topic}`, `#${company.industry}`, "#一人企业", "#创业干货"] : [`#${p.topic}`],
+                  platform_tips: guide ? guide.tips : [],
+                  best_publish_time: guide ? guide.best_time : "建议工作日晚间 20-22 点",
+                  format_recommendation: guide ? guide.format : "图文内容",
+                },
+              };
+
+              return json(brief);
+            }
+
+            case "submit_for_review": {
+              const content = db.queryOne("SELECT * FROM opc_media_content WHERE id = ?", p.content_id) as Record<string, unknown> | null;
+              if (!content) return toolError("内容不存在", "RECORD_NOT_FOUND");
+              db.execute(
+                "UPDATE opc_media_content SET status = 'pending_review', updated_at = ? WHERE id = ?",
+                new Date().toISOString(), p.content_id,
+              );
+              return json(db.queryOne("SELECT * FROM opc_media_content WHERE id = ?", p.content_id));
+            }
+
+            case "approve_content": {
+              const content = db.queryOne("SELECT * FROM opc_media_content WHERE id = ?", p.content_id) as Record<string, unknown> | null;
+              if (!content) return toolError("内容不存在", "RECORD_NOT_FOUND");
+
+              const now = new Date().toISOString();
+              if (p.approved) {
+                db.execute(
+                  "UPDATE opc_media_content SET status = 'approved', reviewer = ?, review_notes = ?, approved_at = ?, updated_at = ? WHERE id = ?",
+                  p.reviewer ?? "", p.review_notes ?? "", now, now, p.content_id,
+                );
+              } else {
+                db.execute(
+                  "UPDATE opc_media_content SET status = 'revision_needed', reviewer = ?, review_notes = ?, updated_at = ? WHERE id = ?",
+                  p.reviewer ?? "", p.review_notes ?? "", now, p.content_id,
+                );
+              }
+
+              return json(db.queryOne("SELECT * FROM opc_media_content WHERE id = ?", p.content_id));
+            }
+
+            case "content_analytics": {
+              const month = p.month ?? new Date().toISOString().slice(0, 7);
+
+              const total = (db.queryOne(
+                "SELECT COUNT(*) as cnt FROM opc_media_content WHERE company_id = ? AND created_at LIKE ?",
+                p.company_id, month + "%",
+              ) as { cnt: number }).cnt;
+
+              const published = (db.queryOne(
+                "SELECT COUNT(*) as cnt FROM opc_media_content WHERE company_id = ? AND status = 'published' AND published_date LIKE ?",
+                p.company_id, month + "%",
+              ) as { cnt: number }).cnt;
+
+              const byPlatform = db.query(
+                "SELECT platform, COUNT(*) as cnt FROM opc_media_content WHERE company_id = ? AND created_at LIKE ? GROUP BY platform",
+                p.company_id, month + "%",
+              );
+
+              const byStatus = db.query(
+                "SELECT status, COUNT(*) as cnt FROM opc_media_content WHERE company_id = ? AND created_at LIKE ? GROUP BY status",
+                p.company_id, month + "%",
+              );
+
+              // 汇总互动指标
+              const allMetrics = db.query(
+                "SELECT metrics FROM opc_media_content WHERE company_id = ? AND status = 'published' AND published_date LIKE ?",
+                p.company_id, month + "%",
+              ) as { metrics: string }[];
+
+              let totalViews = 0, totalLikes = 0, totalComments = 0, totalShares = 0;
+              for (const row of allMetrics) {
+                try {
+                  const m = JSON.parse(row.metrics);
+                  totalViews += m.views ?? 0;
+                  totalLikes += m.likes ?? 0;
+                  totalComments += m.comments ?? 0;
+                  totalShares += m.shares ?? 0;
+                } catch { /* skip */ }
+              }
+
+              return json({
+                ok: true,
+                month,
+                total_content: total,
+                published_count: published,
+                by_platform: byPlatform,
+                by_status: byStatus,
+                engagement: { views: totalViews, likes: totalLikes, comments: totalComments, shares: totalShares },
+              });
+            }
+
             default:
               return toolError(`未知操作: ${(p as { action: string }).action}`, "UNKNOWN_ACTION");
           }

package/src/tools/monitoring-tool.ts

@@ -6,6 +6,9 @@ import { Type, type Static } from "@sinclair/typebox";
 import type { OpenClawPluginApi } from "openclaw/plugin-sdk";
 import type { OpcDatabase } from "../db/index.js";
 import { json, toolError } from "../utils/tool-helper.js";
+import { computeHealthScore, computeGrowthScorecard, getLastBriefing } from "../opc/briefing-builder.js";
+import { detectCompanyStage } from "../opc/stage-detector.js";
+import { getActiveInsights } from "../opc/intelligence-engine.js";
 
 const MonitoringSchema = Type.Union([
   Type.Object({
@@ -52,6 +55,31 @@ const MonitoringSchema = Type.Union([
     action: Type.Literal("delete_metric"),
     metric_id: Type.String({ description: "指标记录 ID" }),
   }),
+  Type.Object({
+    action: Type.Literal("list_insights"),
+    company_id: Type.String({ description: "公司 ID" }),
+    insight_type: Type.Optional(Type.String({ description: "类型筛选: data_gap/trend/risk/opportunity/next_step/staff_observation" })),
+  }),
+  Type.Object({
+    action: Type.Literal("dismiss_insight"),
+    insight_id: Type.String({ description: "洞察 ID" }),
+  }),
+  Type.Object({
+    action: Type.Literal("health_score"),
+    company_id: Type.String({ description: "公司 ID" }),
+  }),
+  Type.Object({
+    action: Type.Literal("growth_scorecard"),
+    company_id: Type.String({ description: "公司 ID" }),
+  }),
+  Type.Object({
+    action: Type.Literal("compare_briefing"),
+    company_id: Type.String({ description: "公司 ID" }),
+  }),
+  Type.Object({
+    action: Type.Literal("strategy_report"),
+    company_id: Type.String({ description: "公司 ID" }),
+  }),
 ]);
 
 type MonitoringParams = Static<typeof MonitoringSchema>;
@@ -64,7 +92,9 @@ export function registerMonitoringTool(api: OpenClawPluginApi, db: OpcDatabase):
       description:
         "运营监控工具。操作: record_metric(记录指标), get_metrics(查询指标), " +
         "create_alert(创建告警), list_alerts(告警列表), " +
-        "dismiss_alert(消除告警), kpi_summary(KPI 汇总), delete_metric(删除指标记录)",
+        "dismiss_alert(消除告警), kpi_summary(KPI 汇总), delete_metric(删除指标记录), " +
+        "list_insights(查看洞察), dismiss_insight(标记洞察已处理), health_score(健康评分), " +
+        "growth_scorecard(增长评分卡), compare_briefing(简报历史对比), strategy_report(战略分析报告)",
       parameters: MonitoringSchema,
       async execute(_toolCallId, params) {
         const p = params as MonitoringParams;
@@ -135,7 +165,7 @@ export function registerMonitoringTool(api: OpenClawPluginApi, db: OpcDatabase):
               ) as { total_income: number; total_expense: number };
 
               const employees = db.queryOne(
-                "SELECT COUNT(*) as count FROM opc_employees WHERE company_id = ? AND status = 'active'",
+                "SELECT COUNT(*) as count FROM opc_hr_records WHERE company_id = ? AND status = 'active'",
                 p.company_id,
               ) as { count: number };
 
@@ -194,6 +224,109 @@ export function registerMonitoringTool(api: OpenClawPluginApi, db: OpcDatabase):
               return json({ ok: true });
             }
 
+            case "list_insights": {
+              let sql = "SELECT * FROM opc_insights WHERE company_id = ? AND status = 'active' AND (expires_at = '' OR expires_at > datetime('now'))";
+              const params2: unknown[] = [p.company_id];
+              if ("insight_type" in p && p.insight_type) { sql += " AND insight_type = ?"; params2.push(p.insight_type); }
+              sql += " ORDER BY priority DESC, created_at DESC LIMIT 20";
+              return json(db.query(sql, ...params2));
+            }
+
+            case "dismiss_insight": {
+              const now = new Date().toISOString();
+              db.execute(
+                "UPDATE opc_insights SET status = 'dismissed', updated_at = ? WHERE id = ?",
+                now, p.insight_id,
+              );
+              const ins = db.queryOne("SELECT * FROM opc_insights WHERE id = ?", p.insight_id);
+              if (!ins) return toolError("洞察不存在", "RECORD_NOT_FOUND");
+              return json(ins);
+            }
+
+            case "health_score": {
+              const score = computeHealthScore(db, p.company_id);
+              return json(score);
+            }
+
+            case "growth_scorecard": {
+              const card = computeGrowthScorecard(db, p.company_id);
+              return json(card);
+            }
+
+            case "compare_briefing": {
+              const last = getLastBriefing(db, p.company_id);
+              if (!last) return json({ message: "尚无历史简报快照，系统将在每日扫描时自动保存" });
+              const currentHealth = computeHealthScore(db, p.company_id);
+              const currentScorecard = computeGrowthScorecard(db, p.company_id);
+              return json({
+                last_date: last.date,
+                health_score: { previous: last.healthScore, current: currentHealth.total, change: currentHealth.total - last.healthScore },
+                income: { previous: last.totalIncome, current: (db.queryOne("SELECT COALESCE(SUM(amount),0) as total FROM opc_transactions WHERE company_id = ? AND type='income'", p.company_id) as { total: number }).total },
+                scorecard: { previous: last.scorecardGrade, current: currentScorecard.overall },
+                dimensions: currentHealth.dimensions,
+              });
+            }
+
+            case "strategy_report": {
+              const stageResult = detectCompanyStage(db, p.company_id);
+              const healthResult = computeHealthScore(db, p.company_id);
+              const scorecardResult = computeGrowthScorecard(db, p.company_id);
+              const insights = getActiveInsights(db, p.company_id, 20);
+
+              const totalIncome = (db.queryOne(
+                "SELECT COALESCE(SUM(amount), 0) as total FROM opc_transactions WHERE company_id = ? AND type = 'income'", p.company_id,
+              ) as { total: number }).total;
+              const totalExpense = (db.queryOne(
+                "SELECT COALESCE(SUM(amount), 0) as total FROM opc_transactions WHERE company_id = ? AND type = 'expense'", p.company_id,
+              ) as { total: number }).total;
+              const revenueMonths = (db.queryOne(
+                "SELECT COUNT(DISTINCT strftime('%Y-%m', transaction_date)) as cnt FROM opc_transactions WHERE company_id = ? AND type = 'income' AND amount > 0",
+                p.company_id,
+              ) as { cnt: number }).cnt;
+              const contactCount = (db.queryOne(
+                "SELECT COUNT(*) as cnt FROM opc_contacts WHERE company_id = ?", p.company_id,
+              ) as { cnt: number }).cnt;
+              const contractCount = (db.queryOne(
+                "SELECT COUNT(*) as cnt FROM opc_contracts WHERE company_id = ?", p.company_id,
+              ) as { cnt: number }).cnt;
+              const contentCount = (db.queryOne(
+                "SELECT COUNT(*) as cnt FROM opc_media_content WHERE company_id = ?", p.company_id,
+              ) as { cnt: number }).cnt;
+
+              // 收入来源分析
+              const counterparties = db.query(
+                "SELECT counterparty, SUM(amount) as total FROM opc_transactions WHERE company_id = ? AND type = 'income' AND counterparty != '' GROUP BY counterparty ORDER BY total DESC LIMIT 5",
+                p.company_id,
+              ) as { counterparty: string; total: number }[];
+              const topClient = counterparties[0];
+              const concentration = topClient && totalIncome > 0
+                ? Math.round((topClient.total / totalIncome) * 100) : 0;
+
+              return json({
+                stage: stageResult,
+                health: healthResult,
+                scorecard: scorecardResult,
+                financials: {
+                  total_income: totalIncome,
+                  total_expense: totalExpense,
+                  net_profit: totalIncome - totalExpense,
+                  profit_rate: totalIncome > 0 ? Math.round(((totalIncome - totalExpense) / totalIncome) * 100) : 0,
+                  revenue_months: revenueMonths,
+                  revenue_sources: counterparties,
+                  top_client_concentration: `${concentration}%`,
+                },
+                operations: {
+                  contact_count: contactCount,
+                  contract_count: contractCount,
+                  content_count: contentCount,
+                },
+                risks: insights.filter(i => i.insight_type === "risk").map(i => ({ title: i.title, priority: i.priority })),
+                opportunities: insights.filter(i => i.insight_type === "opportunity").map(i => ({ title: i.title })),
+                data_gaps: insights.filter(i => i.insight_type === "data_gap").map(i => i.title),
+                staff_observations: insights.filter(i => i.insight_type === "staff_observation").map(i => ({ role: i.staff_role, title: i.title })),
+              });
+            }
+
             default:
               return toolError(`未知操作: ${(p as { action: string }).action}`, "UNKNOWN_ACTION");
           }

package/src/tools/opc-tool.test.ts

@@ -0,0 +1,250 @@
+/**
+ * 星环OPC中心 — opc-tool (核心管理工具) 集成测试
+ */
+
+import { describe, it, expect, beforeEach, afterEach } from "vitest";
+import { createTestDb, factories } from "../__tests__/test-utils.js";
+import { SqliteAdapter } from "../db/sqlite-adapter.js";
+
+describe("opc-tool database integration", () => {
+  let db: SqliteAdapter;
+
+  beforeEach(() => {
+    db = createTestDb();
+  });
+
+  afterEach(() => {
+    db.close();
+  });
+
+  describe("create_company", () => {
+    it("should create a company successfully", () => {
+      const companyData = factories.company({
+        name: "创业公司A",
+        industry: "互联网",
+        owner_name: "张三",
+      });
+
+      const company = db.createCompany(companyData);
+
+      expect(company).not.toBeNull();
+      expect(company.id).toBeDefined();
+      expect(company.name).toBe("创业公司A");
+      expect(company.industry).toBe("互联网");
+      expect(company.owner_name).toBe("张三");
+      expect(company.status).toBe("active");
+    });
+
+    it("should set default values", () => {
+      const companyData = factories.company({
+        name: "最小配置公司",
+        industry: "咨询",
+        owner_name: "李四",
+      });
+
+      const company = db.createCompany(companyData);
+
+      expect(company.registered_capital).toBeDefined();
+      expect(company.created_at).toBeDefined();
+      expect(company.updated_at).toBeDefined();
+    });
+
+    it("should create multiple companies", () => {
+      const company1 = db.createCompany(factories.company({ name: "公司1" }));
+      const company2 = db.createCompany(factories.company({ name: "公司2" }));
+
+      expect(company1.id).not.toBe(company2.id);
+      expect(company1.name).toBe("公司1");
+      expect(company2.name).toBe("公司2");
+    });
+
+    it("should handle Chinese characters in company name", () => {
+      const company = db.createCompany(factories.company({
+        name: "星河科技有限公司",
+        description: "专注于AI技术研发",
+      }));
+
+      expect(company.name).toBe("星河科技有限公司");
+      expect(company.description).toBe("专注于AI技术研发");
+    });
+  });
+
+  describe("get_company", () => {
+    it("should retrieve a company by id", () => {
+      const created = db.createCompany(factories.company({ name: "测试公司" }));
+      const retrieved = db.getCompany(created.id);
+
+      expect(retrieved).not.toBeNull();
+      expect(retrieved!.id).toBe(created.id);
+      expect(retrieved!.name).toBe("测试公司");
+    });
+
+    it("should return null for non-existent company", () => {
+      const result = db.getCompany("non-existent-id");
+      expect(result).toBeNull();
+    });
+  });
+
+  describe("list_companies", () => {
+    it("should list all companies", () => {
+      db.createCompany(factories.company({ name: "公司1" }));
+      db.createCompany(factories.company({ name: "公司2" }));
+      db.createCompany(factories.company({ name: "公司3" }));
+
+      const companies = db.listCompanies();
+      expect(companies.length).toBe(3);
+    });
+
+    it("should filter companies by status", () => {
+      db.createCompany(factories.company({ name: "活跃公司", status: "active" }));
+      db.createCompany(factories.company({ name: "暂停公司", status: "suspended" }));
+      db.createCompany(factories.company({ name: "已收购公司", status: "acquired" }));
+
+      const activeCompanies = db.listCompanies("active");
+      expect(activeCompanies.length).toBe(1);
+      expect(activeCompanies[0].name).toBe("活跃公司");
+
+      const suspendedCompanies = db.listCompanies("suspended");
+      expect(suspendedCompanies.length).toBe(1);
+      expect(suspendedCompanies[0].name).toBe("暂停公司");
+    });
+
+    it("should return empty array when no companies exist", () => {
+      const companies = db.listCompanies();
+      expect(companies).toEqual([]);
+    });
+  });
+
+  describe("update_company", () => {
+    it("should update company status", () => {
+      const company = db.createCompany(factories.company({ status: "active" }));
+
+      db.execute(
+        "UPDATE opc_companies SET status = ?, updated_at = ? WHERE id = ?",
+        "suspended", new Date().toISOString(), company.id
+      );
+
+      const updated = db.getCompany(company.id);
+      expect(updated!.status).toBe("suspended");
+    });
+
+    it("should update company description", () => {
+      const company = db.createCompany(factories.company({ description: "原描述" }));
+
+      const newDescription = "更新后的描述";
+      db.execute(
+        "UPDATE opc_companies SET description = ?, updated_at = ? WHERE id = ?",
+        newDescription, new Date().toISOString(), company.id
+      );
+
+      const updated = db.getCompany(company.id);
+      expect(updated!.description).toBe(newDescription);
+    });
+
+    it("should update registered capital", () => {
+      const company = db.createCompany(factories.company({ registered_capital: 100000 }));
+
+      db.execute(
+        "UPDATE opc_companies SET registered_capital = ?, updated_at = ? WHERE id = ?",
+        500000, new Date().toISOString(), company.id
+      );
+
+      const updated = db.getCompany(company.id);
+      expect(updated!.registered_capital).toBe(500000);
+    });
+  });
+
+  describe("delete_company", () => {
+    it("should delete a company", () => {
+      const company = db.createCompany(factories.company({ name: "待删除公司" }));
+
+      db.execute("DELETE FROM opc_companies WHERE id = ?", company.id);
+
+      const deleted = db.getCompany(company.id);
+      expect(deleted).toBeNull();
+    });
+
+    it("should prevent deleting company with related data (foreign key constraint)", () => {
+      const company = db.createCompany(factories.company());
+
+      // Add related transaction
+      const txId = db.genId();
+      db.execute(
+        `INSERT INTO opc_transactions (id, company_id, type, category, amount, description, counterparty, transaction_date, created_at)
+         VALUES (?, ?, ?, ?, ?, ?, ?, ?, datetime('now'))`,
+        txId, company.id, "income", "service_income", 10000, "测试", "客户", "2026-01-15"
+      );
+
+      // Try to delete company - should fail due to foreign key constraint
+      expect(() => {
+        db.execute("DELETE FROM opc_companies WHERE id = ?", company.id);
+      }).toThrow();
+
+      // Proper way: delete related data first, then delete company
+      db.execute("DELETE FROM opc_transactions WHERE company_id = ?", company.id);
+      db.execute("DELETE FROM opc_companies WHERE id = ?", company.id);
+
+      const deleted = db.getCompany(company.id);
+      expect(deleted).toBeNull();
+    });
+  });
+
+  describe("company search and filtering", () => {
+    beforeEach(() => {
+      db.createCompany(factories.company({ name: "科技公司A", industry: "科技" }));
+      db.createCompany(factories.company({ name: "咨询公司B", industry: "咨询" }));
+      db.createCompany(factories.company({ name: "科技公司C", industry: "科技" }));
+    });
+
+    it("should filter by industry", () => {
+      const techCompanies = db.query(
+        "SELECT * FROM opc_companies WHERE industry = ?",
+        "科技"
+      ) as any[];
+      expect(techCompanies.length).toBe(2);
+    });
+
+    it("should search by name pattern", () => {
+      const companies = db.query(
+        "SELECT * FROM opc_companies WHERE name LIKE ?",
+        "%科技%"
+      ) as any[];
+      expect(companies.length).toBe(2);
+    });
+  });
+
+  describe("company statistics", () => {
+    it("should count companies by status", () => {
+      db.createCompany(factories.company({ status: "active" }));
+      db.createCompany(factories.company({ status: "active" }));
+      db.createCompany(factories.company({ status: "suspended" }));
+      db.createCompany(factories.company({ status: "acquired" }));
+
+      const stats = db.queryOne(
+        `SELECT
+           COUNT(*) as total,
+           COUNT(CASE WHEN status = 'active' THEN 1 END) as active,
+           COUNT(CASE WHEN status = 'suspended' THEN 1 END) as suspended,
+           COUNT(CASE WHEN status = 'acquired' THEN 1 END) as acquired
+         FROM opc_companies`
+      ) as any;
+
+      expect(stats.total).toBe(4);
+      expect(stats.active).toBe(2);
+      expect(stats.suspended).toBe(1);
+      expect(stats.acquired).toBe(1);
+    });
+
+    it("should calculate total registered capital", () => {
+      db.createCompany(factories.company({ registered_capital: 100000 }));
+      db.createCompany(factories.company({ registered_capital: 200000 }));
+      db.createCompany(factories.company({ registered_capital: 300000 }));
+
+      const result = db.queryOne(
+        "SELECT SUM(registered_capital) as total_capital FROM opc_companies"
+      ) as any;
+
+      expect(result.total_capital).toBe(600000);
+    });
+  });
+});