promptfoo

1
const mockDbInstance = {
2
  // Mock any method you use from the dbInstance
3
  // For example:
4
  // query: jest.fn().mockResolvedValue({}),
5
};
6

7
const mockRelations = jest.fn();
8

9
const mockSqliteTable = jest.fn().mockImplementation((tableName, schema) => {
10
  // You can customize this mock based on your testing needs
11
  return { tableName, schema };
12
});
13

14
module.exports = {
15
  prompts: mockSqliteTable('prompts', {
16
    /* schema definition */
17
  }),
18
  promptsRelations: mockRelations,
19
  datasets: mockSqliteTable('datasets', {
20
    /* schema definition */
21
  }),
22
  datasetsRelations: mockRelations,
23
  evals: mockSqliteTable('evals', {
24
    /* schema definition */
25
  }),
26
  evalsRelations: mockRelations,
27
  evalsToPrompts: mockSqliteTable('evals_to_prompts', {
28
    /* schema definition */
29
  }),
30
  evalsToPromptsRelations: mockRelations,
31
  evalsToDatasets: mockSqliteTable('evals_to_datasets', {
32
    /* schema definition */
33
  }),
34
  evalsToDatasetsRelations: mockRelations,
35
  llmOutputs: mockSqliteTable('llm_outputs', {
36
    /* schema definition */
37
  }),
38
  llmOutputsRelations: mockRelations,
39
  getDb: jest.fn(() => mockDbInstance),
40
};
41