Update tests

robisim74 · Mar 27, 2024 · 3be8122 · 3be8122
1 parent 682ae2b
commit 3be8122
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 8 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -190,7 +190,7 @@ export async function gptTranslateJson(options: GptTranslateJsonOptions) {
 
           if (response?.choices) {
             const content = response.choices[0].message?.content;
-            console.log(content);
+
             if (content) {
               const jsonContent = JSON.parse(content);
               translatedTexts = [...translatedTexts, ...jsonContent]

diff --git a/tests/add-lang.test.ts b/tests/add-lang.test.ts
@@ -2,7 +2,7 @@ import { test, describe, expect, vi } from 'vitest';
 
 import { readdir, readFile, writeFile } from 'fs/promises';
 import { normalize } from 'path';
-import { OpenAIApi } from 'openai';
+import OpenAI from 'openai';
 
 import { gptTranslateJson } from '../src/index';
 import { mockAddLangAsset, mockAddLangMetaTranslated, mockAddLangMetaTranslatedLangs, mockAddLangResponse, mockAddLangTranslatedAsset, mockAddTranslationMetaTranslated, mockAddTranslationMetaTranslatedLangs, mockAddTranslationTranslatedAsset } from './mock';
@@ -40,7 +40,7 @@ vi.mock('fs', async () => {
 
 describe('gptTranslateJson', () => {
   test('add lang', async () => {
-    vi.spyOn(OpenAIApi.prototype, 'createChatCompletion').mockImplementationOnce(() => Promise.resolve<any>({ data: mockAddLangResponse }));
+    vi.spyOn(OpenAI.Chat.Completions.prototype, 'create').mockImplementationOnce(() => Promise.resolve<any>(mockAddLangResponse));
 
     await gptTranslateJson({
       apiKey: 'openai_api_key',

diff --git a/tests/add-translation.test.ts b/tests/add-translation.test.ts
@@ -2,7 +2,7 @@ import { test, describe, expect, vi } from 'vitest';
 
 import { readdir, readFile, writeFile } from 'fs/promises';
 import { normalize } from 'path';
-import { OpenAIApi } from 'openai';
+import OpenAI from 'openai';
 
 import { gptTranslateJson } from '../src/index';
 import { mockAddTranslationAsset, mockAddTranslationMetaTranslated, mockAddTranslationMetaTranslatedLangs, mockAddTranslationResponse, mockAddTranslationTranslatedAsset, mockMetaTranslated, mockMetaTranslatedLangs, mockTranslatedAsset } from './mock';
@@ -39,7 +39,7 @@ vi.mock('fs', async () => {
 
 describe('gptTranslateJson', () => {
   test('add translation', async () => {
-    vi.spyOn(OpenAIApi.prototype, 'createChatCompletion').mockImplementationOnce(() => Promise.resolve<any>({ data: mockAddTranslationResponse }));
+    vi.spyOn(OpenAI.Chat.Completions.prototype, 'create').mockImplementationOnce(() => Promise.resolve<any>(mockAddTranslationResponse));
 
     await gptTranslateJson({
       apiKey: 'openai_api_key',

diff --git a/tests/translate.test.ts b/tests/translate.test.ts
@@ -2,7 +2,7 @@ import { test, describe, expect, vi } from 'vitest';
 
 import { readdir, readFile, writeFile } from 'fs/promises';
 import { normalize } from 'path';
-import { OpenAIApi } from 'openai';
+import OpenAI from 'openai';
 
 import { gptTranslateJson } from '../src/index';
 import { mockAsset, mockMetaTranslated, mockMetaTranslatedLangs, mockResponse, mockTranslatedAsset } from './mock';
@@ -31,7 +31,7 @@ vi.mock('fs', async () => {
 
 describe('gptTranslateJson', () => {
   test('translate', async () => {
-    vi.spyOn(OpenAIApi.prototype, 'createChatCompletion').mockImplementationOnce(() => Promise.resolve<any>({ data: mockResponse }));
+    vi.spyOn(OpenAI.Chat.Completions.prototype, 'create').mockImplementationOnce(() => Promise.resolve<any>(mockResponse));
 
     await gptTranslateJson({
       apiKey: 'openai_api_key',

diff --git a/tsconfig.json b/tsconfig.json
@@ -4,7 +4,6 @@
     "useDefineForClassFields": true,
     "module": "ESNext",
     "lib": ["ES2020", "DOM", "DOM.Iterable"],
-    "skipLibCheck": true,
     "moduleResolution": "bundler",
     "allowImportingTsExtensions": true,
     "resolveJsonModule": true,