@talkpilot/core-db 1.3.39 → 1.3.40

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