@talkpilot/core-db 1.3.58 → 1.3.59

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (228) hide show
  1. package/.cursor/rules/development.mdc +65 -65
  2. package/AGENTS.md +49 -49
  3. package/DEVELOPMENT.md +141 -141
  4. package/README.md +779 -778
  5. package/dist/configuration/models/index.d.ts +1 -1
  6. package/dist/configuration/models/index.d.ts.map +1 -1
  7. package/dist/configuration/models/models.types.d.ts +12 -0
  8. package/dist/configuration/models/models.types.d.ts.map +1 -1
  9. package/dist/index.d.ts +0 -1
  10. package/dist/index.d.ts.map +1 -1
  11. package/dist/index.js +0 -1
  12. package/dist/index.js.map +1 -1
  13. package/dist/websitalk/scanResources/scanResources.getters.d.ts +0 -1
  14. package/dist/websitalk/scanResources/scanResources.getters.d.ts.map +1 -1
  15. package/dist/websitalk/scanResources/scanResources.getters.js +1 -6
  16. package/dist/websitalk/scanResources/scanResources.getters.js.map +1 -1
  17. package/dist/websitalk/scans/scans.constants.d.ts +1 -3
  18. package/dist/websitalk/scans/scans.constants.d.ts.map +1 -1
  19. package/dist/websitalk/scans/scans.constants.js +0 -3
  20. package/dist/websitalk/scans/scans.constants.js.map +1 -1
  21. package/dist/websitalk/scans/scans.getters.d.ts +0 -1
  22. package/dist/websitalk/scans/scans.getters.d.ts.map +1 -1
  23. package/dist/websitalk/scans/scans.getters.js +1 -6
  24. package/dist/websitalk/scans/scans.getters.js.map +1 -1
  25. package/dist/websitalk/scans/scans.types.d.ts +0 -1
  26. package/dist/websitalk/scans/scans.types.d.ts.map +1 -1
  27. package/jest.config.js +20 -20
  28. package/package.json +46 -46
  29. package/src/__tests__/setup.ts +20 -20
  30. package/src/bulkWrite/__tests__/bulkWrite.spec.ts +65 -65
  31. package/src/bulkWrite/bulkWrite.ops.ts +45 -45
  32. package/src/bulkWrite/bulkWrite.types.ts +4 -4
  33. package/src/bulkWrite/index.ts +2 -2
  34. package/src/config.ts +9 -9
  35. package/src/configuration/index.ts +17 -17
  36. package/src/configuration/internalModels/index.ts +3 -3
  37. package/src/configuration/internalModels/internalModels.constants.ts +1 -1
  38. package/src/configuration/internalModels/internalModels.getters.ts +33 -33
  39. package/src/configuration/internalModels/internalModels.types.ts +12 -12
  40. package/src/configuration/models/__tests__/models.getters.spec.ts +86 -67
  41. package/src/configuration/models/index.ts +28 -27
  42. package/src/configuration/models/models.constants.ts +15 -15
  43. package/src/configuration/models/models.getters.ts +71 -71
  44. package/src/configuration/models/models.types.ts +93 -80
  45. package/src/configuration/mongodb-client.ts +62 -62
  46. package/src/configuration/prompts/index.ts +2 -2
  47. package/src/configuration/prompts/prompts.getters.ts +49 -49
  48. package/src/configuration/prompts/prompts.types.ts +11 -11
  49. package/src/connection.ts +81 -81
  50. package/src/index.ts +46 -47
  51. package/src/municipal/__tests__/validation.spec.ts +62 -62
  52. package/src/municipal/cities/cities.getters.ts +50 -50
  53. package/src/municipal/cities/cities.types.ts +11 -11
  54. package/src/municipal/cities/index.ts +2 -2
  55. package/src/municipal/departmentsSubjects/__tests__/departmentsSubjects.spec.ts +272 -272
  56. package/src/municipal/departmentsSubjects/departmentsSubjects.getters.ts +317 -317
  57. package/src/municipal/departmentsSubjects/departmentsSubjects.types.ts +73 -73
  58. package/src/municipal/departmentsSubjects/index.ts +9 -9
  59. package/src/municipal/index.ts +22 -22
  60. package/src/municipal/mongodb-client.ts +61 -61
  61. package/src/municipal/muniIssues/__tests__/muniIssues.setters.spec.ts +94 -94
  62. package/src/municipal/muniIssues/__tests__/muniIssues.setup.spec.ts +92 -92
  63. package/src/municipal/muniIssues/index.ts +17 -17
  64. package/src/municipal/muniIssues/muniIssues.constants.ts +18 -18
  65. package/src/municipal/muniIssues/muniIssues.getters.ts +24 -24
  66. package/src/municipal/muniIssues/muniIssues.schema.ts +66 -66
  67. package/src/municipal/muniIssues/muniIssues.setters.ts +24 -24
  68. package/src/municipal/muniIssues/muniIssues.setup.ts +65 -65
  69. package/src/municipal/muniIssues/muniIssues.types.ts +37 -37
  70. package/src/municipal/streets/__tests__/streets.spec.ts +253 -253
  71. package/src/municipal/streets/index.ts +2 -2
  72. package/src/municipal/streets/streets.getters.ts +140 -140
  73. package/src/municipal/streets/streets.types.ts +19 -19
  74. package/src/municipal/systemInstructions/__tests__/getters.spec.ts +113 -113
  75. package/src/municipal/systemInstructions/__tests__/setters.spec.ts +274 -274
  76. package/src/municipal/systemInstructions/index.ts +7 -7
  77. package/src/municipal/systemInstructions/instructions.getters.ts +57 -57
  78. package/src/municipal/systemInstructions/instructions.setters.ts +119 -119
  79. package/src/municipal/systemInstructions/instructions.types.ts +30 -30
  80. package/src/municipal/tickets/__tests__/tickets.getters.spec.ts +30 -30
  81. package/src/municipal/tickets/__tests__/tickets.statistics.spec.ts +99 -99
  82. package/src/municipal/tickets/index.ts +9 -9
  83. package/src/municipal/tickets/tickets.constants.ts +8 -8
  84. package/src/municipal/tickets/tickets.getters.ts +121 -121
  85. package/src/municipal/tickets/tickets.statistics.aggregation.ts +110 -110
  86. package/src/municipal/tickets/tickets.statistics.getters.ts +145 -145
  87. package/src/municipal/tickets/tickets.types.ts +54 -54
  88. package/src/municipal/utils/types.ts +11 -11
  89. package/src/talkpilot/__tests__/db.spec.ts +38 -38
  90. package/src/talkpilot/__tests__/mongodb-client.spec.ts +18 -18
  91. package/src/talkpilot/__tests__/validation.spec.ts +68 -68
  92. package/src/talkpilot/agents/__tests__/agents.getters.spec.ts +29 -29
  93. package/src/talkpilot/agents/agents.getters.ts +34 -34
  94. package/src/talkpilot/agents/agents.types.ts +14 -14
  95. package/src/talkpilot/agents/index.ts +2 -2
  96. package/src/talkpilot/backgroundToolResults/__tests__/backgroundToolResults.getters.spec.ts +147 -147
  97. package/src/talkpilot/backgroundToolResults/backgroundToolResults.getters.ts +65 -65
  98. package/src/talkpilot/backgroundToolResults/backgroundToolResults.types.ts +23 -23
  99. package/src/talkpilot/backgroundToolResults/index.ts +2 -2
  100. package/src/talkpilot/calls/__tests__/callStats.utils.spec.ts +128 -128
  101. package/src/talkpilot/calls/__tests__/calls.dashboard.spec.ts +229 -229
  102. package/src/talkpilot/calls/__tests__/calls.spec.ts +471 -471
  103. package/src/talkpilot/calls/__tests__/calls.statistics.spec.ts +483 -483
  104. package/src/talkpilot/calls/calls.constants.ts +49 -49
  105. package/src/talkpilot/calls/calls.getters.ts +277 -277
  106. package/src/talkpilot/calls/calls.statistics.getters.ts +668 -668
  107. package/src/talkpilot/calls/calls.statistics.types.ts +48 -48
  108. package/src/talkpilot/calls/calls.types.ts +149 -149
  109. package/src/talkpilot/calls/dashboard/calls.dashboard.ts +295 -295
  110. package/src/talkpilot/calls/dashboard/calls.dashboard.types.ts +57 -57
  111. package/src/talkpilot/calls/index.ts +6 -6
  112. package/src/talkpilot/clientAudioBuffers/__tests__/clientAudioBuffer.getters.spec.ts +160 -160
  113. package/src/talkpilot/clientAudioBuffers/clientAudioBuffer.getters.ts +117 -117
  114. package/src/talkpilot/clientAudioBuffers/clientsAudioBuffers.types.ts +25 -25
  115. package/src/talkpilot/clientAudioBuffers/index.ts +2 -2
  116. package/src/talkpilot/clients/__tests__/clients.quota.spec.ts +123 -123
  117. package/src/talkpilot/clients/clients.getters.ts +16 -16
  118. package/src/talkpilot/clients/clients.quota.getters.ts +119 -119
  119. package/src/talkpilot/clients/clients.quota.types.ts +27 -27
  120. package/src/talkpilot/clients/clients.types.ts +18 -18
  121. package/src/talkpilot/clients/index.ts +4 -4
  122. package/src/talkpilot/clientsConfig/__tests__/clientsConfig.getters.spec.ts +423 -423
  123. package/src/talkpilot/clientsConfig/__tests__/clientsConfig.spec.ts +212 -212
  124. package/src/talkpilot/clientsConfig/clientsConfig.constants.ts +2 -2
  125. package/src/talkpilot/clientsConfig/clientsConfig.getters.ts +247 -247
  126. package/src/talkpilot/clientsConfig/clientsConfig.types.ts +136 -136
  127. package/src/talkpilot/clientsConfig/index.ts +3 -3
  128. package/src/talkpilot/contextNotes/__tests__/contextNotes.getters.spec.ts +217 -217
  129. package/src/talkpilot/contextNotes/contextNotes.getters.ts +85 -85
  130. package/src/talkpilot/contextNotes/contextNotes.types.ts +20 -20
  131. package/src/talkpilot/contextNotes/index.ts +2 -2
  132. package/src/talkpilot/flows/__tests__/flows.schema.spec.ts +124 -124
  133. package/src/talkpilot/flows/flows.getter.ts +34 -34
  134. package/src/talkpilot/flows/flows.schema.ts +180 -180
  135. package/src/talkpilot/flows/flows.types.ts +211 -211
  136. package/src/talkpilot/flows/index.ts +7 -7
  137. package/src/talkpilot/groups/__tests__/groups.spec.ts +90 -90
  138. package/src/talkpilot/groups/__tests__/phone.utils.spec.ts +32 -32
  139. package/src/talkpilot/groups/groups.getters.ts +30 -30
  140. package/src/talkpilot/groups/groups.types.ts +29 -29
  141. package/src/talkpilot/groups/index.ts +3 -3
  142. package/src/talkpilot/groups/phone.utils.ts +46 -46
  143. package/src/talkpilot/index.ts +31 -31
  144. package/src/talkpilot/leads/index.ts +2 -2
  145. package/src/talkpilot/leads/leads.getter.ts +6 -6
  146. package/src/talkpilot/leads/leads.schema.ts +33 -33
  147. package/src/talkpilot/leads/leads.types.ts +20 -20
  148. package/src/talkpilot/mongodb-client.ts +78 -78
  149. package/src/talkpilot/phone_numbers/__tests__/phone_numbers.spec.ts +282 -282
  150. package/src/talkpilot/phone_numbers/index.ts +2 -2
  151. package/src/talkpilot/phone_numbers/phone_numbers.getter.ts +176 -176
  152. package/src/talkpilot/phone_numbers/phone_numbers.schema.ts +17 -17
  153. package/src/talkpilot/phone_numbers/phone_numbers.types.ts +30 -30
  154. package/src/talkpilot/plans/__tests__/plans.spec.ts +70 -70
  155. package/src/talkpilot/plans/index.ts +2 -2
  156. package/src/talkpilot/plans/plans.getters.ts +132 -132
  157. package/src/talkpilot/plans/plans.types.ts +89 -89
  158. package/src/talkpilot/products/__tests__/products.getters.spec.ts +44 -44
  159. package/src/talkpilot/products/index.ts +2 -2
  160. package/src/talkpilot/products/products.getters.ts +12 -12
  161. package/src/talkpilot/products/products.types.ts +9 -9
  162. package/src/talkpilot/results/index.ts +7 -7
  163. package/src/talkpilot/results/results.getter.ts +39 -39
  164. package/src/talkpilot/results/results.schema.ts +25 -25
  165. package/src/talkpilot/results/results.types.ts +34 -34
  166. package/src/talkpilot/retry_analyze/__tests__/retryAnalyze.getters.spec.ts +156 -156
  167. package/src/talkpilot/retry_analyze/index.ts +2 -2
  168. package/src/talkpilot/retry_analyze/retryAnalyze.getters.ts +84 -84
  169. package/src/talkpilot/retry_analyze/retryAnalyze.types.ts +13 -13
  170. package/src/talkpilot/sessions/__tests__/sessions.spec.ts +147 -147
  171. package/src/talkpilot/sessions/index.ts +2 -2
  172. package/src/talkpilot/sessions/sessions.getter.ts +92 -92
  173. package/src/talkpilot/sessions/sessions.schema.ts +34 -34
  174. package/src/talkpilot/sessions/sessions.types.ts +30 -30
  175. package/src/talkpilot/subscriptions/__tests__/subscriptions.getters.utils.spec.ts +45 -45
  176. package/src/talkpilot/subscriptions/index.ts +3 -3
  177. package/src/talkpilot/subscriptions/subscriptions.getters.ts +146 -146
  178. package/src/talkpilot/subscriptions/subscriptions.getters.utils.ts +33 -33
  179. package/src/talkpilot/subscriptions/subscriptions.types.ts +66 -66
  180. package/src/talkpilot/utils/__tests__/query.utils.spec.ts +49 -49
  181. package/src/talkpilot/utils/query.utils.ts +21 -21
  182. package/src/test-utils/db-utils.ts +33 -33
  183. package/src/test-utils/factories/index.ts +18 -18
  184. package/src/test-utils/factories/municipal/cities.ts +16 -16
  185. package/src/test-utils/factories/municipal/departmentsSubjects.ts +37 -37
  186. package/src/test-utils/factories/municipal/muniIssues.ts +32 -32
  187. package/src/test-utils/factories/municipal/streets.ts +22 -22
  188. package/src/test-utils/factories/municipal/tickets.ts +39 -39
  189. package/src/test-utils/factories/talkpilot/agents.ts +19 -19
  190. package/src/test-utils/factories/talkpilot/calls.ts +37 -37
  191. package/src/test-utils/factories/talkpilot/clientAudioBuffers.ts +20 -20
  192. package/src/test-utils/factories/talkpilot/clientsConfig.ts +18 -18
  193. package/src/test-utils/factories/talkpilot/contextNotes.ts +40 -40
  194. package/src/test-utils/factories/talkpilot/flows.ts +33 -33
  195. package/src/test-utils/factories/talkpilot/groups.ts +33 -33
  196. package/src/test-utils/factories/talkpilot/phone_numbers.ts +22 -22
  197. package/src/test-utils/factories/talkpilot/sessions.ts +35 -35
  198. package/src/test-utils/factories/websitalk/knowledgeDocuments.ts +29 -29
  199. package/src/test-utils/factories/websitalk/scanResources.ts +26 -26
  200. package/src/test-utils/factories/websitalk/scans.ts +29 -29
  201. package/src/test-utils/factories/websitalk/websiteUrls.ts +17 -17
  202. package/src/utils/date.utils.ts +126 -126
  203. package/src/utils/shared.types.ts +4 -4
  204. package/src/utils/validation.ts +23 -23
  205. package/src/websitalk/index.ts +18 -18
  206. package/src/websitalk/knowledgeDocuments/__tests__/knowledgeDocuments.spec.ts +238 -238
  207. package/src/websitalk/knowledgeDocuments/index.ts +3 -3
  208. package/src/websitalk/knowledgeDocuments/knowledgeDocuments.constants.ts +26 -26
  209. package/src/websitalk/knowledgeDocuments/knowledgeDocuments.getters.ts +95 -95
  210. package/src/websitalk/knowledgeDocuments/knowledgeDocuments.types.ts +75 -75
  211. package/src/websitalk/mongodb-client.ts +61 -61
  212. package/src/websitalk/scanResources/__tests__/scanResources.spec.ts +248 -267
  213. package/src/websitalk/scanResources/index.ts +4 -4
  214. package/src/websitalk/scanResources/scanResources.constants.ts +60 -60
  215. package/src/websitalk/scanResources/scanResources.getters.ts +121 -128
  216. package/src/websitalk/scanResources/scanResources.types.ts +105 -105
  217. package/src/websitalk/scanResources/scanResources.utils.ts +31 -31
  218. package/src/websitalk/scans/__tests__/scans.spec.ts +313 -341
  219. package/src/websitalk/scans/index.ts +4 -4
  220. package/src/websitalk/scans/scans.constants.ts +15 -18
  221. package/src/websitalk/scans/scans.getters.ts +166 -171
  222. package/src/websitalk/scans/scans.types.ts +56 -57
  223. package/src/websitalk/scans/scans.utils.ts +5 -5
  224. package/src/websitalk/websiteUrls/__tests__/websiteUrls.spec.ts +77 -77
  225. package/src/websitalk/websiteUrls/index.ts +2 -2
  226. package/src/websitalk/websiteUrls/websiteUrls.getters.ts +65 -65
  227. package/src/websitalk/websiteUrls/websiteUrls.types.ts +13 -13
  228. package/tsconfig.json +23 -23
package/src/index.ts CHANGED
@@ -1,48 +1,47 @@
1
- // TalkPilot DB exports
2
- export * from "./talkpilot";
3
- export * from "./connection";
4
- export * from "./bulkWrite";
5
-
6
- export {
7
- municipalDataMongodbClient,
8
- getDb as getMunicipalDataDb,
9
- setDb as setMunicipalDataDb,
10
- ObjectId as MunicipalDataObjectId,
11
- } from "./municipal";
12
- export * from "./municipal/cities";
13
- export * from "./municipal/streets";
14
- export * from "./municipal/departmentsSubjects";
15
- export * from "./municipal/tickets";
16
- export * from "./municipal/systemInstructions";
17
- export * from "./municipal/muniIssues";
18
- export * from "./municipal/utils/types";
19
-
20
- export {
21
- websitalkMongodbClient,
22
- getDb as getWebsitalkDb,
23
- setDb as setWebsitalkDb,
24
- ObjectId as WebsitalkObjectId,
25
- } from "./websitalk";
26
- export * from "./websitalk/scans/scans.getters";
27
- export * from "./websitalk/scans/scans.constants";
28
- export type * from "./websitalk/scans/scans.types";
29
- export * from "./websitalk/scanResources";
30
- export * from "./websitalk/websiteUrls";
31
- export * from "./websitalk/knowledgeDocuments/knowledgeDocuments.constants";
32
- export * from "./websitalk/knowledgeDocuments/knowledgeDocuments.getters";
33
- export type * from "./websitalk/knowledgeDocuments/knowledgeDocuments.types";
34
-
35
- export {
36
- configurationMongodbClient,
37
- getDb as getConfigurationDb,
38
- setDb as setConfigurationDb,
39
- } from "./configuration";
40
- export * from "./configuration/prompts";
41
- export * from "./configuration/models";
42
- export * from "./configuration/internalModels";
43
-
44
- export { ObjectId } from "mongodb";
45
- export type { Filter } from "mongodb";
46
- export type { AnyBulkWriteOperation as BulkWriteOp } from "mongodb";
47
- export { MongoBulkWriteError as BulkWriteError } from "mongodb";
1
+ // TalkPilot DB exports
2
+ export * from "./talkpilot";
3
+ export * from "./connection";
4
+ export * from "./bulkWrite";
5
+
6
+ export {
7
+ municipalDataMongodbClient,
8
+ getDb as getMunicipalDataDb,
9
+ setDb as setMunicipalDataDb,
10
+ ObjectId as MunicipalDataObjectId,
11
+ } from "./municipal";
12
+ export * from "./municipal/cities";
13
+ export * from "./municipal/streets";
14
+ export * from "./municipal/departmentsSubjects";
15
+ export * from "./municipal/tickets";
16
+ export * from "./municipal/systemInstructions";
17
+ export * from "./municipal/muniIssues";
18
+ export * from "./municipal/utils/types";
19
+
20
+ export {
21
+ websitalkMongodbClient,
22
+ getDb as getWebsitalkDb,
23
+ setDb as setWebsitalkDb,
24
+ ObjectId as WebsitalkObjectId,
25
+ } from "./websitalk";
26
+ export * from "./websitalk/scans/scans.getters";
27
+ export type * from "./websitalk/scans/scans.types";
28
+ export * from "./websitalk/scanResources";
29
+ export * from "./websitalk/websiteUrls";
30
+ export * from "./websitalk/knowledgeDocuments/knowledgeDocuments.constants";
31
+ export * from "./websitalk/knowledgeDocuments/knowledgeDocuments.getters";
32
+ export type * from "./websitalk/knowledgeDocuments/knowledgeDocuments.types";
33
+
34
+ export {
35
+ configurationMongodbClient,
36
+ getDb as getConfigurationDb,
37
+ setDb as setConfigurationDb,
38
+ } from "./configuration";
39
+ export * from "./configuration/prompts";
40
+ export * from "./configuration/models";
41
+ export * from "./configuration/internalModels";
42
+
43
+ export { ObjectId } from "mongodb";
44
+ export type { Filter } from "mongodb";
45
+ export type { AnyBulkWriteOperation as BulkWriteOp } from "mongodb";
46
+ export { MongoBulkWriteError as BulkWriteError } from "mongodb";
48
47
  export type { OptionalUnlessRequiredId } from "mongodb";
@@ -1,62 +1,62 @@
1
- import { municipalDataMongodbClient } from "../mongodb-client";
2
- import { MongoClient } from "mongodb";
3
-
4
- describe("MunicipalDataMongoDBClient Validation", () => {
5
- const originalEnv = process.env;
6
-
7
- beforeEach(() => {
8
- process.env = { ...originalEnv };
9
- delete process.env.MONGO_URI;
10
- delete process.env.MONGODB_URI;
11
-
12
- // Force reset singleton state
13
- (municipalDataMongodbClient as any).client = null;
14
- (municipalDataMongodbClient as any).db = null;
15
- });
16
-
17
- afterAll(async () => {
18
- process.env = originalEnv;
19
- await municipalDataMongodbClient.disconnect();
20
- });
21
-
22
- it("should succeed with default database name if only URI is provided", async () => {
23
- const mockDb = { databaseName: "municipal-data" };
24
- const connectSpy = jest
25
- .spyOn(MongoClient.prototype, "connect")
26
- .mockResolvedValue({} as any);
27
- const dbSpy = jest
28
- .spyOn(MongoClient.prototype, "db")
29
- .mockReturnValue(mockDb as any);
30
-
31
- const validUri = "mongodb://127.0.0.1:27017";
32
- await municipalDataMongodbClient.connect(validUri);
33
-
34
- expect(municipalDataMongodbClient.getDb()).toBeDefined();
35
- expect(municipalDataMongodbClient.getDb().databaseName).toBe(
36
- "municipal-data",
37
- );
38
-
39
- connectSpy.mockRestore();
40
- dbSpy.mockRestore();
41
- });
42
-
43
- it("should allow overriding the default database name", async () => {
44
- const mockDb = { databaseName: "custom-municipal-db" };
45
- const connectSpy = jest
46
- .spyOn(MongoClient.prototype, "connect")
47
- .mockResolvedValue({} as any);
48
- const dbSpy = jest
49
- .spyOn(MongoClient.prototype, "db")
50
- .mockReturnValue(mockDb as any);
51
-
52
- const validUri = "mongodb://127.0.0.1:27017";
53
- await municipalDataMongodbClient.connect(validUri, "custom-municipal-db");
54
-
55
- expect(municipalDataMongodbClient.getDb().databaseName).toBe(
56
- "custom-municipal-db",
57
- );
58
-
59
- connectSpy.mockRestore();
60
- dbSpy.mockRestore();
61
- });
62
- });
1
+ import { municipalDataMongodbClient } from "../mongodb-client";
2
+ import { MongoClient } from "mongodb";
3
+
4
+ describe("MunicipalDataMongoDBClient Validation", () => {
5
+ const originalEnv = process.env;
6
+
7
+ beforeEach(() => {
8
+ process.env = { ...originalEnv };
9
+ delete process.env.MONGO_URI;
10
+ delete process.env.MONGODB_URI;
11
+
12
+ // Force reset singleton state
13
+ (municipalDataMongodbClient as any).client = null;
14
+ (municipalDataMongodbClient as any).db = null;
15
+ });
16
+
17
+ afterAll(async () => {
18
+ process.env = originalEnv;
19
+ await municipalDataMongodbClient.disconnect();
20
+ });
21
+
22
+ it("should succeed with default database name if only URI is provided", async () => {
23
+ const mockDb = { databaseName: "municipal-data" };
24
+ const connectSpy = jest
25
+ .spyOn(MongoClient.prototype, "connect")
26
+ .mockResolvedValue({} as any);
27
+ const dbSpy = jest
28
+ .spyOn(MongoClient.prototype, "db")
29
+ .mockReturnValue(mockDb as any);
30
+
31
+ const validUri = "mongodb://127.0.0.1:27017";
32
+ await municipalDataMongodbClient.connect(validUri);
33
+
34
+ expect(municipalDataMongodbClient.getDb()).toBeDefined();
35
+ expect(municipalDataMongodbClient.getDb().databaseName).toBe(
36
+ "municipal-data",
37
+ );
38
+
39
+ connectSpy.mockRestore();
40
+ dbSpy.mockRestore();
41
+ });
42
+
43
+ it("should allow overriding the default database name", async () => {
44
+ const mockDb = { databaseName: "custom-municipal-db" };
45
+ const connectSpy = jest
46
+ .spyOn(MongoClient.prototype, "connect")
47
+ .mockResolvedValue({} as any);
48
+ const dbSpy = jest
49
+ .spyOn(MongoClient.prototype, "db")
50
+ .mockReturnValue(mockDb as any);
51
+
52
+ const validUri = "mongodb://127.0.0.1:27017";
53
+ await municipalDataMongodbClient.connect(validUri, "custom-municipal-db");
54
+
55
+ expect(municipalDataMongodbClient.getDb().databaseName).toBe(
56
+ "custom-municipal-db",
57
+ );
58
+
59
+ connectSpy.mockRestore();
60
+ dbSpy.mockRestore();
61
+ });
62
+ });
@@ -1,50 +1,50 @@
1
- import { getDb, ObjectId, City } from "../index";
2
- import { Collection, Filter, ObjectId as MongoObjectId } from "mongodb";
3
-
4
- export const getCitiesCollection = (): Collection<City> => {
5
- return getDb().collection<City>("cities");
6
- };
7
-
8
- export const findCities = async (
9
- filter: Filter<City> = {},
10
- ): Promise<City[]> => {
11
- return await getCitiesCollection().find(filter).toArray();
12
- };
13
-
14
- export const getCityById = async (cityId: string): Promise<City | null> => {
15
- const city = await getCitiesCollection().findOne({
16
- _id: new ObjectId(cityId),
17
- });
18
- return city ? city : null;
19
- };
20
-
21
- export const createCity = async (
22
- cityData: Omit<City, "_id" | "createdAt" | "updatedAt">,
23
- ): Promise<MongoObjectId> => {
24
- const city: Omit<City, "_id"> = {
25
- ...cityData,
26
- createdAt: new Date(),
27
- updatedAt: new Date(),
28
- };
29
- const { insertedId } = await getCitiesCollection().insertOne(city as City);
30
- return insertedId;
31
- };
32
-
33
- export const updateCity = async (
34
- cityId: string,
35
- data: Partial<Omit<City, "_id" | "createdAt" | "updatedAt">>,
36
- ): Promise<City | null> => {
37
- const result = await getCitiesCollection().findOneAndUpdate(
38
- { _id: new ObjectId(cityId) },
39
- { $set: { ...data, updatedAt: new Date() } },
40
- { returnDocument: "after" },
41
- );
42
- return result || null;
43
- };
44
-
45
- export const deleteCity = async (cityId: string): Promise<boolean> => {
46
- const result = await getCitiesCollection().deleteOne({
47
- _id: new ObjectId(cityId),
48
- });
49
- return result.deletedCount > 0;
50
- };
1
+ import { getDb, ObjectId, City } from "../index";
2
+ import { Collection, Filter, ObjectId as MongoObjectId } from "mongodb";
3
+
4
+ export const getCitiesCollection = (): Collection<City> => {
5
+ return getDb().collection<City>("cities");
6
+ };
7
+
8
+ export const findCities = async (
9
+ filter: Filter<City> = {},
10
+ ): Promise<City[]> => {
11
+ return await getCitiesCollection().find(filter).toArray();
12
+ };
13
+
14
+ export const getCityById = async (cityId: string): Promise<City | null> => {
15
+ const city = await getCitiesCollection().findOne({
16
+ _id: new ObjectId(cityId),
17
+ });
18
+ return city ? city : null;
19
+ };
20
+
21
+ export const createCity = async (
22
+ cityData: Omit<City, "_id" | "createdAt" | "updatedAt">,
23
+ ): Promise<MongoObjectId> => {
24
+ const city: Omit<City, "_id"> = {
25
+ ...cityData,
26
+ createdAt: new Date(),
27
+ updatedAt: new Date(),
28
+ };
29
+ const { insertedId } = await getCitiesCollection().insertOne(city as City);
30
+ return insertedId;
31
+ };
32
+
33
+ export const updateCity = async (
34
+ cityId: string,
35
+ data: Partial<Omit<City, "_id" | "createdAt" | "updatedAt">>,
36
+ ): Promise<City | null> => {
37
+ const result = await getCitiesCollection().findOneAndUpdate(
38
+ { _id: new ObjectId(cityId) },
39
+ { $set: { ...data, updatedAt: new Date() } },
40
+ { returnDocument: "after" },
41
+ );
42
+ return result || null;
43
+ };
44
+
45
+ export const deleteCity = async (cityId: string): Promise<boolean> => {
46
+ const result = await getCitiesCollection().deleteOne({
47
+ _id: new ObjectId(cityId),
48
+ });
49
+ return result.deletedCount > 0;
50
+ };
@@ -1,11 +1,11 @@
1
- import { ObjectId, WithId } from "mongodb";
2
-
3
- export type City = {
4
- _id: ObjectId;
5
- createdAt: Date;
6
- updatedAt: Date;
7
- name: string;
8
- displayName: string;
9
- };
10
-
11
- export type CityDoc = WithId<City>;
1
+ import { ObjectId, WithId } from "mongodb";
2
+
3
+ export type City = {
4
+ _id: ObjectId;
5
+ createdAt: Date;
6
+ updatedAt: Date;
7
+ name: string;
8
+ displayName: string;
9
+ };
10
+
11
+ export type CityDoc = WithId<City>;
@@ -1,2 +1,2 @@
1
- export * from "./cities.getters";
2
- export type { City, CityDoc } from "./cities.types";
1
+ export * from "./cities.getters";
2
+ export type { City, CityDoc } from "./cities.types";