@talkpilot/core-db 1.3.59 → 1.3.60

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 (231) 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 -779
  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 +0 -12
  8. package/dist/configuration/models/models.types.d.ts.map +1 -1
  9. package/dist/index.d.ts +1 -0
  10. package/dist/index.d.ts.map +1 -1
  11. package/dist/index.js +1 -0
  12. package/dist/index.js.map +1 -1
  13. package/dist/test-utils/factories/websitalk/scans.d.ts.map +1 -1
  14. package/dist/test-utils/factories/websitalk/scans.js +1 -0
  15. package/dist/test-utils/factories/websitalk/scans.js.map +1 -1
  16. package/dist/websitalk/scanResources/scanResources.getters.d.ts +1 -0
  17. package/dist/websitalk/scanResources/scanResources.getters.d.ts.map +1 -1
  18. package/dist/websitalk/scanResources/scanResources.getters.js +6 -1
  19. package/dist/websitalk/scanResources/scanResources.getters.js.map +1 -1
  20. package/dist/websitalk/scans/scans.constants.d.ts +3 -1
  21. package/dist/websitalk/scans/scans.constants.d.ts.map +1 -1
  22. package/dist/websitalk/scans/scans.constants.js +3 -0
  23. package/dist/websitalk/scans/scans.constants.js.map +1 -1
  24. package/dist/websitalk/scans/scans.getters.d.ts +1 -0
  25. package/dist/websitalk/scans/scans.getters.d.ts.map +1 -1
  26. package/dist/websitalk/scans/scans.getters.js +7 -1
  27. package/dist/websitalk/scans/scans.getters.js.map +1 -1
  28. package/dist/websitalk/scans/scans.types.d.ts +1 -0
  29. package/dist/websitalk/scans/scans.types.d.ts.map +1 -1
  30. package/jest.config.js +20 -20
  31. package/package.json +46 -46
  32. package/src/__tests__/setup.ts +20 -20
  33. package/src/bulkWrite/__tests__/bulkWrite.spec.ts +65 -65
  34. package/src/bulkWrite/bulkWrite.ops.ts +45 -45
  35. package/src/bulkWrite/bulkWrite.types.ts +4 -4
  36. package/src/bulkWrite/index.ts +2 -2
  37. package/src/config.ts +9 -9
  38. package/src/configuration/index.ts +17 -17
  39. package/src/configuration/internalModels/index.ts +3 -3
  40. package/src/configuration/internalModels/internalModels.constants.ts +1 -1
  41. package/src/configuration/internalModels/internalModels.getters.ts +33 -33
  42. package/src/configuration/internalModels/internalModels.types.ts +12 -12
  43. package/src/configuration/models/__tests__/models.getters.spec.ts +67 -86
  44. package/src/configuration/models/index.ts +27 -28
  45. package/src/configuration/models/models.constants.ts +15 -15
  46. package/src/configuration/models/models.getters.ts +71 -71
  47. package/src/configuration/models/models.types.ts +80 -93
  48. package/src/configuration/mongodb-client.ts +62 -62
  49. package/src/configuration/prompts/index.ts +2 -2
  50. package/src/configuration/prompts/prompts.getters.ts +49 -49
  51. package/src/configuration/prompts/prompts.types.ts +11 -11
  52. package/src/connection.ts +81 -81
  53. package/src/index.ts +47 -46
  54. package/src/municipal/__tests__/validation.spec.ts +62 -62
  55. package/src/municipal/cities/cities.getters.ts +50 -50
  56. package/src/municipal/cities/cities.types.ts +11 -11
  57. package/src/municipal/cities/index.ts +2 -2
  58. package/src/municipal/departmentsSubjects/__tests__/departmentsSubjects.spec.ts +272 -272
  59. package/src/municipal/departmentsSubjects/departmentsSubjects.getters.ts +317 -317
  60. package/src/municipal/departmentsSubjects/departmentsSubjects.types.ts +73 -73
  61. package/src/municipal/departmentsSubjects/index.ts +9 -9
  62. package/src/municipal/index.ts +22 -22
  63. package/src/municipal/mongodb-client.ts +61 -61
  64. package/src/municipal/muniIssues/__tests__/muniIssues.setters.spec.ts +94 -94
  65. package/src/municipal/muniIssues/__tests__/muniIssues.setup.spec.ts +92 -92
  66. package/src/municipal/muniIssues/index.ts +17 -17
  67. package/src/municipal/muniIssues/muniIssues.constants.ts +18 -18
  68. package/src/municipal/muniIssues/muniIssues.getters.ts +24 -24
  69. package/src/municipal/muniIssues/muniIssues.schema.ts +66 -66
  70. package/src/municipal/muniIssues/muniIssues.setters.ts +24 -24
  71. package/src/municipal/muniIssues/muniIssues.setup.ts +65 -65
  72. package/src/municipal/muniIssues/muniIssues.types.ts +37 -37
  73. package/src/municipal/streets/__tests__/streets.spec.ts +253 -253
  74. package/src/municipal/streets/index.ts +2 -2
  75. package/src/municipal/streets/streets.getters.ts +140 -140
  76. package/src/municipal/streets/streets.types.ts +19 -19
  77. package/src/municipal/systemInstructions/__tests__/getters.spec.ts +113 -113
  78. package/src/municipal/systemInstructions/__tests__/setters.spec.ts +274 -274
  79. package/src/municipal/systemInstructions/index.ts +7 -7
  80. package/src/municipal/systemInstructions/instructions.getters.ts +57 -57
  81. package/src/municipal/systemInstructions/instructions.setters.ts +119 -119
  82. package/src/municipal/systemInstructions/instructions.types.ts +30 -30
  83. package/src/municipal/tickets/__tests__/tickets.getters.spec.ts +30 -30
  84. package/src/municipal/tickets/__tests__/tickets.statistics.spec.ts +99 -99
  85. package/src/municipal/tickets/index.ts +9 -9
  86. package/src/municipal/tickets/tickets.constants.ts +8 -8
  87. package/src/municipal/tickets/tickets.getters.ts +121 -121
  88. package/src/municipal/tickets/tickets.statistics.aggregation.ts +110 -110
  89. package/src/municipal/tickets/tickets.statistics.getters.ts +145 -145
  90. package/src/municipal/tickets/tickets.types.ts +54 -54
  91. package/src/municipal/utils/types.ts +11 -11
  92. package/src/talkpilot/__tests__/db.spec.ts +38 -38
  93. package/src/talkpilot/__tests__/mongodb-client.spec.ts +18 -18
  94. package/src/talkpilot/__tests__/validation.spec.ts +68 -68
  95. package/src/talkpilot/agents/__tests__/agents.getters.spec.ts +29 -29
  96. package/src/talkpilot/agents/agents.getters.ts +34 -34
  97. package/src/talkpilot/agents/agents.types.ts +14 -14
  98. package/src/talkpilot/agents/index.ts +2 -2
  99. package/src/talkpilot/backgroundToolResults/__tests__/backgroundToolResults.getters.spec.ts +147 -147
  100. package/src/talkpilot/backgroundToolResults/backgroundToolResults.getters.ts +65 -65
  101. package/src/talkpilot/backgroundToolResults/backgroundToolResults.types.ts +23 -23
  102. package/src/talkpilot/backgroundToolResults/index.ts +2 -2
  103. package/src/talkpilot/calls/__tests__/callStats.utils.spec.ts +128 -128
  104. package/src/talkpilot/calls/__tests__/calls.dashboard.spec.ts +229 -229
  105. package/src/talkpilot/calls/__tests__/calls.spec.ts +471 -471
  106. package/src/talkpilot/calls/__tests__/calls.statistics.spec.ts +483 -483
  107. package/src/talkpilot/calls/calls.constants.ts +49 -49
  108. package/src/talkpilot/calls/calls.getters.ts +277 -277
  109. package/src/talkpilot/calls/calls.statistics.getters.ts +668 -668
  110. package/src/talkpilot/calls/calls.statistics.types.ts +48 -48
  111. package/src/talkpilot/calls/calls.types.ts +149 -149
  112. package/src/talkpilot/calls/dashboard/calls.dashboard.ts +295 -295
  113. package/src/talkpilot/calls/dashboard/calls.dashboard.types.ts +57 -57
  114. package/src/talkpilot/calls/index.ts +6 -6
  115. package/src/talkpilot/clientAudioBuffers/__tests__/clientAudioBuffer.getters.spec.ts +160 -160
  116. package/src/talkpilot/clientAudioBuffers/clientAudioBuffer.getters.ts +117 -117
  117. package/src/talkpilot/clientAudioBuffers/clientsAudioBuffers.types.ts +25 -25
  118. package/src/talkpilot/clientAudioBuffers/index.ts +2 -2
  119. package/src/talkpilot/clients/__tests__/clients.quota.spec.ts +123 -123
  120. package/src/talkpilot/clients/clients.getters.ts +16 -16
  121. package/src/talkpilot/clients/clients.quota.getters.ts +119 -119
  122. package/src/talkpilot/clients/clients.quota.types.ts +27 -27
  123. package/src/talkpilot/clients/clients.types.ts +18 -18
  124. package/src/talkpilot/clients/index.ts +4 -4
  125. package/src/talkpilot/clientsConfig/__tests__/clientsConfig.getters.spec.ts +423 -423
  126. package/src/talkpilot/clientsConfig/__tests__/clientsConfig.spec.ts +212 -212
  127. package/src/talkpilot/clientsConfig/clientsConfig.constants.ts +2 -2
  128. package/src/talkpilot/clientsConfig/clientsConfig.getters.ts +247 -247
  129. package/src/talkpilot/clientsConfig/clientsConfig.types.ts +136 -136
  130. package/src/talkpilot/clientsConfig/index.ts +3 -3
  131. package/src/talkpilot/contextNotes/__tests__/contextNotes.getters.spec.ts +217 -217
  132. package/src/talkpilot/contextNotes/contextNotes.getters.ts +85 -85
  133. package/src/talkpilot/contextNotes/contextNotes.types.ts +20 -20
  134. package/src/talkpilot/contextNotes/index.ts +2 -2
  135. package/src/talkpilot/flows/__tests__/flows.schema.spec.ts +124 -124
  136. package/src/talkpilot/flows/flows.getter.ts +34 -34
  137. package/src/talkpilot/flows/flows.schema.ts +180 -180
  138. package/src/talkpilot/flows/flows.types.ts +211 -211
  139. package/src/talkpilot/flows/index.ts +7 -7
  140. package/src/talkpilot/groups/__tests__/groups.spec.ts +90 -90
  141. package/src/talkpilot/groups/__tests__/phone.utils.spec.ts +32 -32
  142. package/src/talkpilot/groups/groups.getters.ts +30 -30
  143. package/src/talkpilot/groups/groups.types.ts +29 -29
  144. package/src/talkpilot/groups/index.ts +3 -3
  145. package/src/talkpilot/groups/phone.utils.ts +46 -46
  146. package/src/talkpilot/index.ts +31 -31
  147. package/src/talkpilot/leads/index.ts +2 -2
  148. package/src/talkpilot/leads/leads.getter.ts +6 -6
  149. package/src/talkpilot/leads/leads.schema.ts +33 -33
  150. package/src/talkpilot/leads/leads.types.ts +20 -20
  151. package/src/talkpilot/mongodb-client.ts +78 -78
  152. package/src/talkpilot/phone_numbers/__tests__/phone_numbers.spec.ts +282 -282
  153. package/src/talkpilot/phone_numbers/index.ts +2 -2
  154. package/src/talkpilot/phone_numbers/phone_numbers.getter.ts +176 -176
  155. package/src/talkpilot/phone_numbers/phone_numbers.schema.ts +17 -17
  156. package/src/talkpilot/phone_numbers/phone_numbers.types.ts +30 -30
  157. package/src/talkpilot/plans/__tests__/plans.spec.ts +70 -70
  158. package/src/talkpilot/plans/index.ts +2 -2
  159. package/src/talkpilot/plans/plans.getters.ts +132 -132
  160. package/src/talkpilot/plans/plans.types.ts +89 -89
  161. package/src/talkpilot/products/__tests__/products.getters.spec.ts +44 -44
  162. package/src/talkpilot/products/index.ts +2 -2
  163. package/src/talkpilot/products/products.getters.ts +12 -12
  164. package/src/talkpilot/products/products.types.ts +9 -9
  165. package/src/talkpilot/results/index.ts +7 -7
  166. package/src/talkpilot/results/results.getter.ts +39 -39
  167. package/src/talkpilot/results/results.schema.ts +25 -25
  168. package/src/talkpilot/results/results.types.ts +34 -34
  169. package/src/talkpilot/retry_analyze/__tests__/retryAnalyze.getters.spec.ts +156 -156
  170. package/src/talkpilot/retry_analyze/index.ts +2 -2
  171. package/src/talkpilot/retry_analyze/retryAnalyze.getters.ts +84 -84
  172. package/src/talkpilot/retry_analyze/retryAnalyze.types.ts +13 -13
  173. package/src/talkpilot/sessions/__tests__/sessions.spec.ts +147 -147
  174. package/src/talkpilot/sessions/index.ts +2 -2
  175. package/src/talkpilot/sessions/sessions.getter.ts +92 -92
  176. package/src/talkpilot/sessions/sessions.schema.ts +34 -34
  177. package/src/talkpilot/sessions/sessions.types.ts +30 -30
  178. package/src/talkpilot/subscriptions/__tests__/subscriptions.getters.utils.spec.ts +45 -45
  179. package/src/talkpilot/subscriptions/index.ts +3 -3
  180. package/src/talkpilot/subscriptions/subscriptions.getters.ts +146 -146
  181. package/src/talkpilot/subscriptions/subscriptions.getters.utils.ts +33 -33
  182. package/src/talkpilot/subscriptions/subscriptions.types.ts +66 -66
  183. package/src/talkpilot/utils/__tests__/query.utils.spec.ts +49 -49
  184. package/src/talkpilot/utils/query.utils.ts +21 -21
  185. package/src/test-utils/db-utils.ts +33 -33
  186. package/src/test-utils/factories/index.ts +18 -18
  187. package/src/test-utils/factories/municipal/cities.ts +16 -16
  188. package/src/test-utils/factories/municipal/departmentsSubjects.ts +37 -37
  189. package/src/test-utils/factories/municipal/muniIssues.ts +32 -32
  190. package/src/test-utils/factories/municipal/streets.ts +22 -22
  191. package/src/test-utils/factories/municipal/tickets.ts +39 -39
  192. package/src/test-utils/factories/talkpilot/agents.ts +19 -19
  193. package/src/test-utils/factories/talkpilot/calls.ts +37 -37
  194. package/src/test-utils/factories/talkpilot/clientAudioBuffers.ts +20 -20
  195. package/src/test-utils/factories/talkpilot/clientsConfig.ts +18 -18
  196. package/src/test-utils/factories/talkpilot/contextNotes.ts +40 -40
  197. package/src/test-utils/factories/talkpilot/flows.ts +33 -33
  198. package/src/test-utils/factories/talkpilot/groups.ts +33 -33
  199. package/src/test-utils/factories/talkpilot/phone_numbers.ts +22 -22
  200. package/src/test-utils/factories/talkpilot/sessions.ts +35 -35
  201. package/src/test-utils/factories/websitalk/knowledgeDocuments.ts +29 -29
  202. package/src/test-utils/factories/websitalk/scanResources.ts +26 -26
  203. package/src/test-utils/factories/websitalk/scans.ts +30 -29
  204. package/src/test-utils/factories/websitalk/websiteUrls.ts +17 -17
  205. package/src/utils/date.utils.ts +126 -126
  206. package/src/utils/shared.types.ts +4 -4
  207. package/src/utils/validation.ts +23 -23
  208. package/src/websitalk/index.ts +18 -18
  209. package/src/websitalk/knowledgeDocuments/__tests__/knowledgeDocuments.spec.ts +238 -238
  210. package/src/websitalk/knowledgeDocuments/index.ts +3 -3
  211. package/src/websitalk/knowledgeDocuments/knowledgeDocuments.constants.ts +26 -26
  212. package/src/websitalk/knowledgeDocuments/knowledgeDocuments.getters.ts +95 -95
  213. package/src/websitalk/knowledgeDocuments/knowledgeDocuments.types.ts +75 -75
  214. package/src/websitalk/mongodb-client.ts +61 -61
  215. package/src/websitalk/scanResources/__tests__/scanResources.spec.ts +267 -248
  216. package/src/websitalk/scanResources/index.ts +4 -4
  217. package/src/websitalk/scanResources/scanResources.constants.ts +60 -60
  218. package/src/websitalk/scanResources/scanResources.getters.ts +128 -121
  219. package/src/websitalk/scanResources/scanResources.types.ts +105 -105
  220. package/src/websitalk/scanResources/scanResources.utils.ts +31 -31
  221. package/src/websitalk/scans/__tests__/scans.spec.ts +49 -0
  222. package/src/websitalk/scans/index.ts +4 -4
  223. package/src/websitalk/scans/scans.constants.ts +18 -15
  224. package/src/websitalk/scans/scans.getters.ts +172 -166
  225. package/src/websitalk/scans/scans.types.ts +61 -56
  226. package/src/websitalk/scans/scans.utils.ts +5 -5
  227. package/src/websitalk/websiteUrls/__tests__/websiteUrls.spec.ts +77 -77
  228. package/src/websitalk/websiteUrls/index.ts +2 -2
  229. package/src/websitalk/websiteUrls/websiteUrls.getters.ts +65 -65
  230. package/src/websitalk/websiteUrls/websiteUrls.types.ts +13 -13
  231. package/tsconfig.json +23 -23
package/src/index.ts CHANGED
@@ -1,47 +1,48 @@
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";
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";
47
48
  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";