@talkpilot/core-db 1.3.21 → 1.3.22

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 (190) hide show
  1. package/.cursor/rules/development.mdc +65 -65
  2. package/DEVELOPMENT.md +141 -141
  3. package/README.md +494 -500
  4. package/dist/talkpilot/calls/calls.constants.d.ts +0 -2
  5. package/dist/talkpilot/calls/calls.constants.d.ts.map +1 -1
  6. package/dist/talkpilot/calls/calls.constants.js +1 -3
  7. package/dist/talkpilot/calls/calls.constants.js.map +1 -1
  8. package/dist/talkpilot/calls/calls.getters.d.ts +0 -3
  9. package/dist/talkpilot/calls/calls.getters.d.ts.map +1 -1
  10. package/dist/talkpilot/calls/calls.getters.js +1 -12
  11. package/dist/talkpilot/calls/calls.getters.js.map +1 -1
  12. package/dist/talkpilot/calls/calls.types.d.ts +1 -10
  13. package/dist/talkpilot/calls/calls.types.d.ts.map +1 -1
  14. package/dist/websitalk/websiteUrls/websiteUrls.context.d.ts +11 -0
  15. package/dist/websitalk/websiteUrls/websiteUrls.context.d.ts.map +1 -0
  16. package/dist/websitalk/websiteUrls/websiteUrls.context.js +35 -0
  17. package/dist/websitalk/websiteUrls/websiteUrls.context.js.map +1 -0
  18. package/dist/websitalk/websiteUrls/websiteUrls.setters.d.ts +7 -0
  19. package/dist/websitalk/websiteUrls/websiteUrls.setters.d.ts.map +1 -0
  20. package/dist/websitalk/websiteUrls/websiteUrls.setters.js +29 -0
  21. package/dist/websitalk/websiteUrls/websiteUrls.setters.js.map +1 -0
  22. package/jest.config.js +20 -20
  23. package/package.json +46 -46
  24. package/src/__tests__/setup.ts +20 -20
  25. package/src/connection.ts +54 -54
  26. package/src/index.ts +27 -27
  27. package/src/municipal/__tests__/validation.spec.ts +62 -62
  28. package/src/municipal/cities/cities.getters.ts +50 -50
  29. package/src/municipal/cities/cities.types.ts +11 -11
  30. package/src/municipal/cities/index.ts +2 -2
  31. package/src/municipal/departmentsSubjects/departmentsSubjects.getters.ts +282 -282
  32. package/src/municipal/departmentsSubjects/departmentsSubjects.types.ts +72 -72
  33. package/src/municipal/departmentsSubjects/index.ts +9 -9
  34. package/src/municipal/index.ts +22 -22
  35. package/src/municipal/mongodb-client.ts +61 -61
  36. package/src/municipal/muniIssues/__tests__/muniIssues.setters.spec.ts +94 -94
  37. package/src/municipal/muniIssues/__tests__/muniIssues.setup.spec.ts +92 -92
  38. package/src/municipal/muniIssues/index.ts +17 -17
  39. package/src/municipal/muniIssues/muniIssues.constants.ts +18 -18
  40. package/src/municipal/muniIssues/muniIssues.getters.ts +24 -24
  41. package/src/municipal/muniIssues/muniIssues.schema.ts +66 -66
  42. package/src/municipal/muniIssues/muniIssues.setters.ts +24 -24
  43. package/src/municipal/muniIssues/muniIssues.setup.ts +65 -65
  44. package/src/municipal/muniIssues/muniIssues.types.ts +37 -37
  45. package/src/municipal/streets/index.ts +2 -2
  46. package/src/municipal/streets/streets.getters.ts +125 -125
  47. package/src/municipal/streets/streets.types.ts +18 -18
  48. package/src/municipal/systemInstructions/__tests__/getters.spec.ts +113 -113
  49. package/src/municipal/systemInstructions/__tests__/setters.spec.ts +274 -274
  50. package/src/municipal/systemInstructions/index.ts +7 -7
  51. package/src/municipal/systemInstructions/instructions.getters.ts +57 -57
  52. package/src/municipal/systemInstructions/instructions.setters.ts +119 -119
  53. package/src/municipal/systemInstructions/instructions.types.ts +30 -30
  54. package/src/municipal/tickets/__tests__/tickets.getters.spec.ts +30 -30
  55. package/src/municipal/tickets/__tests__/tickets.statistics.spec.ts +99 -99
  56. package/src/municipal/tickets/index.ts +9 -9
  57. package/src/municipal/tickets/tickets.constants.ts +8 -8
  58. package/src/municipal/tickets/tickets.getters.ts +121 -121
  59. package/src/municipal/tickets/tickets.statistics.aggregation.ts +110 -110
  60. package/src/municipal/tickets/tickets.statistics.getters.ts +145 -145
  61. package/src/municipal/tickets/tickets.types.ts +54 -54
  62. package/src/municipal/utils/types.ts +11 -11
  63. package/src/talkpilot/__tests__/db.spec.ts +38 -38
  64. package/src/talkpilot/__tests__/mongodb-client.spec.ts +18 -18
  65. package/src/talkpilot/__tests__/validation.spec.ts +68 -68
  66. package/src/talkpilot/agents/__tests__/agents.getters.spec.ts +29 -29
  67. package/src/talkpilot/agents/agents.getters.ts +34 -34
  68. package/src/talkpilot/agents/agents.types.ts +14 -14
  69. package/src/talkpilot/agents/index.ts +2 -2
  70. package/src/talkpilot/backgroundToolResults/__tests__/backgroundToolResults.getters.spec.ts +147 -147
  71. package/src/talkpilot/backgroundToolResults/backgroundToolResults.getters.ts +65 -65
  72. package/src/talkpilot/backgroundToolResults/backgroundToolResults.types.ts +23 -23
  73. package/src/talkpilot/backgroundToolResults/index.ts +2 -2
  74. package/src/talkpilot/calls/__tests__/callStats.utils.spec.ts +128 -128
  75. package/src/talkpilot/calls/__tests__/calls.dashboard.spec.ts +229 -229
  76. package/src/talkpilot/calls/__tests__/calls.spec.ts +283 -471
  77. package/src/talkpilot/calls/__tests__/calls.statistics.spec.ts +483 -483
  78. package/src/talkpilot/calls/calls.constants.ts +32 -35
  79. package/src/talkpilot/calls/calls.getters.ts +258 -277
  80. package/src/talkpilot/calls/calls.statistics.getters.ts +668 -668
  81. package/src/talkpilot/calls/calls.statistics.types.ts +48 -48
  82. package/src/talkpilot/calls/calls.types.ts +130 -141
  83. package/src/talkpilot/calls/dashboard/calls.dashboard.ts +295 -295
  84. package/src/talkpilot/calls/dashboard/calls.dashboard.types.ts +57 -57
  85. package/src/talkpilot/calls/index.ts +6 -6
  86. package/src/talkpilot/clientAudioBuffers/__tests__/clientAudioBuffer.getters.spec.ts +160 -160
  87. package/src/talkpilot/clientAudioBuffers/clientAudioBuffer.getters.ts +117 -117
  88. package/src/talkpilot/clientAudioBuffers/clientsAudioBuffers.types.ts +25 -25
  89. package/src/talkpilot/clientAudioBuffers/index.ts +2 -2
  90. package/src/talkpilot/clients/clients.getters.ts +16 -16
  91. package/src/talkpilot/clients/clients.types.ts +14 -14
  92. package/src/talkpilot/clients/index.ts +2 -2
  93. package/src/talkpilot/clientsConfig/__tests__/clientsConfig.getters.spec.ts +105 -105
  94. package/src/talkpilot/clientsConfig/__tests__/clientsConfig.spec.ts +186 -186
  95. package/src/talkpilot/clientsConfig/clientsConfig.getters.ts +79 -79
  96. package/src/talkpilot/clientsConfig/clientsConfig.types.ts +127 -127
  97. package/src/talkpilot/clientsConfig/index.ts +2 -2
  98. package/src/talkpilot/contextNotes/__tests__/contextNotes.getters.spec.ts +217 -217
  99. package/src/talkpilot/contextNotes/contextNotes.getters.ts +85 -85
  100. package/src/talkpilot/contextNotes/contextNotes.types.ts +20 -20
  101. package/src/talkpilot/contextNotes/index.ts +2 -2
  102. package/src/talkpilot/flows/__tests__/flows.schema.spec.ts +99 -99
  103. package/src/talkpilot/flows/flows.getter.ts +14 -14
  104. package/src/talkpilot/flows/flows.schema.ts +155 -155
  105. package/src/talkpilot/flows/flows.types.ts +189 -189
  106. package/src/talkpilot/flows/index.ts +2 -2
  107. package/src/talkpilot/groups/__tests__/groups.spec.ts +90 -90
  108. package/src/talkpilot/groups/__tests__/phone.utils.spec.ts +32 -32
  109. package/src/talkpilot/groups/groups.getters.ts +30 -30
  110. package/src/talkpilot/groups/groups.types.ts +29 -29
  111. package/src/talkpilot/groups/index.ts +3 -3
  112. package/src/talkpilot/groups/phone.utils.ts +46 -46
  113. package/src/talkpilot/index.ts +31 -31
  114. package/src/talkpilot/leads/index.ts +2 -2
  115. package/src/talkpilot/leads/leads.getter.ts +6 -6
  116. package/src/talkpilot/leads/leads.schema.ts +33 -33
  117. package/src/talkpilot/leads/leads.types.ts +20 -20
  118. package/src/talkpilot/mongodb-client.ts +78 -78
  119. package/src/talkpilot/phone_numbers/__tests__/phone_numbers.spec.ts +252 -252
  120. package/src/talkpilot/phone_numbers/index.ts +2 -2
  121. package/src/talkpilot/phone_numbers/phone_numbers.getter.ts +171 -171
  122. package/src/talkpilot/phone_numbers/phone_numbers.schema.ts +17 -17
  123. package/src/talkpilot/phone_numbers/phone_numbers.types.ts +30 -30
  124. package/src/talkpilot/plans/__tests__/plans.spec.ts +70 -70
  125. package/src/talkpilot/plans/index.ts +2 -2
  126. package/src/talkpilot/plans/plans.getters.ts +132 -132
  127. package/src/talkpilot/plans/plans.types.ts +89 -89
  128. package/src/talkpilot/products/__tests__/products.getters.spec.ts +44 -44
  129. package/src/talkpilot/products/index.ts +2 -2
  130. package/src/talkpilot/products/products.getters.ts +12 -12
  131. package/src/talkpilot/products/products.types.ts +9 -9
  132. package/src/talkpilot/results/index.ts +7 -7
  133. package/src/talkpilot/results/results.getter.ts +39 -39
  134. package/src/talkpilot/results/results.schema.ts +25 -25
  135. package/src/talkpilot/results/results.types.ts +34 -34
  136. package/src/talkpilot/retry_analyze/__tests__/retryAnalyze.getters.spec.ts +156 -156
  137. package/src/talkpilot/retry_analyze/index.ts +2 -2
  138. package/src/talkpilot/retry_analyze/retryAnalyze.getters.ts +84 -84
  139. package/src/talkpilot/retry_analyze/retryAnalyze.types.ts +13 -13
  140. package/src/talkpilot/sessions/__tests__/sessions.spec.ts +147 -147
  141. package/src/talkpilot/sessions/index.ts +2 -2
  142. package/src/talkpilot/sessions/sessions.getter.ts +92 -92
  143. package/src/talkpilot/sessions/sessions.schema.ts +34 -34
  144. package/src/talkpilot/sessions/sessions.types.ts +30 -30
  145. package/src/talkpilot/subscriptions/__tests__/subscriptions.getters.utils.spec.ts +45 -45
  146. package/src/talkpilot/subscriptions/index.ts +3 -3
  147. package/src/talkpilot/subscriptions/subscriptions.getters.ts +146 -146
  148. package/src/talkpilot/subscriptions/subscriptions.getters.utils.ts +33 -33
  149. package/src/talkpilot/subscriptions/subscriptions.types.ts +66 -66
  150. package/src/talkpilot/utils/__tests__/query.utils.spec.ts +49 -49
  151. package/src/talkpilot/utils/query.utils.ts +21 -21
  152. package/src/test-utils/db-utils.ts +26 -26
  153. package/src/test-utils/factories/index.ts +16 -16
  154. package/src/test-utils/factories/municipal/cities.ts +16 -16
  155. package/src/test-utils/factories/municipal/departmentsSubjects.ts +37 -37
  156. package/src/test-utils/factories/municipal/muniIssues.ts +32 -32
  157. package/src/test-utils/factories/municipal/streets.ts +22 -22
  158. package/src/test-utils/factories/municipal/tickets.ts +39 -39
  159. package/src/test-utils/factories/talkpilot/agents.ts +19 -19
  160. package/src/test-utils/factories/talkpilot/calls.ts +37 -37
  161. package/src/test-utils/factories/talkpilot/clientAudioBuffers.ts +20 -20
  162. package/src/test-utils/factories/talkpilot/clientsConfig.ts +18 -18
  163. package/src/test-utils/factories/talkpilot/contextNotes.ts +40 -40
  164. package/src/test-utils/factories/talkpilot/flows.ts +33 -33
  165. package/src/test-utils/factories/talkpilot/groups.ts +33 -33
  166. package/src/test-utils/factories/talkpilot/phone_numbers.ts +22 -22
  167. package/src/test-utils/factories/talkpilot/sessions.ts +35 -35
  168. package/src/test-utils/factories/websitalk/scans.ts +24 -24
  169. package/src/test-utils/factories/websitalk/websiteUrls.ts +17 -17
  170. package/src/utils/date.utils.ts +126 -126
  171. package/src/utils/shared.types.ts +4 -4
  172. package/src/utils/validation.ts +23 -23
  173. package/src/websitalk/index.ts +16 -16
  174. package/src/websitalk/mongodb-client.ts +61 -61
  175. package/src/websitalk/scans/__tests__/scans.spec.ts +313 -313
  176. package/src/websitalk/scans/index.ts +4 -4
  177. package/src/websitalk/scans/scans.constants.ts +15 -15
  178. package/src/websitalk/scans/scans.getters.ts +160 -160
  179. package/src/websitalk/scans/scans.types.ts +45 -45
  180. package/src/websitalk/scans/scans.utils.ts +5 -5
  181. package/src/websitalk/websiteUrls/__tests__/websiteUrls.spec.ts +77 -77
  182. package/src/websitalk/websiteUrls/index.ts +2 -2
  183. package/src/websitalk/websiteUrls/websiteUrls.getters.ts +65 -65
  184. package/src/websitalk/websiteUrls/websiteUrls.types.ts +13 -13
  185. package/tsconfig.json +23 -23
  186. package/.claude/settings.local.json +0 -13
  187. package/dist/talkpilot/leads/leads.setter.d.ts +0 -4
  188. package/dist/talkpilot/leads/leads.setter.d.ts.map +0 -1
  189. package/dist/talkpilot/leads/leads.setter.js +0 -16
  190. package/dist/talkpilot/leads/leads.setter.js.map +0 -1
@@ -1,313 +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
- } 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
- });
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
+ });
@@ -1,4 +1,4 @@
1
- export * from "./scans.constants";
2
- export * from "./scans.getters";
3
- export * from "./scans.types";
4
- export * from "./scans.utils";
1
+ export * from "./scans.constants";
2
+ export * from "./scans.getters";
3
+ export * from "./scans.types";
4
+ export * from "./scans.utils";
@@ -1,15 +1,15 @@
1
- export const SCAN_STATUSES = {
2
- CHECKING: "CHECKING",
3
- PREPARING: "PREPARING",
4
- SCANNING: "SCANNING",
5
- PROCESSING: "PROCESSING",
6
- COMPLETED: "COMPLETED",
7
- FAILED: "FAILED",
8
- OVERWRITTEN: "OVERWRITTEN",
9
- } as const;
10
-
11
- export const FINISHED_SCAN_STATUSES = [
12
- SCAN_STATUSES.COMPLETED,
13
- SCAN_STATUSES.FAILED,
14
- SCAN_STATUSES.OVERWRITTEN,
15
- ] as const;
1
+ export const SCAN_STATUSES = {
2
+ CHECKING: "CHECKING",
3
+ PREPARING: "PREPARING",
4
+ SCANNING: "SCANNING",
5
+ PROCESSING: "PROCESSING",
6
+ COMPLETED: "COMPLETED",
7
+ FAILED: "FAILED",
8
+ OVERWRITTEN: "OVERWRITTEN",
9
+ } as const;
10
+
11
+ export const FINISHED_SCAN_STATUSES = [
12
+ SCAN_STATUSES.COMPLETED,
13
+ SCAN_STATUSES.FAILED,
14
+ SCAN_STATUSES.OVERWRITTEN,
15
+ ] as const;