@talkpilot/core-db 1.3.84 → 1.3.86

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 (229) 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 +847 -847
  5. package/dist/talkpilot/flows/flows.schema.d.ts +9 -0
  6. package/dist/talkpilot/flows/flows.schema.d.ts.map +1 -1
  7. package/dist/talkpilot/flows/flows.schema.js +7 -0
  8. package/dist/talkpilot/flows/flows.schema.js.map +1 -1
  9. package/dist/talkpilot/flows/flows.types.d.ts +3 -0
  10. package/dist/talkpilot/flows/flows.types.d.ts.map +1 -1
  11. package/dist/talkpilot/subscriptions/subscriptions.utils.d.ts +4 -0
  12. package/dist/talkpilot/subscriptions/subscriptions.utils.d.ts.map +1 -0
  13. package/dist/talkpilot/subscriptions/subscriptions.utils.js +20 -0
  14. package/dist/talkpilot/subscriptions/subscriptions.utils.js.map +1 -0
  15. package/jest.config.js +20 -20
  16. package/package.json +46 -46
  17. package/src/__tests__/setup.ts +20 -20
  18. package/src/bulkWrite/__tests__/bulkWrite.spec.ts +65 -65
  19. package/src/bulkWrite/bulkWrite.ops.ts +45 -45
  20. package/src/bulkWrite/bulkWrite.types.ts +4 -4
  21. package/src/bulkWrite/index.ts +2 -2
  22. package/src/config.ts +9 -9
  23. package/src/configuration/index.ts +17 -17
  24. package/src/configuration/internalModels/index.ts +3 -3
  25. package/src/configuration/internalModels/internalModels.constants.ts +1 -1
  26. package/src/configuration/internalModels/internalModels.getters.ts +33 -33
  27. package/src/configuration/internalModels/internalModels.types.ts +12 -12
  28. package/src/configuration/models/__tests__/models.getters.spec.ts +67 -67
  29. package/src/configuration/models/index.ts +27 -27
  30. package/src/configuration/models/models.constants.ts +20 -20
  31. package/src/configuration/models/models.getters.ts +71 -71
  32. package/src/configuration/models/models.types.ts +80 -80
  33. package/src/configuration/mongodb-client.ts +62 -62
  34. package/src/configuration/prompts/index.ts +2 -2
  35. package/src/configuration/prompts/prompts.getters.ts +49 -49
  36. package/src/configuration/prompts/prompts.types.ts +11 -11
  37. package/src/connection.ts +81 -81
  38. package/src/index.ts +48 -48
  39. package/src/municipal/__tests__/validation.spec.ts +62 -62
  40. package/src/municipal/cities/cities.getters.ts +50 -50
  41. package/src/municipal/cities/cities.types.ts +11 -11
  42. package/src/municipal/cities/index.ts +2 -2
  43. package/src/municipal/departmentsSubjects/__tests__/departmentsSubjects.spec.ts +272 -272
  44. package/src/municipal/departmentsSubjects/departmentsSubjects.getters.ts +317 -317
  45. package/src/municipal/departmentsSubjects/departmentsSubjects.types.ts +73 -73
  46. package/src/municipal/departmentsSubjects/index.ts +9 -9
  47. package/src/municipal/index.ts +23 -23
  48. package/src/municipal/mongodb-client.ts +61 -61
  49. package/src/municipal/muniIssues/__tests__/muniIssues.getters.spec.ts +87 -87
  50. package/src/municipal/muniIssues/__tests__/muniIssues.setters.spec.ts +219 -219
  51. package/src/municipal/muniIssues/__tests__/muniIssues.setup.spec.ts +93 -93
  52. package/src/municipal/muniIssues/index.ts +24 -24
  53. package/src/municipal/muniIssues/muniIssues.constants.ts +27 -27
  54. package/src/municipal/muniIssues/muniIssues.getters.ts +73 -73
  55. package/src/municipal/muniIssues/muniIssues.schema.ts +85 -85
  56. package/src/municipal/muniIssues/muniIssues.setters.ts +87 -87
  57. package/src/municipal/muniIssues/muniIssues.setup.ts +65 -65
  58. package/src/municipal/muniIssues/muniIssues.types.ts +67 -67
  59. package/src/municipal/residents/__tests__/residents.spec.ts +92 -92
  60. package/src/municipal/residents/index.ts +4 -4
  61. package/src/municipal/residents/residents.constants.ts +2 -2
  62. package/src/municipal/residents/residents.getters.ts +42 -42
  63. package/src/municipal/residents/residents.schema.ts +28 -28
  64. package/src/municipal/residents/residents.types.ts +17 -17
  65. package/src/municipal/streets/__tests__/streets.spec.ts +253 -253
  66. package/src/municipal/streets/index.ts +2 -2
  67. package/src/municipal/streets/streets.getters.ts +140 -140
  68. package/src/municipal/streets/streets.types.ts +19 -19
  69. package/src/municipal/systemInstructions/__tests__/getters.spec.ts +113 -113
  70. package/src/municipal/systemInstructions/__tests__/setters.spec.ts +274 -274
  71. package/src/municipal/systemInstructions/index.ts +7 -7
  72. package/src/municipal/systemInstructions/instructions.getters.ts +57 -57
  73. package/src/municipal/systemInstructions/instructions.setters.ts +119 -119
  74. package/src/municipal/systemInstructions/instructions.types.ts +30 -30
  75. package/src/municipal/tickets/__tests__/tickets.getters.spec.ts +30 -30
  76. package/src/municipal/tickets/__tests__/tickets.statistics.spec.ts +99 -99
  77. package/src/municipal/tickets/index.ts +9 -9
  78. package/src/municipal/tickets/tickets.constants.ts +8 -8
  79. package/src/municipal/tickets/tickets.getters.ts +121 -121
  80. package/src/municipal/tickets/tickets.statistics.aggregation.ts +110 -110
  81. package/src/municipal/tickets/tickets.statistics.getters.ts +145 -145
  82. package/src/municipal/tickets/tickets.types.ts +54 -54
  83. package/src/municipal/utils/types.ts +11 -11
  84. package/src/talkpilot/__tests__/db.spec.ts +38 -38
  85. package/src/talkpilot/__tests__/mongodb-client.spec.ts +18 -18
  86. package/src/talkpilot/__tests__/validation.spec.ts +68 -68
  87. package/src/talkpilot/agents/__tests__/agents.getters.spec.ts +29 -29
  88. package/src/talkpilot/agents/agents.getters.ts +34 -34
  89. package/src/talkpilot/agents/agents.types.ts +14 -14
  90. package/src/talkpilot/agents/index.ts +2 -2
  91. package/src/talkpilot/backgroundToolResults/__tests__/backgroundToolResults.getters.spec.ts +147 -147
  92. package/src/talkpilot/backgroundToolResults/backgroundToolResults.getters.ts +65 -65
  93. package/src/talkpilot/backgroundToolResults/backgroundToolResults.types.ts +23 -23
  94. package/src/talkpilot/backgroundToolResults/index.ts +2 -2
  95. package/src/talkpilot/callTokenUsage/__tests__/callTokenUsage.getters.spec.ts +57 -57
  96. package/src/talkpilot/callTokenUsage/callTokenUsage.getters.ts +42 -42
  97. package/src/talkpilot/callTokenUsage/callTokenUsage.types.ts +28 -28
  98. package/src/talkpilot/callTokenUsage/index.ts +2 -2
  99. package/src/talkpilot/calls/__tests__/callStats.utils.spec.ts +128 -128
  100. package/src/talkpilot/calls/__tests__/calls.dashboard.spec.ts +229 -229
  101. package/src/talkpilot/calls/__tests__/calls.spec.ts +519 -519
  102. package/src/talkpilot/calls/__tests__/calls.statistics.spec.ts +483 -483
  103. package/src/talkpilot/calls/calls.constants.ts +57 -57
  104. package/src/talkpilot/calls/calls.getters.ts +295 -295
  105. package/src/talkpilot/calls/calls.statistics.getters.ts +668 -668
  106. package/src/talkpilot/calls/calls.statistics.types.ts +48 -48
  107. package/src/talkpilot/calls/calls.types.ts +168 -168
  108. package/src/talkpilot/calls/dashboard/calls.dashboard.ts +295 -295
  109. package/src/talkpilot/calls/dashboard/calls.dashboard.types.ts +57 -57
  110. package/src/talkpilot/calls/index.ts +6 -6
  111. package/src/talkpilot/clientAudioBuffers/__tests__/clientAudioBuffer.getters.spec.ts +160 -160
  112. package/src/talkpilot/clientAudioBuffers/clientAudioBuffer.getters.ts +117 -117
  113. package/src/talkpilot/clientAudioBuffers/clientsAudioBuffers.types.ts +25 -25
  114. package/src/talkpilot/clientAudioBuffers/index.ts +2 -2
  115. package/src/talkpilot/clients/__tests__/clients.quota.spec.ts +123 -123
  116. package/src/talkpilot/clients/clients.getters.ts +19 -19
  117. package/src/talkpilot/clients/clients.quota.getters.ts +119 -119
  118. package/src/talkpilot/clients/clients.quota.types.ts +27 -27
  119. package/src/talkpilot/clients/clients.types.ts +18 -18
  120. package/src/talkpilot/clients/index.ts +4 -4
  121. package/src/talkpilot/clientsConfig/__tests__/clientsConfig.getters.spec.ts +423 -423
  122. package/src/talkpilot/clientsConfig/__tests__/clientsConfig.spec.ts +212 -212
  123. package/src/talkpilot/clientsConfig/clientsConfig.constants.ts +2 -2
  124. package/src/talkpilot/clientsConfig/clientsConfig.getters.ts +247 -247
  125. package/src/talkpilot/clientsConfig/clientsConfig.types.ts +137 -137
  126. package/src/talkpilot/clientsConfig/index.ts +3 -3
  127. package/src/talkpilot/contextNotes/__tests__/contextNotes.getters.spec.ts +217 -217
  128. package/src/talkpilot/contextNotes/contextNotes.getters.ts +85 -85
  129. package/src/talkpilot/contextNotes/contextNotes.types.ts +20 -20
  130. package/src/talkpilot/contextNotes/index.ts +2 -2
  131. package/src/talkpilot/flows/__tests__/flows.schema.spec.ts +163 -152
  132. package/src/talkpilot/flows/flows.getter.ts +34 -34
  133. package/src/talkpilot/flows/flows.schema.ts +203 -196
  134. package/src/talkpilot/flows/flows.types.ts +221 -220
  135. package/src/talkpilot/flows/index.ts +7 -7
  136. package/src/talkpilot/groups/__tests__/groups.spec.ts +90 -90
  137. package/src/talkpilot/groups/__tests__/phone.utils.spec.ts +32 -32
  138. package/src/talkpilot/groups/groups.getters.ts +30 -30
  139. package/src/talkpilot/groups/groups.types.ts +29 -29
  140. package/src/talkpilot/groups/index.ts +3 -3
  141. package/src/talkpilot/groups/phone.utils.ts +46 -46
  142. package/src/talkpilot/index.ts +32 -32
  143. package/src/talkpilot/leads/index.ts +2 -2
  144. package/src/talkpilot/leads/leads.getter.ts +6 -6
  145. package/src/talkpilot/leads/leads.schema.ts +33 -33
  146. package/src/talkpilot/leads/leads.types.ts +20 -20
  147. package/src/talkpilot/mongodb-client.ts +78 -78
  148. package/src/talkpilot/phone_numbers/__tests__/phone_numbers.spec.ts +530 -530
  149. package/src/talkpilot/phone_numbers/index.ts +2 -2
  150. package/src/talkpilot/phone_numbers/phone_numbers.getter.ts +240 -240
  151. package/src/talkpilot/phone_numbers/phone_numbers.schema.ts +17 -17
  152. package/src/talkpilot/phone_numbers/phone_numbers.types.ts +30 -30
  153. package/src/talkpilot/phone_numbers/phone_numbers.utils.ts +9 -9
  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 +19 -19
  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/residents.ts +20 -20
  188. package/src/test-utils/factories/municipal/streets.ts +22 -22
  189. package/src/test-utils/factories/municipal/tickets.ts +39 -39
  190. package/src/test-utils/factories/talkpilot/agents.ts +19 -19
  191. package/src/test-utils/factories/talkpilot/calls.ts +37 -37
  192. package/src/test-utils/factories/talkpilot/clientAudioBuffers.ts +20 -20
  193. package/src/test-utils/factories/talkpilot/clientsConfig.ts +18 -18
  194. package/src/test-utils/factories/talkpilot/contextNotes.ts +40 -40
  195. package/src/test-utils/factories/talkpilot/flows.ts +33 -33
  196. package/src/test-utils/factories/talkpilot/groups.ts +33 -33
  197. package/src/test-utils/factories/talkpilot/phone_numbers.ts +22 -22
  198. package/src/test-utils/factories/talkpilot/sessions.ts +35 -35
  199. package/src/test-utils/factories/websitalk/knowledgeDocuments.ts +29 -29
  200. package/src/test-utils/factories/websitalk/scanResources.ts +26 -26
  201. package/src/test-utils/factories/websitalk/scans.ts +30 -30
  202. package/src/test-utils/factories/websitalk/websiteUrls.ts +17 -17
  203. package/src/utils/date.utils.ts +126 -126
  204. package/src/utils/shared.types.ts +4 -4
  205. package/src/utils/validation.ts +23 -23
  206. package/src/websitalk/index.ts +18 -18
  207. package/src/websitalk/knowledgeDocuments/__tests__/knowledgeDocuments.spec.ts +238 -238
  208. package/src/websitalk/knowledgeDocuments/index.ts +3 -3
  209. package/src/websitalk/knowledgeDocuments/knowledgeDocuments.constants.ts +26 -26
  210. package/src/websitalk/knowledgeDocuments/knowledgeDocuments.getters.ts +95 -95
  211. package/src/websitalk/knowledgeDocuments/knowledgeDocuments.types.ts +75 -75
  212. package/src/websitalk/mongodb-client.ts +61 -61
  213. package/src/websitalk/scanResources/__tests__/scanResources.spec.ts +267 -267
  214. package/src/websitalk/scanResources/index.ts +4 -4
  215. package/src/websitalk/scanResources/scanResources.constants.ts +60 -60
  216. package/src/websitalk/scanResources/scanResources.getters.ts +128 -128
  217. package/src/websitalk/scanResources/scanResources.types.ts +105 -105
  218. package/src/websitalk/scanResources/scanResources.utils.ts +31 -31
  219. package/src/websitalk/scans/__tests__/scans.spec.ts +362 -362
  220. package/src/websitalk/scans/index.ts +4 -4
  221. package/src/websitalk/scans/scans.constants.ts +18 -18
  222. package/src/websitalk/scans/scans.getters.ts +172 -172
  223. package/src/websitalk/scans/scans.types.ts +58 -58
  224. package/src/websitalk/scans/scans.utils.ts +5 -5
  225. package/src/websitalk/websiteUrls/__tests__/websiteUrls.spec.ts +77 -77
  226. package/src/websitalk/websiteUrls/index.ts +2 -2
  227. package/src/websitalk/websiteUrls/websiteUrls.getters.ts +65 -65
  228. package/src/websitalk/websiteUrls/websiteUrls.types.ts +13 -13
  229. package/tsconfig.json +23 -23
@@ -1,362 +1,362 @@
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
- pausedAt: null,
144
- });
145
- expect(result?.createdAt).toBeInstanceOf(Date);
146
- expect(result?.updatedAt).toBeInstanceOf(Date);
147
- });
148
- });
149
-
150
- describe("pausedAt", () => {
151
- it("is set on pause and cleared back to null on resume, via updateScanDoc", async () => {
152
- const scan = createScanFixture({ status: "SCANNING", pausedAt: null });
153
- const { insertedId } = await getScansCollection().insertOne(scan);
154
-
155
- const pausedAt = new Date();
156
- const paused = await updateScanDoc(insertedId, {
157
- status: "PAUSED",
158
- pausedAt,
159
- });
160
- expect(paused?.pausedAt).toEqual(pausedAt);
161
-
162
- const resumed = await updateScanDoc(insertedId, {
163
- status: "SCANNING",
164
- pausedAt: null,
165
- });
166
- expect(resumed?.pausedAt).toBeNull();
167
- });
168
- });
169
-
170
- describe("updateScanDoc()", () => {
171
- it("should update the given fields and bump updatedAt", async () => {
172
- const scan = createScanFixture({ status: "SCANNING", pagesScanned: 0 });
173
- const { insertedId } = await getScansCollection().insertOne(scan);
174
-
175
- const result = await updateScanDoc(insertedId, {
176
- status: "COMPLETED",
177
- pagesScanned: 10,
178
- });
179
-
180
- expect(result).toMatchObject({
181
- _id: insertedId,
182
- status: "COMPLETED",
183
- pagesScanned: 10,
184
- });
185
- expect(result?.updatedAt.getTime()).toBeGreaterThan(
186
- scan.updatedAt.getTime(),
187
- );
188
- });
189
-
190
- it("should return null when the scan does not exist", async () => {
191
- const result = await updateScanDoc(new ObjectId(), { status: "FAILED" });
192
-
193
- expect(result).toBeNull();
194
- });
195
- });
196
-
197
- describe("in-progress scan helpers", () => {
198
- const websiteUrlId = "website-123";
199
-
200
- it("should return website scan history newest first", 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: "COMPLETED",
210
- createdAt: new Date("2023-05-15"),
211
- }),
212
- ]);
213
-
214
- const scans = await getScansByWebsiteUrl(websiteUrlId);
215
- expect(scans).toHaveLength(2);
216
- expect(scans[0].createdAt.getTime()).toBeGreaterThan(
217
- scans[1].createdAt.getTime(),
218
- );
219
- });
220
-
221
- it("should return the newest scan for a website", async () => {
222
- await getScansCollection().insertMany([
223
- createScanFixture({
224
- websiteUrlId,
225
- status: "COMPLETED",
226
- createdAt: new Date("2023-05-01"),
227
- }),
228
- createScanFixture({
229
- websiteUrlId,
230
- status: "FAILED",
231
- createdAt: new Date("2023-05-15"),
232
- }),
233
- ]);
234
-
235
- const latest = await getLatestScanByWebsiteUrl(websiteUrlId);
236
- expect(latest?.status).toBe("FAILED");
237
- });
238
-
239
- it("should return in-progress scan and return null when only completed exists", async () => {
240
- await getScansCollection().insertMany([
241
- createScanFixture({
242
- websiteUrlId,
243
- status: "COMPLETED",
244
- createdAt: new Date("2023-05-01"),
245
- }),
246
- createScanFixture({
247
- websiteUrlId,
248
- status: "SCANNING",
249
- createdAt: new Date("2023-05-15"),
250
- }),
251
- ]);
252
-
253
- const inProgressScan = await getInProgressScanByWebsiteUrl(websiteUrlId);
254
- expect(inProgressScan?.status).toBe("SCANNING");
255
-
256
- await getScansCollection().deleteMany({ websiteUrlId });
257
- await getScansCollection().insertOne(
258
- createScanFixture({
259
- websiteUrlId,
260
- status: "COMPLETED",
261
- createdAt: new Date("2023-05-20"),
262
- }),
263
- );
264
-
265
- expect(await getInProgressScanByWebsiteUrl(websiteUrlId)).toBeNull();
266
- });
267
-
268
- it("should return all in-progress scans across websites for a client", async () => {
269
- const clientId = "multiScanClient";
270
- await getScansCollection().insertMany([
271
- createScanFixture({
272
- clientId,
273
- websiteUrlId: "website-a",
274
- status: "SCANNING",
275
- }),
276
- createScanFixture({
277
- clientId,
278
- websiteUrlId: "website-b",
279
- status: "PROCESSING",
280
- }),
281
- createScanFixture({
282
- clientId,
283
- websiteUrlId: "website-c",
284
- status: "COMPLETED",
285
- }),
286
- ]);
287
-
288
- const scans = await findInProgressScansByClient(clientId);
289
- expect(scans).toHaveLength(2);
290
- expect(scans.map((s) => s.websiteUrlId).sort()).toEqual([
291
- "website-a",
292
- "website-b",
293
- ]);
294
- });
295
- });
296
-
297
- describe("website delete helpers", () => {
298
- it("should return the newest completed scan only", async () => {
299
- const websiteUrlId = new ObjectId().toString();
300
- await getScansCollection().insertMany([
301
- createScanFixture({
302
- websiteUrlId,
303
- status: "FAILED",
304
- createdAt: new Date("2023-05-25"),
305
- }),
306
- createScanFixture({
307
- websiteUrlId,
308
- status: "COMPLETED",
309
- createdAt: new Date("2023-05-20"),
310
- }),
311
- createScanFixture({
312
- websiteUrlId,
313
- status: "COMPLETED",
314
- createdAt: new Date("2023-05-22"),
315
- }),
316
- ]);
317
-
318
- const latest = await getLatestCompletedScanByWebsiteUrl(websiteUrlId);
319
- expect(latest?.status).toBe("COMPLETED");
320
- expect(latest?.createdAt).toEqual(new Date("2023-05-22"));
321
- });
322
-
323
- it("should delete all scans for the website and return count", async () => {
324
- const websiteUrlId = new ObjectId().toString();
325
- await getScansCollection().insertMany([
326
- createScanFixture({ websiteUrlId }),
327
- createScanFixture({ websiteUrlId }),
328
- createScanFixture({ websiteUrlId: new ObjectId().toString() }),
329
- ]);
330
-
331
- const deleted = await deleteScansByWebsiteUrl(websiteUrlId);
332
- expect(deleted).toBe(2);
333
- expect(await countScans({ websiteUrlId })).toBe(0);
334
- });
335
- });
336
-
337
- describe("deleteScanById()", () => {
338
- it("should delete the scan and return true", async () => {
339
- const { insertedId } =
340
- await getScansCollection().insertOne(createScanFixture());
341
-
342
- expect(await deleteScanById(insertedId)).toBe(true);
343
- expect(
344
- await getScansCollection().findOne({ _id: insertedId }),
345
- ).toBeNull();
346
- });
347
-
348
- it("should return false when the scan does not exist", async () => {
349
- expect(await deleteScanById(new ObjectId())).toBe(false);
350
- });
351
- });
352
-
353
- describe("FINISHED_SCAN_STATUSES", () => {
354
- it("should include CANCELED — canceling is final, not resumable", () => {
355
- expect(FINISHED_SCAN_STATUSES).toContain(SCAN_STATUSES.CANCELED);
356
- });
357
-
358
- it("should NOT include PAUSED — pausing is reversible", () => {
359
- expect(FINISHED_SCAN_STATUSES).not.toContain(SCAN_STATUSES.PAUSED);
360
- });
361
- });
362
- });
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
+ pausedAt: null,
144
+ });
145
+ expect(result?.createdAt).toBeInstanceOf(Date);
146
+ expect(result?.updatedAt).toBeInstanceOf(Date);
147
+ });
148
+ });
149
+
150
+ describe("pausedAt", () => {
151
+ it("is set on pause and cleared back to null on resume, via updateScanDoc", async () => {
152
+ const scan = createScanFixture({ status: "SCANNING", pausedAt: null });
153
+ const { insertedId } = await getScansCollection().insertOne(scan);
154
+
155
+ const pausedAt = new Date();
156
+ const paused = await updateScanDoc(insertedId, {
157
+ status: "PAUSED",
158
+ pausedAt,
159
+ });
160
+ expect(paused?.pausedAt).toEqual(pausedAt);
161
+
162
+ const resumed = await updateScanDoc(insertedId, {
163
+ status: "SCANNING",
164
+ pausedAt: null,
165
+ });
166
+ expect(resumed?.pausedAt).toBeNull();
167
+ });
168
+ });
169
+
170
+ describe("updateScanDoc()", () => {
171
+ it("should update the given fields and bump updatedAt", async () => {
172
+ const scan = createScanFixture({ status: "SCANNING", pagesScanned: 0 });
173
+ const { insertedId } = await getScansCollection().insertOne(scan);
174
+
175
+ const result = await updateScanDoc(insertedId, {
176
+ status: "COMPLETED",
177
+ pagesScanned: 10,
178
+ });
179
+
180
+ expect(result).toMatchObject({
181
+ _id: insertedId,
182
+ status: "COMPLETED",
183
+ pagesScanned: 10,
184
+ });
185
+ expect(result?.updatedAt.getTime()).toBeGreaterThan(
186
+ scan.updatedAt.getTime(),
187
+ );
188
+ });
189
+
190
+ it("should return null when the scan does not exist", async () => {
191
+ const result = await updateScanDoc(new ObjectId(), { status: "FAILED" });
192
+
193
+ expect(result).toBeNull();
194
+ });
195
+ });
196
+
197
+ describe("in-progress scan helpers", () => {
198
+ const websiteUrlId = "website-123";
199
+
200
+ it("should return website scan history newest first", 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: "COMPLETED",
210
+ createdAt: new Date("2023-05-15"),
211
+ }),
212
+ ]);
213
+
214
+ const scans = await getScansByWebsiteUrl(websiteUrlId);
215
+ expect(scans).toHaveLength(2);
216
+ expect(scans[0].createdAt.getTime()).toBeGreaterThan(
217
+ scans[1].createdAt.getTime(),
218
+ );
219
+ });
220
+
221
+ it("should return the newest scan for a website", async () => {
222
+ await getScansCollection().insertMany([
223
+ createScanFixture({
224
+ websiteUrlId,
225
+ status: "COMPLETED",
226
+ createdAt: new Date("2023-05-01"),
227
+ }),
228
+ createScanFixture({
229
+ websiteUrlId,
230
+ status: "FAILED",
231
+ createdAt: new Date("2023-05-15"),
232
+ }),
233
+ ]);
234
+
235
+ const latest = await getLatestScanByWebsiteUrl(websiteUrlId);
236
+ expect(latest?.status).toBe("FAILED");
237
+ });
238
+
239
+ it("should return in-progress scan and return null when only completed exists", async () => {
240
+ await getScansCollection().insertMany([
241
+ createScanFixture({
242
+ websiteUrlId,
243
+ status: "COMPLETED",
244
+ createdAt: new Date("2023-05-01"),
245
+ }),
246
+ createScanFixture({
247
+ websiteUrlId,
248
+ status: "SCANNING",
249
+ createdAt: new Date("2023-05-15"),
250
+ }),
251
+ ]);
252
+
253
+ const inProgressScan = await getInProgressScanByWebsiteUrl(websiteUrlId);
254
+ expect(inProgressScan?.status).toBe("SCANNING");
255
+
256
+ await getScansCollection().deleteMany({ websiteUrlId });
257
+ await getScansCollection().insertOne(
258
+ createScanFixture({
259
+ websiteUrlId,
260
+ status: "COMPLETED",
261
+ createdAt: new Date("2023-05-20"),
262
+ }),
263
+ );
264
+
265
+ expect(await getInProgressScanByWebsiteUrl(websiteUrlId)).toBeNull();
266
+ });
267
+
268
+ it("should return all in-progress scans across websites for a client", async () => {
269
+ const clientId = "multiScanClient";
270
+ await getScansCollection().insertMany([
271
+ createScanFixture({
272
+ clientId,
273
+ websiteUrlId: "website-a",
274
+ status: "SCANNING",
275
+ }),
276
+ createScanFixture({
277
+ clientId,
278
+ websiteUrlId: "website-b",
279
+ status: "PROCESSING",
280
+ }),
281
+ createScanFixture({
282
+ clientId,
283
+ websiteUrlId: "website-c",
284
+ status: "COMPLETED",
285
+ }),
286
+ ]);
287
+
288
+ const scans = await findInProgressScansByClient(clientId);
289
+ expect(scans).toHaveLength(2);
290
+ expect(scans.map((s) => s.websiteUrlId).sort()).toEqual([
291
+ "website-a",
292
+ "website-b",
293
+ ]);
294
+ });
295
+ });
296
+
297
+ describe("website delete helpers", () => {
298
+ it("should return the newest completed scan only", async () => {
299
+ const websiteUrlId = new ObjectId().toString();
300
+ await getScansCollection().insertMany([
301
+ createScanFixture({
302
+ websiteUrlId,
303
+ status: "FAILED",
304
+ createdAt: new Date("2023-05-25"),
305
+ }),
306
+ createScanFixture({
307
+ websiteUrlId,
308
+ status: "COMPLETED",
309
+ createdAt: new Date("2023-05-20"),
310
+ }),
311
+ createScanFixture({
312
+ websiteUrlId,
313
+ status: "COMPLETED",
314
+ createdAt: new Date("2023-05-22"),
315
+ }),
316
+ ]);
317
+
318
+ const latest = await getLatestCompletedScanByWebsiteUrl(websiteUrlId);
319
+ expect(latest?.status).toBe("COMPLETED");
320
+ expect(latest?.createdAt).toEqual(new Date("2023-05-22"));
321
+ });
322
+
323
+ it("should delete all scans for the website and return count", async () => {
324
+ const websiteUrlId = new ObjectId().toString();
325
+ await getScansCollection().insertMany([
326
+ createScanFixture({ websiteUrlId }),
327
+ createScanFixture({ websiteUrlId }),
328
+ createScanFixture({ websiteUrlId: new ObjectId().toString() }),
329
+ ]);
330
+
331
+ const deleted = await deleteScansByWebsiteUrl(websiteUrlId);
332
+ expect(deleted).toBe(2);
333
+ expect(await countScans({ websiteUrlId })).toBe(0);
334
+ });
335
+ });
336
+
337
+ describe("deleteScanById()", () => {
338
+ it("should delete the scan and return true", async () => {
339
+ const { insertedId } =
340
+ await getScansCollection().insertOne(createScanFixture());
341
+
342
+ expect(await deleteScanById(insertedId)).toBe(true);
343
+ expect(
344
+ await getScansCollection().findOne({ _id: insertedId }),
345
+ ).toBeNull();
346
+ });
347
+
348
+ it("should return false when the scan does not exist", async () => {
349
+ expect(await deleteScanById(new ObjectId())).toBe(false);
350
+ });
351
+ });
352
+
353
+ describe("FINISHED_SCAN_STATUSES", () => {
354
+ it("should include CANCELED — canceling is final, not resumable", () => {
355
+ expect(FINISHED_SCAN_STATUSES).toContain(SCAN_STATUSES.CANCELED);
356
+ });
357
+
358
+ it("should NOT include PAUSED — pausing is reversible", () => {
359
+ expect(FINISHED_SCAN_STATUSES).not.toContain(SCAN_STATUSES.PAUSED);
360
+ });
361
+ });
362
+ });