@talkpilot/core-db 1.3.23 → 1.3.24

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