@kanopi/wp-ai-indexer 1.0.0

package/tests/mocks/openai.mock.ts

@@ -0,0 +1,52 @@
+import { vi } from 'vitest';
+
+/**
+ * Mock OpenAI client
+ */
+export const createMockOpenAI = () => {
+  return {
+    embeddings: {
+      create: vi.fn().mockResolvedValue({
+        data: [{ embedding: new Array(1536).fill(0.1) }],
+      }),
+    },
+  };
+};
+
+/**
+ * Mock batch embeddings response
+ */
+export const createMockBatchEmbeddings = (count: number) => {
+  return {
+    data: Array(count).fill(null).map(() => ({
+      embedding: new Array(1536).fill(0.1),
+    })),
+  };
+};
+
+/**
+ * Mock rate limit error
+ */
+export const createRateLimitError = () => {
+  const error: any = new Error('Rate limit exceeded');
+  error.status = 429;
+  return error;
+};
+
+/**
+ * Mock server error
+ */
+export const createServerError = () => {
+  const error: any = new Error('Internal server error');
+  error.status = 500;
+  return error;
+};
+
+/**
+ * Mock network error
+ */
+export const createNetworkError = () => {
+  const error: any = new Error('Connection reset');
+  error.code = 'ECONNRESET';
+  return error;
+};

package/tests/mocks/pinecone.mock.ts

@@ -0,0 +1,76 @@
+import { vi } from 'vitest';
+
+/**
+ * Mock Pinecone client
+ */
+export const createMockPinecone = () => {
+  const mockNamespace = {
+    upsert: vi.fn().mockResolvedValue({}),
+    deleteMany: vi.fn().mockResolvedValue({}),
+    fetch: vi.fn().mockResolvedValue({ records: {} }),
+    listPaginated: vi.fn().mockResolvedValue({ vectors: [] }),
+    query: vi.fn().mockResolvedValue({ matches: [] }),
+  };
+
+  const mockIndex = {
+    namespace: vi.fn().mockReturnValue(mockNamespace),
+    describeIndexStats: vi.fn().mockResolvedValue({
+      namespaces: {},
+      dimension: 1536,
+      indexFullness: 0,
+      totalVectorCount: 0,
+    }),
+  };
+
+  return {
+    index: vi.fn().mockReturnValue(mockIndex),
+    _mockIndex: mockIndex,
+    _mockNamespace: mockNamespace,
+  };
+};
+
+/**
+ * Mock vector data
+ */
+export const createMockVector = (id: string, postId: number, chunkIndex: number) => {
+  return {
+    id,
+    values: new Array(1536).fill(0.1),
+    metadata: {
+      post_id: postId,
+      post_type: 'post',
+      title: 'Test Post',
+      url: 'https://example.com/test',
+      chunk: 'Test content',
+      domain: 'example.com',
+      schema_version: 1,
+      post_date: '2024-01-01',
+      post_modified: '2024-01-01',
+      author_id: 1,
+      chunk_index: chunkIndex,
+    },
+  };
+};
+
+/**
+ * Mock list response with pagination
+ */
+export const createMockListResponse = (vectorIds: string[], hasNext: boolean = false) => {
+  return {
+    vectors: vectorIds.map(id => ({ id })),
+    pagination: hasNext ? { next: 'next-token' } : undefined,
+  };
+};
+
+/**
+ * Mock fetch response
+ */
+export const createMockFetchResponse = (vectors: any[]) => {
+  const records: Record<string, any> = {};
+
+  vectors.forEach(vector => {
+    records[vector.id] = vector;
+  });
+
+  return { records };
+};

package/tests/setup.ts

@@ -0,0 +1,20 @@
+import { beforeAll, afterAll, afterEach } from 'vitest';
+import nock from 'nock';
+import dotenv from 'dotenv';
+
+// Load test environment
+dotenv.config({ path: '.env.test' });
+
+// Clean nock after each test
+afterEach(() => {
+  nock.cleanAll();
+});
+
+// Prevent real HTTP calls
+beforeAll(() => {
+  nock.disableNetConnect();
+});
+
+afterAll(() => {
+  nock.enableNetConnect();
+});

package/vitest.config.ts

@@ -0,0 +1,29 @@
+import { defineConfig } from 'vitest/config';
+
+export default defineConfig({
+  test: {
+    globals: true,
+    environment: 'node',
+    setupFiles: ['./tests/setup.ts'],
+    coverage: {
+      provider: 'v8',
+      reporter: ['text', 'json', 'html', 'lcov'],
+      exclude: [
+        'node_modules/',
+        'dist/',
+        'tests/',
+        '**/*.test.ts',
+        '**/*.config.ts',
+        'bin/',
+      ],
+      thresholds: {
+        statements: 80,
+        branches: 75,
+        functions: 85,
+        lines: 80,
+      },
+    },
+    include: ['tests/**/*.test.ts'],
+    testTimeout: 10000,
+  },
+});