@studious-lms/server 1.16.0 → 1.17.0

package/tests/unit/services/openaiDocumentGen.test.ts

@@ -0,0 +1,65 @@
+import { describe, expect, it } from 'vitest';
+import {
+  extractContainerGeneratedFiles,
+  mimeTypeForFormat,
+  normalizeDocumentFormat,
+} from '../../../src/services/openaiDocumentGen.js';
+import type { Response } from 'openai/resources/responses/responses.js';
+
+describe('openaiDocumentGen', () => {
+  it('normalizes unknown formats to pdf', () => {
+    expect(normalizeDocumentFormat('docx')).toBe('docx');
+    expect(normalizeDocumentFormat('.PDF')).toBe('pdf');
+    expect(normalizeDocumentFormat('bogus')).toBe('pdf');
+  });
+
+  it('maps formats to mime types', () => {
+    expect(mimeTypeForFormat('pdf')).toBe('application/pdf');
+    expect(mimeTypeForFormat('xlsx')).toBe(
+      'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
+    );
+    expect(mimeTypeForFormat('png')).toBe('image/png');
+  });
+
+  it('collects container_file_citation annotations for the requested extension', () => {
+    const response = {
+      output: [
+        {
+          type: 'message',
+          content: [
+            {
+              type: 'output_text',
+              text: 'Here is your spreadsheet.',
+              annotations: [
+                {
+                  type: 'container_file_citation',
+                  container_id: 'cntr_1',
+                  file_id: 'cfile_1',
+                  filename: 'Grades.xlsx',
+                  start_index: 0,
+                  end_index: 0,
+                },
+                {
+                  type: 'container_file_citation',
+                  container_id: 'cntr_1',
+                  file_id: 'cfile_2',
+                  filename: 'Notes.pdf',
+                  start_index: 0,
+                  end_index: 0,
+                },
+              ],
+            },
+          ],
+        },
+      ],
+    } as unknown as Response;
+
+    expect(extractContainerGeneratedFiles(response, 'xlsx')).toEqual([
+      {
+        containerId: 'cntr_1',
+        fileId: 'cfile_1',
+        filename: 'Grades.xlsx',
+      },
+    ]);
+  });
+});