add test (#28)

Author: sakuyacatcat

app/service/stampService.ts

@@ -1,4 +1,3 @@
-// stampService.js
 import { StampRepository } from '../repository/stampRepository';
 
 export class StampService {

jest.config.js

@@ -1,3 +1,4 @@
 module.exports = {
-  testMatch: ["**/tests/**/*.[jt]s?(x)", "**/?(*.)+(spec|test).[tj]s?(x)"],
+  preset: 'ts-jest',
+  testMatch: ['**/tests/**/*.[jt]s?(x)', '**/?(*.)+(spec|test).[tj]s?(x)'],
 };

package.json

@@ -9,6 +9,7 @@
     "dev:monorepo": "next dev",
     "build": "next build",
     "start": "next start",
+    "clean": "rimraf node_modules",
     "format": "prettier --write 'app/**/*.{ts,tsx,json,md}'",
     "format:fix": "prettier --write 'app/**/*.{ts,tsx,json,md}'",
     "format:check": "prettier --check 'app/**/*.{ts,tsx,js,jsx,json,css}'",
@@ -44,6 +45,7 @@
   },
   "devDependencies": {
     "@frames.js/debugger": "^0.1.9",
+    "@types/jest": "^29.5.12",
     "@types/node": "^20.11.30",
     "@types/react": "^18.2.0",
     "@types/react-dom": "^18.2.0",

tests/sample.test.ts

@@ -0,0 +1,41 @@
+// StampRepository.test.ts
+import OpenAI from 'openai';
+import { StampRepository } from '../app/repository/stampRepository';
+
+describe('StampRepository', () => {
+  let openAIInstance: OpenAI;
+  let stampRepository: StampRepository;
+  const mockImageUrl = 'https://example.com/image.png';
+
+  beforeEach(() => {
+    openAIInstance = new OpenAI({ apiKey: 'dummy-api-key' });
+    stampRepository = new StampRepository(openAIInstance);
+  });
+
+  it('should generate an image URL successfully', async () => {
+    jest.spyOn(openAIInstance.images, 'generate').mockResolvedValue({
+      created: 0,
+      data: [{ url: mockImageUrl }],
+    });
+
+    const prompt = 'test prompt';
+    const imageUrl = await stampRepository.generateImage(prompt);
+    expect(imageUrl).toBe(mockImageUrl);
+    expect(openAIInstance.images.generate).toHaveBeenCalledWith({
+      model: 'dall-e-3',
+      prompt,
+      n: 1,
+      size: '1024x1024',
+    });
+  });
+
+  it('should throw an error when image generation fails', async () => {
+    jest
+      .spyOn(openAIInstance.images, 'generate')
+      .mockRejectedValue(new Error('API call failed'));
+
+    await expect(stampRepository.generateImage('test prompt')).rejects.toThrow(
+      'Image generation failed'
+    );
+  });
+});

tests/stampRepository.test.ts

@@ -0,0 +1,31 @@
+// stampService.test.js
+import { StampRepository } from '../app/repository/stampRepository';
+import { StampService } from '../app/service/stampService';
+
+describe('StampService', () => {
+  let service: StampService;
+  const template = 'Template: ';
+  const mockImageUrl = 'https://example.com/image.png';
+
+  beforeEach(() => {
+    jest.clearAllMocks();
+    service = new StampService(template, StampRepository.prototype);
+
+    jest
+      .spyOn(StampRepository.prototype, 'generateImage')
+      .mockImplementation((prompt) => {
+        return Promise.resolve(mockImageUrl);
+      });
+  });
+
+  it('should generate an image URL using the provided template and input', async () => {
+    const input = 'test input';
+    const result = await service.generateImageUrl(input);
+
+    const expectedPrompt = template + input;
+    expect(StampRepository.prototype.generateImage).toHaveBeenCalledWith(
+      expectedPrompt
+    );
+    expect(result).toBe(mockImageUrl);
+  });
+});

tests/stampRepository.test.ts_tmp

@@ -2490,6 +2490,14 @@
   dependencies:
     "@types/istanbul-lib-report" "*"
 
+"@types/jest@^29.5.12":
+  version "29.5.12"
+  resolved "https://registry.yarnpkg.com/@types/jest/-/jest-29.5.12.tgz#7f7dc6eb4cf246d2474ed78744b05d06ce025544"
+  integrity sha512-eDC8bTvT/QhYdxJAulQikueigY5AsdBRH2yDKW3yveW7svY3+DzN84/2NUgkw10RTiJbWqZrTtoGVdYlvFJdLw==
+  dependencies:
+    expect "^29.0.0"
+    pretty-format "^29.0.0"
+
 "@types/json5@^0.0.29":
   version "0.0.29"
   resolved "https://registry.npmjs.org/@types/json5/-/json5-0.0.29.tgz"
@@ -4867,7 +4875,7 @@ exit@^0.1.2:
   resolved "https://registry.npmjs.org/exit/-/exit-0.1.2.tgz"
   integrity sha512-Zk/eNKV2zbjpKzrsQ+n1G6poVbErQxJ0LBOJXaKZ1EViLzH+hrLu9cdXI4zw9dBQJslwBEpbQ2P1oS7nDxs6jQ==
 
-expect@^29.7.0:
+expect@^29.0.0, expect@^29.7.0:
   version "29.7.0"
   resolved "https://registry.npmjs.org/expect/-/expect-29.7.0.tgz"
   integrity sha512-2Zks0hf1VLFYI1kbh0I5jP3KHHyCHpkfyHBzsSXRFgl/Bg9mWYfMW8oD+PdMPlEwy5HNsR9JutYy6pMeOh61nw==
@@ -7859,7 +7867,7 @@ prettier@^3.2.5:
   resolved "https://registry.npmjs.org/prettier/-/prettier-3.2.5.tgz"
   integrity sha512-3/GWa9aOC0YeD7LUfvOG2NiDyhOWRvt1k+rcKhOuYnMY24iiCphgneUfJDyFXd6rZCAnuLBv6UeAULtrhT/F4A==
 
-pretty-format@^29.7.0:
+pretty-format@^29.0.0, pretty-format@^29.7.0:
   version "29.7.0"
   resolved "https://registry.npmjs.org/pretty-format/-/pretty-format-29.7.0.tgz"
   integrity sha512-Pdlw/oPxN+aXdmM9R00JVC9WVFoCLTKJvDVLgmJ+qAffBMxsV85l/Lu7sNx4zSzPyoL2euImuEwHhOXdEgNFZQ==