@talkpilot/core-db 1.3.19 → 1.3.21

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