@talkpilot/core-db 1.3.39 → 1.3.41

package/src/test-utils/db-utils.ts

@@ -1,33 +1,33 @@
-import { MongoMemoryServer } from "mongodb-memory-server";
-import { MongoClient, Db } from "mongodb";
-import {
-  setDb,
-  setMunicipalDataDb,
-  setWebsitalkDb,
-  setConfigurationDb,
-} from "../index";
-
-let mongoServer: MongoMemoryServer;
-let client: MongoClient;
-
-export async function initTestDb(): Promise<Db> {
-  mongoServer = await MongoMemoryServer.create({
-    instance: { ip: "127.0.0.1" },
-  });
-  client = new MongoClient(mongoServer.getUri());
-  await client.connect();
-  const talkpilotDb = client.db();
-  const municipalDb = client.db("municipal-data");
-  const websitalkDb = client.db("website-talk");
-  const configurationDb = client.db("configuration");
-  setDb(talkpilotDb);
-  setMunicipalDataDb(municipalDb);
-  setWebsitalkDb(websitalkDb);
-  setConfigurationDb(configurationDb);
-  return talkpilotDb;
-}
-
-export async function closeTestDb(): Promise<void> {
-  if (client) await client.close();
-  if (mongoServer) await mongoServer.stop();
-}
+import { MongoMemoryServer } from "mongodb-memory-server";
+import { MongoClient, Db } from "mongodb";
+import {
+  setDb,
+  setMunicipalDataDb,
+  setWebsitalkDb,
+  setConfigurationDb,
+} from "../index";
+
+let mongoServer: MongoMemoryServer;
+let client: MongoClient;
+
+export async function initTestDb(): Promise<Db> {
+  mongoServer = await MongoMemoryServer.create({
+    instance: { ip: "127.0.0.1" },
+  });
+  client = new MongoClient(mongoServer.getUri());
+  await client.connect();
+  const talkpilotDb = client.db();
+  const municipalDb = client.db("municipal-data");
+  const websitalkDb = client.db("website-talk");
+  const configurationDb = client.db("configuration");
+  setDb(talkpilotDb);
+  setMunicipalDataDb(municipalDb);
+  setWebsitalkDb(websitalkDb);
+  setConfigurationDb(configurationDb);
+  return talkpilotDb;
+}
+
+export async function closeTestDb(): Promise<void> {
+  if (client) await client.close();
+  if (mongoServer) await mongoServer.stop();
+}

package/src/test-utils/factories/index.ts

@@ -1,16 +1,17 @@
-export * from "./talkpilot/agents";
-export * from "./talkpilot/calls";
-export * from "./talkpilot/clientAudioBuffers";
-export * from "./talkpilot/clientsConfig";
-export * from "./talkpilot/flows";
-export * from "./talkpilot/contextNotes";
-export * from "./talkpilot/groups";
-export * from "./talkpilot/phone_numbers";
-export * from "./talkpilot/sessions";
-export * from "./municipal/cities";
-export * from "./municipal/departmentsSubjects";
-export * from "./municipal/muniIssues";
-export * from "./municipal/streets";
-export * from "./municipal/tickets";
-export * from "./websitalk/scans";
-export * from "./websitalk/websiteUrls";
+export * from "./talkpilot/agents";
+export * from "./talkpilot/calls";
+export * from "./talkpilot/clientAudioBuffers";
+export * from "./talkpilot/clientsConfig";
+export * from "./talkpilot/flows";
+export * from "./talkpilot/contextNotes";
+export * from "./talkpilot/groups";
+export * from "./talkpilot/phone_numbers";
+export * from "./talkpilot/sessions";
+export * from "./municipal/cities";
+export * from "./municipal/departmentsSubjects";
+export * from "./municipal/muniIssues";
+export * from "./municipal/streets";
+export * from "./municipal/tickets";
+export * from "./websitalk/scans";
+export * from "./websitalk/websiteUrls";
+export * from "./websitalk/knowledgeDocuments";

package/src/test-utils/factories/municipal/cities.ts

@@ -1,16 +1,16 @@
-import { Factory } from "fishery";
-import { ObjectId } from "mongodb";
-import { faker } from "@faker-js/faker";
-import type { City } from "../../../municipal";
-
-export const cityFactory = Factory.define<City>(() => ({
-  _id: new ObjectId(),
-  name: faker.location.city().toLowerCase().replace(/\s/g, "_"),
-  displayName: faker.location.city(),
-  createdAt: new Date(),
-  updatedAt: new Date(),
-}));
-
-export function createCity(overrides?: Partial<City>): City {
-  return cityFactory.build(overrides);
-}
+import { Factory } from "fishery";
+import { ObjectId } from "mongodb";
+import { faker } from "@faker-js/faker";
+import type { City } from "../../../municipal";
+
+export const cityFactory = Factory.define<City>(() => ({
+  _id: new ObjectId(),
+  name: faker.location.city().toLowerCase().replace(/\s/g, "_"),
+  displayName: faker.location.city(),
+  createdAt: new Date(),
+  updatedAt: new Date(),
+}));
+
+export function createCity(overrides?: Partial<City>): City {
+  return cityFactory.build(overrides);
+}

package/src/test-utils/factories/municipal/departmentsSubjects.ts

@@ -1,37 +1,37 @@
-import { Factory } from "fishery";
-import { ObjectId } from "mongodb";
-import { faker } from "@faker-js/faker";
-import type { DepartmentSubject } from "../../../municipal";
-import { CityName } from "../../../municipal";
-
-const VALID_CITIES: CityName[] = [
-  "ashdod",
-  "maltar",
-  "billit",
-  "hashkelon",
-  "tests",
-];
-
-export const departmentSubjectFactory = Factory.define<DepartmentSubject>(
-  () => ({
-    _id: new ObjectId(),
-    createdAt: new Date(),
-    updatedAt: new Date(),
-    subjectName: faker.lorem.words(2),
-    subject_id: faker.string.numeric(4),
-    sub_subject_name: faker.lorem.word(),
-    sub_subject_id: faker.string.numeric(4),
-    descriptions: [faker.lorem.sentence()],
-    cityName: faker.helpers.arrayElement(VALID_CITIES),
-    guidelines: null,
-    instructions: [],
-    communications: [],
-    sla: faker.number.int({ min: 24, max: 720 }),
-  }),
-);
-
-export function createDepartmentSubject(
-  overrides?: Partial<DepartmentSubject>,
-): DepartmentSubject {
-  return departmentSubjectFactory.build(overrides);
-}
+import { Factory } from "fishery";
+import { ObjectId } from "mongodb";
+import { faker } from "@faker-js/faker";
+import type { DepartmentSubject } from "../../../municipal";
+import { CityName } from "../../../municipal";
+
+const VALID_CITIES: CityName[] = [
+  "ashdod",
+  "maltar",
+  "billit",
+  "hashkelon",
+  "tests",
+];
+
+export const departmentSubjectFactory = Factory.define<DepartmentSubject>(
+  () => ({
+    _id: new ObjectId(),
+    createdAt: new Date(),
+    updatedAt: new Date(),
+    subjectName: faker.lorem.words(2),
+    subject_id: faker.string.numeric(4),
+    sub_subject_name: faker.lorem.word(),
+    sub_subject_id: faker.string.numeric(4),
+    descriptions: [faker.lorem.sentence()],
+    cityName: faker.helpers.arrayElement(VALID_CITIES),
+    guidelines: null,
+    instructions: [],
+    communications: [],
+    sla: faker.number.int({ min: 24, max: 720 }),
+  }),
+);
+
+export function createDepartmentSubject(
+  overrides?: Partial<DepartmentSubject>,
+): DepartmentSubject {
+  return departmentSubjectFactory.build(overrides);
+}

package/src/test-utils/factories/municipal/muniIssues.ts

@@ -1,32 +1,32 @@
-import { Factory } from "fishery";
-import { faker } from "@faker-js/faker";
-import type { CreateMuniIssueInput } from "../../../municipal";
-
-/**
- * Test factory for `createMuniIssue` input.
- * `jira.key` uses `sequence` so each build gets a unique key (the real key
- * comes from Jira in production).
- */
-export const muniIssueInputFactory = Factory.define<CreateMuniIssueInput>(
-  ({ sequence }) => ({
-    callSid: `CA${faker.string.alphanumeric(10)}`,
-    clientId: faker.string.uuid(),
-    cityName: faker.location.city(),
-    issueContent: {
-      issuerName: faker.person.fullName(),
-      issuerRole: faker.person.jobTitle(),
-      issueDescription: faker.lorem.sentence(),
-      desiredBehavior: faker.lorem.sentence(),
-    },
-    jira: {
-      key: `CP-${sequence}`,
-      syncStatus: "created",
-    },
-  }),
-);
-
-export function createMuniIssueInput(
-  overrides?: Partial<CreateMuniIssueInput>,
-): CreateMuniIssueInput {
-  return muniIssueInputFactory.build(overrides);
-}
+import { Factory } from "fishery";
+import { faker } from "@faker-js/faker";
+import type { CreateMuniIssueInput } from "../../../municipal";
+
+/**
+ * Test factory for `createMuniIssue` input.
+ * `jira.key` uses `sequence` so each build gets a unique key (the real key
+ * comes from Jira in production).
+ */
+export const muniIssueInputFactory = Factory.define<CreateMuniIssueInput>(
+  ({ sequence }) => ({
+    callSid: `CA${faker.string.alphanumeric(10)}`,
+    clientId: faker.string.uuid(),
+    cityName: faker.location.city(),
+    issueContent: {
+      issuerName: faker.person.fullName(),
+      issuerRole: faker.person.jobTitle(),
+      issueDescription: faker.lorem.sentence(),
+      desiredBehavior: faker.lorem.sentence(),
+    },
+    jira: {
+      key: `CP-${sequence}`,
+      syncStatus: "created",
+    },
+  }),
+);
+
+export function createMuniIssueInput(
+  overrides?: Partial<CreateMuniIssueInput>,
+): CreateMuniIssueInput {
+  return muniIssueInputFactory.build(overrides);
+}

package/src/test-utils/factories/municipal/streets.ts

@@ -1,22 +1,22 @@
-import { Factory } from "fishery";
-import { ObjectId } from "mongodb";
-import { faker } from "@faker-js/faker";
-import type { Street } from "../../../municipal";
-
-export const streetFactory = Factory.define<Street>(() => ({
-  _id: new ObjectId(),
-  createdAt: new Date(),
-  updatedAt: new Date(),
-  id: faker.string.uuid(),
-  name: faker.location.street(),
-  cityName: faker.helpers.arrayElement([
-    "ashdod",
-    "maltar",
-    "billit",
-    "hashkelon",
-  ]),
-}));
-
-export function createStreet(overrides?: Partial<Street>): Street {
-  return streetFactory.build(overrides);
-}
+import { Factory } from "fishery";
+import { ObjectId } from "mongodb";
+import { faker } from "@faker-js/faker";
+import type { Street } from "../../../municipal";
+
+export const streetFactory = Factory.define<Street>(() => ({
+  _id: new ObjectId(),
+  createdAt: new Date(),
+  updatedAt: new Date(),
+  id: faker.string.uuid(),
+  name: faker.location.street(),
+  cityName: faker.helpers.arrayElement([
+    "ashdod",
+    "maltar",
+    "billit",
+    "hashkelon",
+  ]),
+}));
+
+export function createStreet(overrides?: Partial<Street>): Street {
+  return streetFactory.build(overrides);
+}

package/src/test-utils/factories/municipal/tickets.ts

@@ -1,39 +1,39 @@
-import { Factory } from "fishery";
-import { ObjectId } from "mongodb";
-import { faker } from "@faker-js/faker";
-import type { Ticket } from "../../../municipal";
-import { CityName } from "../../../municipal";
-
-const VALID_CITIES: CityName[] = [
-  "ashdod",
-  "maltar",
-  "billit",
-  "hashkelon",
-  "tests",
-];
-
-export const ticketFactory = Factory.define<Ticket>(() => ({
-  _id: new ObjectId(),
-  createdAt: new Date(),
-  updatedAt: new Date(),
-  cityName: faker.helpers.arrayElement(VALID_CITIES),
-  externalCallFields: {
-    first_name: faker.person.firstName(),
-    last_name: faker.person.lastName(),
-    event_description: faker.lorem.sentence(),
-    street: faker.location.street(),
-    house_number: faker.location.buildingNumber(),
-    event_subject_id: faker.string.numeric(4),
-    event_sub_subject_id: faker.string.numeric(4),
-    image1: faker.image.url(),
-    image2: faker.image.url(),
-    image3: faker.image.url(),
-    file1: faker.system.filePath(),
-    file2: faker.system.filePath(),
-    file3: faker.system.filePath(),
-  },
-}));
-
-export function createTicket(overrides?: Partial<Ticket>): Ticket {
-  return ticketFactory.build(overrides);
-}
+import { Factory } from "fishery";
+import { ObjectId } from "mongodb";
+import { faker } from "@faker-js/faker";
+import type { Ticket } from "../../../municipal";
+import { CityName } from "../../../municipal";
+
+const VALID_CITIES: CityName[] = [
+  "ashdod",
+  "maltar",
+  "billit",
+  "hashkelon",
+  "tests",
+];
+
+export const ticketFactory = Factory.define<Ticket>(() => ({
+  _id: new ObjectId(),
+  createdAt: new Date(),
+  updatedAt: new Date(),
+  cityName: faker.helpers.arrayElement(VALID_CITIES),
+  externalCallFields: {
+    first_name: faker.person.firstName(),
+    last_name: faker.person.lastName(),
+    event_description: faker.lorem.sentence(),
+    street: faker.location.street(),
+    house_number: faker.location.buildingNumber(),
+    event_subject_id: faker.string.numeric(4),
+    event_sub_subject_id: faker.string.numeric(4),
+    image1: faker.image.url(),
+    image2: faker.image.url(),
+    image3: faker.image.url(),
+    file1: faker.system.filePath(),
+    file2: faker.system.filePath(),
+    file3: faker.system.filePath(),
+  },
+}));
+
+export function createTicket(overrides?: Partial<Ticket>): Ticket {
+  return ticketFactory.build(overrides);
+}

package/src/test-utils/factories/talkpilot/agents.ts

@@ -1,19 +1,19 @@
-import { Factory } from "fishery";
-import { ObjectId } from "mongodb";
-import { faker } from "@faker-js/faker";
-import type { Agent } from "../../../talkpilot";
-
-export const agentFactory = Factory.define<Agent>(() => ({
-  name: faker.person.fullName(),
-  language: "en",
-  voice: "alloy",
-  instructions: faker.lorem.paragraph(),
-  labels: [],
-  clientId: new ObjectId().toString(),
-  role: "assistant",
-  gender: faker.helpers.arrayElement(["male", "female"]),
-}));
-
-export function createAgent(overrides?: Partial<Agent>): Agent {
-  return agentFactory.build(overrides);
-}
+import { Factory } from "fishery";
+import { ObjectId } from "mongodb";
+import { faker } from "@faker-js/faker";
+import type { Agent } from "../../../talkpilot";
+
+export const agentFactory = Factory.define<Agent>(() => ({
+  name: faker.person.fullName(),
+  language: "en",
+  voice: "alloy",
+  instructions: faker.lorem.paragraph(),
+  labels: [],
+  clientId: new ObjectId().toString(),
+  role: "assistant",
+  gender: faker.helpers.arrayElement(["male", "female"]),
+}));
+
+export function createAgent(overrides?: Partial<Agent>): Agent {
+  return agentFactory.build(overrides);
+}

package/src/test-utils/factories/talkpilot/calls.ts

@@ -1,37 +1,37 @@
-import { Factory } from "fishery";
-import { ObjectId } from "mongodb";
-import { faker } from "@faker-js/faker";
-import type { Call } from "../../../talkpilot";
-
-export const callFactory = Factory.define<Call>(() => ({
-  callSid: faker.string.uuid(),
-  flowId: new ObjectId(),
-  clientId: faker.string.uuid(),
-  sessionId: null,
-  resultId: new ObjectId(),
-  customerPhoneNumber: faker.phone.number(),
-  agentPhoneNumber: faker.phone.number(),
-  isIncomingCall: false,
-  isOutgoingCall: true,
-  callLength: faker.number.int({ min: 0, max: 3600 }),
-  env: "test",
-  updatedAt: new Date(),
-  createdAt: new Date(),
-}));
-
-export function createOutGoingCallDoc(overrides?: Partial<Call>): Call {
-  return callFactory.build({
-    ...overrides,
-    isIncomingCall: false,
-    isOutgoingCall: true,
-  });
-}
-
-export function createIncomingCallDoc(overrides?: Partial<Call>): Call {
-  return callFactory.build({
-    ...overrides,
-    isIncomingCall: true,
-    isOutgoingCall: false,
-    sessionId: overrides?.sessionId ?? null,
-  });
-}
+import { Factory } from "fishery";
+import { ObjectId } from "mongodb";
+import { faker } from "@faker-js/faker";
+import type { Call } from "../../../talkpilot";
+
+export const callFactory = Factory.define<Call>(() => ({
+  callSid: faker.string.uuid(),
+  flowId: new ObjectId(),
+  clientId: faker.string.uuid(),
+  sessionId: null,
+  resultId: new ObjectId(),
+  customerPhoneNumber: faker.phone.number(),
+  agentPhoneNumber: faker.phone.number(),
+  isIncomingCall: false,
+  isOutgoingCall: true,
+  callLength: faker.number.int({ min: 0, max: 3600 }),
+  env: "test",
+  updatedAt: new Date(),
+  createdAt: new Date(),
+}));
+
+export function createOutGoingCallDoc(overrides?: Partial<Call>): Call {
+  return callFactory.build({
+    ...overrides,
+    isIncomingCall: false,
+    isOutgoingCall: true,
+  });
+}
+
+export function createIncomingCallDoc(overrides?: Partial<Call>): Call {
+  return callFactory.build({
+    ...overrides,
+    isIncomingCall: true,
+    isOutgoingCall: false,
+    sessionId: overrides?.sessionId ?? null,
+  });
+}

package/src/test-utils/factories/talkpilot/clientAudioBuffers.ts

@@ -1,20 +1,20 @@
-import { Factory } from "fishery";
-import { faker } from "@faker-js/faker";
-import { CreateAudioBufferInput } from "../../../talkpilot";
-
-export const audioBufferInputFactory = Factory.define<CreateAudioBufferInput>(
-  () => ({
-    callSid: "CA1234567890abcdef1234567890abcdef",
-    clientId: faker.string.uuid(),
-    bucketStartMs: faker.number.int({ min: 0, max: 1000000 }),
-    bucketDurationMs: faker.helpers.arrayElement([500, 1000]),
-    data: Buffer.from(faker.string.alphanumeric({ length: 100 })),
-    ttlMs: faker.number.int({ min: 1000, max: 3600000 }),
-  }),
-);
-
-export function createAudioBufferInput(
-  overrides?: Partial<CreateAudioBufferInput>,
-): CreateAudioBufferInput {
-  return audioBufferInputFactory.build(overrides ?? {});
-}
+import { Factory } from "fishery";
+import { faker } from "@faker-js/faker";
+import { CreateAudioBufferInput } from "../../../talkpilot";
+
+export const audioBufferInputFactory = Factory.define<CreateAudioBufferInput>(
+  () => ({
+    callSid: "CA1234567890abcdef1234567890abcdef",
+    clientId: faker.string.uuid(),
+    bucketStartMs: faker.number.int({ min: 0, max: 1000000 }),
+    bucketDurationMs: faker.helpers.arrayElement([500, 1000]),
+    data: Buffer.from(faker.string.alphanumeric({ length: 100 })),
+    ttlMs: faker.number.int({ min: 1000, max: 3600000 }),
+  }),
+);
+
+export function createAudioBufferInput(
+  overrides?: Partial<CreateAudioBufferInput>,
+): CreateAudioBufferInput {
+  return audioBufferInputFactory.build(overrides ?? {});
+}

package/src/test-utils/factories/talkpilot/clientsConfig.ts

@@ -1,18 +1,18 @@
-import { Factory } from "fishery";
-import { ObjectId } from "mongodb";
-import { faker } from "@faker-js/faker";
-import type { ClientConfig, ClientConfigDoc } from "../../../talkpilot";
-
-export const clientConfigFactory = Factory.define<ClientConfigDoc>(() => ({
-  _id: new ObjectId(),
-  clientId: faker.string.uuid(),
-  products: {},
-}));
-
-export function createClientConfig(
-  overrides?: Partial<ClientConfig>,
-): ClientConfigDoc {
-  return clientConfigFactory.build({
-    ...overrides,
-  });
-}
+import { Factory } from "fishery";
+import { ObjectId } from "mongodb";
+import { faker } from "@faker-js/faker";
+import type { ClientConfig, ClientConfigDoc } from "../../../talkpilot";
+
+export const clientConfigFactory = Factory.define<ClientConfigDoc>(() => ({
+  _id: new ObjectId(),
+  clientId: faker.string.uuid(),
+  products: {},
+}));
+
+export function createClientConfig(
+  overrides?: Partial<ClientConfig>,
+): ClientConfigDoc {
+  return clientConfigFactory.build({
+    ...overrides,
+  });
+}