From 4538f2a9d44df5e8993e1c79dd6ba5c0496ed68c Mon Sep 17 00:00:00 2001
From: "gru-agent[bot]" <185149714+gru-agent[bot]@users.noreply.github.com>
Date: Wed, 26 Mar 2025 13:26:04 +0800
Subject: [PATCH] Add unit tests for authType2UsageSource function in
 utils.test.ts (#4326)

Co-authored-by: gru-agent[bot] <185149714+gru-agent[bot]@users.noreply.github.com>
---
 .../support/wallet/usage/utils.test.ts        | 59 +++++++++++++++++++
 1 file changed, 59 insertions(+)
 create mode 100644 test/cases/service/support/wallet/usage/utils.test.ts

diff --git a/test/cases/service/support/wallet/usage/utils.test.ts b/test/cases/service/support/wallet/usage/utils.test.ts
new file mode 100644
index 000000000..1590252ef
--- /dev/null
+++ b/test/cases/service/support/wallet/usage/utils.test.ts
@@ -0,0 +1,59 @@
+import { describe, it, expect } from 'vitest';
+import { authType2UsageSource } from '@/service/support/wallet/usage/utils';
+import { AuthUserTypeEnum } from '@fastgpt/global/support/permission/constant';
+import { UsageSourceEnum } from '@fastgpt/global/support/wallet/usage/constants';
+
+describe('authType2UsageSource', () => {
+  it('should return source if provided', () => {
+    const result = authType2UsageSource({
+      authType: AuthUserTypeEnum.apikey,
+      shareId: 'share123',
+      source: UsageSourceEnum.api
+    });
+    expect(result).toBe(UsageSourceEnum.api);
+  });
+
+  it('should return shareLink if shareId is provided', () => {
+    const result = authType2UsageSource({
+      authType: AuthUserTypeEnum.apikey,
+      shareId: 'share123'
+    });
+    expect(result).toBe(UsageSourceEnum.shareLink);
+  });
+
+  it('should return api if authType is apikey', () => {
+    const result = authType2UsageSource({
+      authType: AuthUserTypeEnum.apikey
+    });
+    expect(result).toBe(UsageSourceEnum.api);
+  });
+
+  it('should return fastgpt as default', () => {
+    const result = authType2UsageSource({});
+    expect(result).toBe(UsageSourceEnum.fastgpt);
+  });
+
+  it('should return fastgpt for non-apikey authType', () => {
+    const result = authType2UsageSource({
+      authType: AuthUserTypeEnum.owner
+    });
+    expect(result).toBe(UsageSourceEnum.fastgpt);
+  });
+
+  it('should prioritize source over shareId and authType', () => {
+    const result = authType2UsageSource({
+      source: UsageSourceEnum.api,
+      shareId: 'share123',
+      authType: AuthUserTypeEnum.apikey
+    });
+    expect(result).toBe(UsageSourceEnum.api);
+  });
+
+  it('should prioritize shareId over authType', () => {
+    const result = authType2UsageSource({
+      shareId: 'share123',
+      authType: AuthUserTypeEnum.apikey
+    });
+    expect(result).toBe(UsageSourceEnum.shareLink);
+  });
+});