@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
@@ -1,341 +1,313 @@
1
- import {
2
- getScansCollection,
3
- getScansByClientAndDateRange,
4
- getScanFieldsByClientAndDateRange,
5
- findScansByQuery,
6
- countScans,
7
- createScanDoc,
8
- updateScanDoc,
9
- findInProgressScansByClient,
10
- getScansByWebsiteUrl,
11
- getLatestScanByWebsiteUrl,
12
- getInProgressScanByWebsiteUrl,
13
- getLatestCompletedScanByWebsiteUrl,
14
- deleteScansByWebsiteUrl,
15
- deleteScanById,
16
- } from "../scans.getters";
17
- import { FINISHED_SCAN_STATUSES, SCAN_STATUSES } from "../scans.constants";
18
- import { createScanFixture } from "../../../test-utils/factories";
19
- import { ObjectId } from "mongodb";
20
-
21
- describe("db.websitalk.scans", () => {
22
- describe("getScansByClientAndDateRange()", () => {
23
- it("should return full scan documents within the date range", async () => {
24
- const clientId = "dateRangeClient";
25
- const startDate = new Date("2023-05-01");
26
- const endDate = new Date("2023-05-31");
27
-
28
- const scanInside = createScanFixture({ clientId, status: "COMPLETED" });
29
- const scanBefore = createScanFixture({ clientId });
30
- const scanAfter = createScanFixture({ clientId });
31
-
32
- await getScansCollection().insertMany([
33
- { ...scanInside, createdAt: new Date("2023-05-15") },
34
- { ...scanBefore, createdAt: new Date("2023-04-30") },
35
- { ...scanAfter, createdAt: new Date("2023-06-01") },
36
- ]);
37
-
38
- const result = await getScansByClientAndDateRange(
39
- clientId,
40
- startDate,
41
- endDate,
42
- );
43
-
44
- expect(result.length).toBe(1);
45
- expect(result[0]).toMatchObject({
46
- clientId,
47
- websiteUrlId: scanInside.websiteUrlId,
48
- status: "COMPLETED",
49
- });
50
- expect(result[0]).toHaveProperty("siteChrome");
51
- expect(result[0]).toHaveProperty("skipped");
52
- });
53
-
54
- it("should respect the date range boundaries (inclusive)", async () => {
55
- const clientId = "boundaryClient";
56
- const onStart = createScanFixture({ clientId });
57
- const onEnd = createScanFixture({ clientId });
58
-
59
- await getScansCollection().insertMany([
60
- { ...onStart, createdAt: new Date("2023-05-01") },
61
- { ...onEnd, createdAt: new Date("2023-05-31") },
62
- ]);
63
-
64
- const result = await getScansByClientAndDateRange(
65
- clientId,
66
- new Date("2023-05-01"),
67
- new Date("2023-05-31"),
68
- );
69
-
70
- expect(result.length).toBe(2);
71
- });
72
- });
73
-
74
- describe("getScanFieldsByClientAndDateRange()", () => {
75
- it("should return only the requested fields (plus _id)", async () => {
76
- const clientId = "projectionClient";
77
- const scan = createScanFixture({
78
- clientId,
79
- status: "COMPLETED",
80
- siteChrome: { url: "https://example.com", sourceType: "site_chrome" },
81
- skipped: [{ path: "/broken", reason: "HTTP 500" }],
82
- });
83
-
84
- await getScansCollection().insertOne({
85
- ...scan,
86
- createdAt: new Date("2023-05-15"),
87
- });
88
-
89
- const [result] = await getScanFieldsByClientAndDateRange(
90
- clientId,
91
- new Date("2023-05-01"),
92
- new Date("2023-05-31"),
93
- ["clientId", "websiteUrlId", "status"],
94
- );
95
-
96
- expect(result).toMatchObject({
97
- clientId,
98
- websiteUrlId: scan.websiteUrlId,
99
- status: "COMPLETED",
100
- });
101
- expect(result).toHaveProperty("_id");
102
- expect(result).not.toHaveProperty("siteChrome");
103
- expect(result).not.toHaveProperty("skipped");
104
- expect(result).not.toHaveProperty("createdAt");
105
- });
106
- });
107
-
108
- describe("findScansByQuery() / countScans()", () => {
109
- it("should find scans matching a raw filter and count them", async () => {
110
- const clientId = "queryClient";
111
- const completed = createScanFixture({ clientId, status: "COMPLETED" });
112
- const scanning = createScanFixture({ clientId, status: "SCANNING" });
113
-
114
- await getScansCollection().insertMany([completed, scanning]);
115
-
116
- const result = await findScansByQuery({ clientId, status: "COMPLETED" });
117
- const count = await countScans({ clientId });
118
-
119
- expect(result.length).toBe(1);
120
- expect(result[0].websiteUrlId).toBe(completed.websiteUrlId);
121
- expect(count).toBe(2);
122
- });
123
- });
124
-
125
- describe("createScanDoc()", () => {
126
- it("should insert a scan with default values for the rest of the fields", async () => {
127
- const clientId = "createClient";
128
- const websiteUrlId = new ObjectId().toString();
129
-
130
- const { insertedId } = await createScanDoc({ clientId, websiteUrlId });
131
-
132
- const result = await getScansCollection().findOne({ _id: insertedId });
133
-
134
- expect(result).toMatchObject({
135
- clientId,
136
- websiteUrlId,
137
- status: "CHECKING",
138
- activeTasksCount: 0,
139
- pagesScanned: 0,
140
- pagesSkipped: 0,
141
- skipped: [],
142
- siteChrome: {},
143
- });
144
- expect(result?.createdAt).toBeInstanceOf(Date);
145
- expect(result?.updatedAt).toBeInstanceOf(Date);
146
- });
147
- });
148
-
149
- describe("updateScanDoc()", () => {
150
- it("should update the given fields and bump updatedAt", async () => {
151
- const scan = createScanFixture({ status: "SCANNING", pagesScanned: 0 });
152
- const { insertedId } = await getScansCollection().insertOne(scan);
153
-
154
- const result = await updateScanDoc(insertedId, {
155
- status: "COMPLETED",
156
- pagesScanned: 10,
157
- });
158
-
159
- expect(result).toMatchObject({
160
- _id: insertedId,
161
- status: "COMPLETED",
162
- pagesScanned: 10,
163
- });
164
- expect(result?.updatedAt.getTime()).toBeGreaterThan(
165
- scan.updatedAt.getTime(),
166
- );
167
- });
168
-
169
- it("should return null when the scan does not exist", async () => {
170
- const result = await updateScanDoc(new ObjectId(), { status: "FAILED" });
171
-
172
- expect(result).toBeNull();
173
- });
174
- });
175
-
176
- describe("in-progress scan helpers", () => {
177
- const websiteUrlId = "website-123";
178
-
179
- it("should return website scan history newest first", async () => {
180
- await getScansCollection().insertMany([
181
- createScanFixture({
182
- websiteUrlId,
183
- status: "COMPLETED",
184
- createdAt: new Date("2023-05-01"),
185
- }),
186
- createScanFixture({
187
- websiteUrlId,
188
- status: "COMPLETED",
189
- createdAt: new Date("2023-05-15"),
190
- }),
191
- ]);
192
-
193
- const scans = await getScansByWebsiteUrl(websiteUrlId);
194
- expect(scans).toHaveLength(2);
195
- expect(scans[0].createdAt.getTime()).toBeGreaterThan(
196
- scans[1].createdAt.getTime(),
197
- );
198
- });
199
-
200
- it("should return the newest scan for a website", async () => {
201
- await getScansCollection().insertMany([
202
- createScanFixture({
203
- websiteUrlId,
204
- status: "COMPLETED",
205
- createdAt: new Date("2023-05-01"),
206
- }),
207
- createScanFixture({
208
- websiteUrlId,
209
- status: "FAILED",
210
- createdAt: new Date("2023-05-15"),
211
- }),
212
- ]);
213
-
214
- const latest = await getLatestScanByWebsiteUrl(websiteUrlId);
215
- expect(latest?.status).toBe("FAILED");
216
- });
217
-
218
- it("should return in-progress scan and return null when only completed exists", async () => {
219
- await getScansCollection().insertMany([
220
- createScanFixture({
221
- websiteUrlId,
222
- status: "COMPLETED",
223
- createdAt: new Date("2023-05-01"),
224
- }),
225
- createScanFixture({
226
- websiteUrlId,
227
- status: "SCANNING",
228
- createdAt: new Date("2023-05-15"),
229
- }),
230
- ]);
231
-
232
- const inProgressScan = await getInProgressScanByWebsiteUrl(websiteUrlId);
233
- expect(inProgressScan?.status).toBe("SCANNING");
234
-
235
- await getScansCollection().deleteMany({ websiteUrlId });
236
- await getScansCollection().insertOne(
237
- createScanFixture({
238
- websiteUrlId,
239
- status: "COMPLETED",
240
- createdAt: new Date("2023-05-20"),
241
- }),
242
- );
243
-
244
- expect(await getInProgressScanByWebsiteUrl(websiteUrlId)).toBeNull();
245
- });
246
-
247
- it("should return all in-progress scans across websites for a client", async () => {
248
- const clientId = "multiScanClient";
249
- await getScansCollection().insertMany([
250
- createScanFixture({
251
- clientId,
252
- websiteUrlId: "website-a",
253
- status: "SCANNING",
254
- }),
255
- createScanFixture({
256
- clientId,
257
- websiteUrlId: "website-b",
258
- status: "PROCESSING",
259
- }),
260
- createScanFixture({
261
- clientId,
262
- websiteUrlId: "website-c",
263
- status: "COMPLETED",
264
- }),
265
- ]);
266
-
267
- const scans = await findInProgressScansByClient(clientId);
268
- expect(scans).toHaveLength(2);
269
- expect(scans.map((s) => s.websiteUrlId).sort()).toEqual([
270
- "website-a",
271
- "website-b",
272
- ]);
273
- });
274
- });
275
-
276
- describe("website delete helpers", () => {
277
- it("should return the newest completed scan only", async () => {
278
- const websiteUrlId = new ObjectId().toString();
279
- await getScansCollection().insertMany([
280
- createScanFixture({
281
- websiteUrlId,
282
- status: "FAILED",
283
- createdAt: new Date("2023-05-25"),
284
- }),
285
- createScanFixture({
286
- websiteUrlId,
287
- status: "COMPLETED",
288
- createdAt: new Date("2023-05-20"),
289
- }),
290
- createScanFixture({
291
- websiteUrlId,
292
- status: "COMPLETED",
293
- createdAt: new Date("2023-05-22"),
294
- }),
295
- ]);
296
-
297
- const latest = await getLatestCompletedScanByWebsiteUrl(websiteUrlId);
298
- expect(latest?.status).toBe("COMPLETED");
299
- expect(latest?.createdAt).toEqual(new Date("2023-05-22"));
300
- });
301
-
302
- it("should delete all scans for the website and return count", async () => {
303
- const websiteUrlId = new ObjectId().toString();
304
- await getScansCollection().insertMany([
305
- createScanFixture({ websiteUrlId }),
306
- createScanFixture({ websiteUrlId }),
307
- createScanFixture({ websiteUrlId: new ObjectId().toString() }),
308
- ]);
309
-
310
- const deleted = await deleteScansByWebsiteUrl(websiteUrlId);
311
- expect(deleted).toBe(2);
312
- expect(await countScans({ websiteUrlId })).toBe(0);
313
- });
314
- });
315
-
316
- describe("deleteScanById()", () => {
317
- it("should delete the scan and return true", async () => {
318
- const { insertedId } =
319
- await getScansCollection().insertOne(createScanFixture());
320
-
321
- expect(await deleteScanById(insertedId)).toBe(true);
322
- expect(
323
- await getScansCollection().findOne({ _id: insertedId }),
324
- ).toBeNull();
325
- });
326
-
327
- it("should return false when the scan does not exist", async () => {
328
- expect(await deleteScanById(new ObjectId())).toBe(false);
329
- });
330
- });
331
-
332
- describe("FINISHED_SCAN_STATUSES", () => {
333
- it("should include CANCELED — canceling is final, not resumable", () => {
334
- expect(FINISHED_SCAN_STATUSES).toContain(SCAN_STATUSES.CANCELED);
335
- });
336
-
337
- it("should NOT include PAUSED — pausing is reversible", () => {
338
- expect(FINISHED_SCAN_STATUSES).not.toContain(SCAN_STATUSES.PAUSED);
339
- });
340
- });
341
- });
1
+ import {
2
+ getScansCollection,
3
+ getScansByClientAndDateRange,
4
+ getScanFieldsByClientAndDateRange,
5
+ findScansByQuery,
6
+ countScans,
7
+ createScanDoc,
8
+ updateScanDoc,
9
+ findInProgressScansByClient,
10
+ getScansByWebsiteUrl,
11
+ getLatestScanByWebsiteUrl,
12
+ getInProgressScanByWebsiteUrl,
13
+ getLatestCompletedScanByWebsiteUrl,
14
+ deleteScansByWebsiteUrl,
15
+ } from "../scans.getters";
16
+ import { createScanFixture } from "../../../test-utils/factories";
17
+ import { ObjectId } from "mongodb";
18
+
19
+ describe("db.websitalk.scans", () => {
20
+ describe("getScansByClientAndDateRange()", () => {
21
+ it("should return full scan documents within the date range", async () => {
22
+ const clientId = "dateRangeClient";
23
+ const startDate = new Date("2023-05-01");
24
+ const endDate = new Date("2023-05-31");
25
+
26
+ const scanInside = createScanFixture({ clientId, status: "COMPLETED" });
27
+ const scanBefore = createScanFixture({ clientId });
28
+ const scanAfter = createScanFixture({ clientId });
29
+
30
+ await getScansCollection().insertMany([
31
+ { ...scanInside, createdAt: new Date("2023-05-15") },
32
+ { ...scanBefore, createdAt: new Date("2023-04-30") },
33
+ { ...scanAfter, createdAt: new Date("2023-06-01") },
34
+ ]);
35
+
36
+ const result = await getScansByClientAndDateRange(
37
+ clientId,
38
+ startDate,
39
+ endDate,
40
+ );
41
+
42
+ expect(result.length).toBe(1);
43
+ expect(result[0]).toMatchObject({
44
+ clientId,
45
+ websiteUrlId: scanInside.websiteUrlId,
46
+ status: "COMPLETED",
47
+ });
48
+ expect(result[0]).toHaveProperty("siteChrome");
49
+ expect(result[0]).toHaveProperty("skipped");
50
+ });
51
+
52
+ it("should respect the date range boundaries (inclusive)", async () => {
53
+ const clientId = "boundaryClient";
54
+ const onStart = createScanFixture({ clientId });
55
+ const onEnd = createScanFixture({ clientId });
56
+
57
+ await getScansCollection().insertMany([
58
+ { ...onStart, createdAt: new Date("2023-05-01") },
59
+ { ...onEnd, createdAt: new Date("2023-05-31") },
60
+ ]);
61
+
62
+ const result = await getScansByClientAndDateRange(
63
+ clientId,
64
+ new Date("2023-05-01"),
65
+ new Date("2023-05-31"),
66
+ );
67
+
68
+ expect(result.length).toBe(2);
69
+ });
70
+ });
71
+
72
+ describe("getScanFieldsByClientAndDateRange()", () => {
73
+ it("should return only the requested fields (plus _id)", async () => {
74
+ const clientId = "projectionClient";
75
+ const scan = createScanFixture({
76
+ clientId,
77
+ status: "COMPLETED",
78
+ siteChrome: { url: "https://example.com", sourceType: "site_chrome" },
79
+ skipped: [{ path: "/broken", reason: "HTTP 500" }],
80
+ });
81
+
82
+ await getScansCollection().insertOne({
83
+ ...scan,
84
+ createdAt: new Date("2023-05-15"),
85
+ });
86
+
87
+ const [result] = await getScanFieldsByClientAndDateRange(
88
+ clientId,
89
+ new Date("2023-05-01"),
90
+ new Date("2023-05-31"),
91
+ ["clientId", "websiteUrlId", "status"],
92
+ );
93
+
94
+ expect(result).toMatchObject({
95
+ clientId,
96
+ websiteUrlId: scan.websiteUrlId,
97
+ status: "COMPLETED",
98
+ });
99
+ expect(result).toHaveProperty("_id");
100
+ expect(result).not.toHaveProperty("siteChrome");
101
+ expect(result).not.toHaveProperty("skipped");
102
+ expect(result).not.toHaveProperty("createdAt");
103
+ });
104
+ });
105
+
106
+ describe("findScansByQuery() / countScans()", () => {
107
+ it("should find scans matching a raw filter and count them", async () => {
108
+ const clientId = "queryClient";
109
+ const completed = createScanFixture({ clientId, status: "COMPLETED" });
110
+ const scanning = createScanFixture({ clientId, status: "SCANNING" });
111
+
112
+ await getScansCollection().insertMany([completed, scanning]);
113
+
114
+ const result = await findScansByQuery({ clientId, status: "COMPLETED" });
115
+ const count = await countScans({ clientId });
116
+
117
+ expect(result.length).toBe(1);
118
+ expect(result[0].websiteUrlId).toBe(completed.websiteUrlId);
119
+ expect(count).toBe(2);
120
+ });
121
+ });
122
+
123
+ describe("createScanDoc()", () => {
124
+ it("should insert a scan with default values for the rest of the fields", async () => {
125
+ const clientId = "createClient";
126
+ const websiteUrlId = new ObjectId().toString();
127
+
128
+ const { insertedId } = await createScanDoc({ clientId, websiteUrlId });
129
+
130
+ const result = await getScansCollection().findOne({ _id: insertedId });
131
+
132
+ expect(result).toMatchObject({
133
+ clientId,
134
+ websiteUrlId,
135
+ status: "CHECKING",
136
+ activeTasksCount: 0,
137
+ pagesScanned: 0,
138
+ pagesSkipped: 0,
139
+ skipped: [],
140
+ siteChrome: {},
141
+ });
142
+ expect(result?.createdAt).toBeInstanceOf(Date);
143
+ expect(result?.updatedAt).toBeInstanceOf(Date);
144
+ });
145
+ });
146
+
147
+ describe("updateScanDoc()", () => {
148
+ it("should update the given fields and bump updatedAt", async () => {
149
+ const scan = createScanFixture({ status: "SCANNING", pagesScanned: 0 });
150
+ const { insertedId } = await getScansCollection().insertOne(scan);
151
+
152
+ const result = await updateScanDoc(insertedId, {
153
+ status: "COMPLETED",
154
+ pagesScanned: 10,
155
+ });
156
+
157
+ expect(result).toMatchObject({
158
+ _id: insertedId,
159
+ status: "COMPLETED",
160
+ pagesScanned: 10,
161
+ });
162
+ expect(result?.updatedAt.getTime()).toBeGreaterThan(
163
+ scan.updatedAt.getTime(),
164
+ );
165
+ });
166
+
167
+ it("should return null when the scan does not exist", async () => {
168
+ const result = await updateScanDoc(new ObjectId(), { status: "FAILED" });
169
+
170
+ expect(result).toBeNull();
171
+ });
172
+ });
173
+
174
+ describe("in-progress scan helpers", () => {
175
+ const websiteUrlId = "website-123";
176
+
177
+ it("should return website scan history newest first", async () => {
178
+ await getScansCollection().insertMany([
179
+ createScanFixture({
180
+ websiteUrlId,
181
+ status: "COMPLETED",
182
+ createdAt: new Date("2023-05-01"),
183
+ }),
184
+ createScanFixture({
185
+ websiteUrlId,
186
+ status: "COMPLETED",
187
+ createdAt: new Date("2023-05-15"),
188
+ }),
189
+ ]);
190
+
191
+ const scans = await getScansByWebsiteUrl(websiteUrlId);
192
+ expect(scans).toHaveLength(2);
193
+ expect(scans[0].createdAt.getTime()).toBeGreaterThan(
194
+ scans[1].createdAt.getTime(),
195
+ );
196
+ });
197
+
198
+ it("should return the newest scan for a website", async () => {
199
+ await getScansCollection().insertMany([
200
+ createScanFixture({
201
+ websiteUrlId,
202
+ status: "COMPLETED",
203
+ createdAt: new Date("2023-05-01"),
204
+ }),
205
+ createScanFixture({
206
+ websiteUrlId,
207
+ status: "FAILED",
208
+ createdAt: new Date("2023-05-15"),
209
+ }),
210
+ ]);
211
+
212
+ const latest = await getLatestScanByWebsiteUrl(websiteUrlId);
213
+ expect(latest?.status).toBe("FAILED");
214
+ });
215
+
216
+ it("should return in-progress scan and return null when only completed exists", async () => {
217
+ await getScansCollection().insertMany([
218
+ createScanFixture({
219
+ websiteUrlId,
220
+ status: "COMPLETED",
221
+ createdAt: new Date("2023-05-01"),
222
+ }),
223
+ createScanFixture({
224
+ websiteUrlId,
225
+ status: "SCANNING",
226
+ createdAt: new Date("2023-05-15"),
227
+ }),
228
+ ]);
229
+
230
+ const inProgressScan = await getInProgressScanByWebsiteUrl(websiteUrlId);
231
+ expect(inProgressScan?.status).toBe("SCANNING");
232
+
233
+ await getScansCollection().deleteMany({ websiteUrlId });
234
+ await getScansCollection().insertOne(
235
+ createScanFixture({
236
+ websiteUrlId,
237
+ status: "COMPLETED",
238
+ createdAt: new Date("2023-05-20"),
239
+ }),
240
+ );
241
+
242
+ expect(await getInProgressScanByWebsiteUrl(websiteUrlId)).toBeNull();
243
+ });
244
+
245
+ it("should return all in-progress scans across websites for a client", async () => {
246
+ const clientId = "multiScanClient";
247
+ await getScansCollection().insertMany([
248
+ createScanFixture({
249
+ clientId,
250
+ websiteUrlId: "website-a",
251
+ status: "SCANNING",
252
+ }),
253
+ createScanFixture({
254
+ clientId,
255
+ websiteUrlId: "website-b",
256
+ status: "PROCESSING",
257
+ }),
258
+ createScanFixture({
259
+ clientId,
260
+ websiteUrlId: "website-c",
261
+ status: "COMPLETED",
262
+ }),
263
+ ]);
264
+
265
+ const scans = await findInProgressScansByClient(clientId);
266
+ expect(scans).toHaveLength(2);
267
+ expect(scans.map((s) => s.websiteUrlId).sort()).toEqual([
268
+ "website-a",
269
+ "website-b",
270
+ ]);
271
+ });
272
+ });
273
+
274
+ describe("website delete helpers", () => {
275
+ it("should return the newest completed scan only", async () => {
276
+ const websiteUrlId = new ObjectId().toString();
277
+ await getScansCollection().insertMany([
278
+ createScanFixture({
279
+ websiteUrlId,
280
+ status: "FAILED",
281
+ createdAt: new Date("2023-05-25"),
282
+ }),
283
+ createScanFixture({
284
+ websiteUrlId,
285
+ status: "COMPLETED",
286
+ createdAt: new Date("2023-05-20"),
287
+ }),
288
+ createScanFixture({
289
+ websiteUrlId,
290
+ status: "COMPLETED",
291
+ createdAt: new Date("2023-05-22"),
292
+ }),
293
+ ]);
294
+
295
+ const latest = await getLatestCompletedScanByWebsiteUrl(websiteUrlId);
296
+ expect(latest?.status).toBe("COMPLETED");
297
+ expect(latest?.createdAt).toEqual(new Date("2023-05-22"));
298
+ });
299
+
300
+ it("should delete all scans for the website and return count", async () => {
301
+ const websiteUrlId = new ObjectId().toString();
302
+ await getScansCollection().insertMany([
303
+ createScanFixture({ websiteUrlId }),
304
+ createScanFixture({ websiteUrlId }),
305
+ createScanFixture({ websiteUrlId: new ObjectId().toString() }),
306
+ ]);
307
+
308
+ const deleted = await deleteScansByWebsiteUrl(websiteUrlId);
309
+ expect(deleted).toBe(2);
310
+ expect(await countScans({ websiteUrlId })).toBe(0);
311
+ });
312
+ });
313
+ });