konnosDashboard
supamonitor.utils.test.ts308 lines · main
1import { safeSql } from '@supabase/pg-meta'
2import { describe, expect, it } from 'vitest'
3
4import type { ParsedLogEntry } from '../QueryInsights.types'
5import {
6 aggregateLogsByQuery,
7 parseSupamonitorLogs,
8 transformLogsToChartData,
9} from './supamonitor.utils'
10
11const makeSampleLog = (overrides: Partial<ParsedLogEntry> = {}): any => ({
12 timestamp: '2025-01-01T00:00:00Z',
13 application_name: 'test_app',
14 calls: 10,
15 database_name: 'test_db',
16 query: 'SELECT 1',
17 query_id: 1,
18 total_exec_time: 100,
19 total_plan_time: 20,
20 user_name: 'postgres',
21 mean_exec_time: 10,
22 mean_plan_time: 2,
23 min_exec_time: 1,
24 max_exec_time: 50,
25 min_plan_time: 0.5,
26 max_plan_time: 5,
27 p50_exec_time: 8,
28 p95_exec_time: 40,
29 p50_plan_time: 1.5,
30 p95_plan_time: 4,
31 ...overrides,
32})
33
34describe('parseSupamonitorLogs', () => {
35 it('returns empty array for null or empty input', () => {
36 expect(parseSupamonitorLogs(null as any)).toEqual([])
37 expect(parseSupamonitorLogs([])).toEqual([])
38 })
39
40 it('parses log entries preserving all fields', () => {
41 const raw = [makeSampleLog()]
42 const result = parseSupamonitorLogs(raw)
43
44 expect(result).toHaveLength(1)
45 expect(result[0]).toEqual({
46 timestamp: '2025-01-01T00:00:00Z',
47 application_name: 'test_app',
48 calls: 10,
49 database_name: 'test_db',
50 query: 'SELECT 1',
51 query_id: 1,
52 total_exec_time: 100,
53 total_plan_time: 20,
54 user_name: 'postgres',
55 mean_exec_time: 10,
56 mean_plan_time: 2,
57 min_exec_time: 1,
58 max_exec_time: 50,
59 min_plan_time: 0.5,
60 max_plan_time: 5,
61 p50_exec_time: 8,
62 p95_exec_time: 40,
63 p50_plan_time: 1.5,
64 p95_plan_time: 4,
65 })
66 })
67
68 it('handles multiple log entries', () => {
69 const raw = [makeSampleLog(), makeSampleLog({ query: safeSql`SELECT 2`, query_id: 2 })]
70 const result = parseSupamonitorLogs(raw)
71 expect(result).toHaveLength(2)
72 })
73})
74
75describe('transformLogsToChartData', () => {
76 it('returns empty array for null or empty input', () => {
77 expect(transformLogsToChartData(null as any)).toEqual([])
78 expect(transformLogsToChartData([])).toEqual([])
79 })
80
81 it('filters out entries with no timestamp', () => {
82 const logs: ParsedLogEntry[] = [{ query: safeSql`SELECT 1`, calls: 5 }]
83 const result = transformLogsToChartData(logs)
84 expect(result).toEqual([])
85 })
86
87 it('filters out entries with invalid timestamp', () => {
88 const logs: ParsedLogEntry[] = [{ timestamp: 'not-a-date', calls: 5 }]
89 const result = transformLogsToChartData(logs)
90 expect(result).toEqual([])
91 })
92
93 it('transforms a valid log entry into a chart data point', () => {
94 const logs: ParsedLogEntry[] = [
95 {
96 timestamp: '2025-01-01T00:00:00Z',
97 mean_exec_time: 10,
98 mean_plan_time: 2,
99 min_exec_time: 1,
100 max_exec_time: 50,
101 min_plan_time: 0.5,
102 max_plan_time: 5,
103 p50_exec_time: 8,
104 p95_exec_time: 40,
105 p50_plan_time: 1.5,
106 p95_plan_time: 4,
107 calls: 10,
108 },
109 ]
110
111 const result = transformLogsToChartData(logs)
112
113 expect(result).toHaveLength(1)
114 expect(result[0]).toEqual({
115 period_start: new Date('2025-01-01T00:00:00Z').getTime(),
116 timestamp: '2025-01-01T00:00:00Z',
117 query_latency: 12, // 10 + 2
118 mean_time: 10,
119 min_time: 1.5, // 1 + 0.5
120 max_time: 55, // 50 + 5
121 stddev_time: 0,
122 p50_time: 9.5, // 8 + 1.5
123 p95_time: 44, // 40 + 4
124 rows_read: 0,
125 calls: 10,
126 cache_hits: 0,
127 cache_misses: 0,
128 })
129 })
130
131 it('defaults missing numeric fields to 0', () => {
132 const logs: ParsedLogEntry[] = [
133 {
134 timestamp: '2025-06-01T12:00:00Z',
135 },
136 ]
137
138 const result = transformLogsToChartData(logs)
139
140 expect(result).toHaveLength(1)
141 expect(result[0].query_latency).toBe(0)
142 expect(result[0].calls).toBe(0)
143 expect(result[0].min_time).toBe(0)
144 expect(result[0].max_time).toBe(0)
145 })
146
147 it('sorts results by period_start ascending', () => {
148 const logs: ParsedLogEntry[] = [
149 { timestamp: '2025-01-03T00:00:00Z', mean_exec_time: 1 },
150 { timestamp: '2025-01-01T00:00:00Z', mean_exec_time: 2 },
151 { timestamp: '2025-01-02T00:00:00Z', mean_exec_time: 3 },
152 ]
153
154 const result = transformLogsToChartData(logs)
155
156 expect(result).toHaveLength(3)
157 expect(result[0].timestamp).toBe('2025-01-01T00:00:00Z')
158 expect(result[1].timestamp).toBe('2025-01-02T00:00:00Z')
159 expect(result[2].timestamp).toBe('2025-01-03T00:00:00Z')
160 })
161})
162
163describe('aggregateLogsByQuery', () => {
164 it('returns empty array for null or empty input', () => {
165 expect(aggregateLogsByQuery(null as any)).toEqual([])
166 expect(aggregateLogsByQuery([])).toEqual([])
167 })
168
169 it('skips entries with empty or whitespace-only queries', () => {
170 const logs: ParsedLogEntry[] = [
171 { query: safeSql``, calls: 5 },
172 { query: safeSql` `, calls: 3 },
173 ]
174 const result = aggregateLogsByQuery(logs)
175 expect(result).toEqual([])
176 })
177
178 it('aggregates a single log entry correctly', () => {
179 const logs: ParsedLogEntry[] = [
180 {
181 query: safeSql`SELECT 1`,
182 user_name: 'postgres',
183 application_name: 'app',
184 calls: 10,
185 total_exec_time: 100,
186 total_plan_time: 20,
187 min_exec_time: 1,
188 max_exec_time: 50,
189 min_plan_time: 0.5,
190 max_plan_time: 5,
191 },
192 ]
193
194 const result = aggregateLogsByQuery(logs)
195
196 expect(result).toHaveLength(1)
197 expect(result[0].query).toBe('SELECT 1')
198 expect(result[0].rolname).toBe('postgres')
199 expect(result[0].application_name).toBe('app')
200 expect(result[0].calls).toBe(10)
201 expect(result[0].total_time).toBe(120)
202 expect(result[0].mean_time).toBe(12)
203 expect(result[0].min_time).toBe(1.5)
204 expect(result[0].max_time).toBe(55)
205 expect(result[0].prop_total_time).toBe(100)
206 })
207
208 it('aggregates multiple entries for the same query', () => {
209 const logs: ParsedLogEntry[] = [
210 {
211 query: safeSql`SELECT 1`,
212 user_name: 'postgres',
213 calls: 5,
214 total_exec_time: 50,
215 total_plan_time: 10,
216 min_exec_time: 2,
217 max_exec_time: 20,
218 min_plan_time: 1,
219 max_plan_time: 3,
220 },
221 {
222 query: safeSql`SELECT 1`,
223 user_name: 'postgres',
224 calls: 10,
225 total_exec_time: 100,
226 total_plan_time: 20,
227 min_exec_time: 1,
228 max_exec_time: 50,
229 min_plan_time: 0.5,
230 max_plan_time: 5,
231 },
232 ]
233
234 const result = aggregateLogsByQuery(logs)
235
236 expect(result).toHaveLength(1)
237 expect(result[0].calls).toBe(15) // 5 + 10
238 expect(result[0].total_time).toBe(180) // (50+10) + (100+20)
239 expect(result[0].mean_time).toBe(12) // 180 / 15
240 expect(result[0].min_time).toBe(1.5) // min(2+1, 1+0.5) = 1.5
241 expect(result[0].max_time).toBe(55) // max(20+3, 50+5) = 55
242 expect(result[0]._count).toBe(2) // 2 log entries
243 })
244
245 it('normalizes whitespace differences in queries', () => {
246 const logs: ParsedLogEntry[] = [
247 { query: safeSql`SELECT 1`, calls: 5, total_exec_time: 50, total_plan_time: 0 },
248 { query: safeSql`SELECT 1`, calls: 3, total_exec_time: 30, total_plan_time: 0 },
249 ]
250
251 const result = aggregateLogsByQuery(logs)
252
253 expect(result).toHaveLength(1)
254 expect(result[0].calls).toBe(8)
255 })
256
257 it('sorts results by total_time descending', () => {
258 const logs: ParsedLogEntry[] = [
259 { query: safeSql`SELECT 1`, calls: 1, total_exec_time: 10, total_plan_time: 0 },
260 { query: safeSql`SELECT 2`, calls: 1, total_exec_time: 100, total_plan_time: 0 },
261 { query: safeSql`SELECT 3`, calls: 1, total_exec_time: 50, total_plan_time: 0 },
262 ]
263
264 const result = aggregateLogsByQuery(logs)
265
266 expect(result).toHaveLength(3)
267 expect(result[0].query).toBe('SELECT 2')
268 expect(result[1].query).toBe('SELECT 3')
269 expect(result[2].query).toBe('SELECT 1')
270 })
271
272 it('calculates prop_total_time as percentage of total execution', () => {
273 const logs: ParsedLogEntry[] = [
274 { query: safeSql`SELECT 1`, calls: 1, total_exec_time: 75, total_plan_time: 0 },
275 { query: safeSql`SELECT 2`, calls: 1, total_exec_time: 25, total_plan_time: 0 },
276 ]
277
278 const result = aggregateLogsByQuery(logs)
279
280 expect(result[0].prop_total_time).toBe(75)
281 expect(result[1].prop_total_time).toBe(25)
282 })
283
284 it('handles zero calls gracefully (mean_time defaults to 0)', () => {
285 const logs: ParsedLogEntry[] = [
286 { query: safeSql`SELECT 1`, calls: 0, total_exec_time: 100, total_plan_time: 0 },
287 ]
288
289 const result = aggregateLogsByQuery(logs)
290
291 expect(result).toHaveLength(1)
292 expect(result[0].mean_time).toBe(0)
293 })
294
295 it('sets static fields correctly', () => {
296 const logs: ParsedLogEntry[] = [
297 { query: safeSql`SELECT 1`, calls: 1, total_exec_time: 10, total_plan_time: 0 },
298 ]
299
300 const result = aggregateLogsByQuery(logs)
301
302 expect(result[0].rows_read).toBe(0)
303 expect(result[0].cache_hit_rate).toBe(0)
304 expect(result[0].index_advisor_result).toBeNull()
305 expect(result[0]._total_cache_hits).toBe(0)
306 expect(result[0]._total_cache_misses).toBe(0)
307 })
308})