observability-toolkit 1.8.2 → 1.8.5

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 (259) hide show
  1. package/README.md +60 -0
  2. package/dist/backends/index.d.ts +43 -0
  3. package/dist/backends/index.d.ts.map +1 -1
  4. package/dist/backends/index.js +41 -0
  5. package/dist/backends/index.js.map +1 -1
  6. package/dist/backends/index.test.d.ts +5 -0
  7. package/dist/backends/index.test.d.ts.map +1 -0
  8. package/dist/backends/index.test.js +156 -0
  9. package/dist/backends/index.test.js.map +1 -0
  10. package/dist/backends/local-jsonl-boolean-search.test.js +15 -12
  11. package/dist/backends/local-jsonl-boolean-search.test.js.map +1 -1
  12. package/dist/backends/local-jsonl-cache.test.d.ts +2 -0
  13. package/dist/backends/local-jsonl-cache.test.d.ts.map +1 -0
  14. package/dist/backends/local-jsonl-cache.test.js +295 -0
  15. package/dist/backends/local-jsonl-cache.test.js.map +1 -0
  16. package/dist/backends/local-jsonl-circuit-breaker.test.d.ts +2 -0
  17. package/dist/backends/local-jsonl-circuit-breaker.test.d.ts.map +1 -0
  18. package/dist/backends/local-jsonl-circuit-breaker.test.js +180 -0
  19. package/dist/backends/local-jsonl-circuit-breaker.test.js.map +1 -0
  20. package/dist/backends/local-jsonl-export.test.d.ts +2 -0
  21. package/dist/backends/local-jsonl-export.test.d.ts.map +1 -0
  22. package/dist/backends/local-jsonl-export.test.js +704 -0
  23. package/dist/backends/local-jsonl-export.test.js.map +1 -0
  24. package/dist/backends/local-jsonl-index.test.d.ts +2 -0
  25. package/dist/backends/local-jsonl-index.test.d.ts.map +1 -0
  26. package/dist/backends/local-jsonl-index.test.js +554 -0
  27. package/dist/backends/local-jsonl-index.test.js.map +1 -0
  28. package/dist/backends/local-jsonl-logs.test.d.ts +2 -0
  29. package/dist/backends/local-jsonl-logs.test.d.ts.map +1 -0
  30. package/dist/backends/local-jsonl-logs.test.js +612 -0
  31. package/dist/backends/local-jsonl-logs.test.js.map +1 -0
  32. package/dist/backends/local-jsonl-metrics.test.d.ts +2 -0
  33. package/dist/backends/local-jsonl-metrics.test.d.ts.map +1 -0
  34. package/dist/backends/local-jsonl-metrics.test.js +876 -0
  35. package/dist/backends/local-jsonl-metrics.test.js.map +1 -0
  36. package/dist/backends/local-jsonl-traces.test.d.ts +2 -0
  37. package/dist/backends/local-jsonl-traces.test.d.ts.map +1 -0
  38. package/dist/backends/local-jsonl-traces.test.js +1729 -0
  39. package/dist/backends/local-jsonl-traces.test.js.map +1 -0
  40. package/dist/backends/local-jsonl.d.ts +9 -0
  41. package/dist/backends/local-jsonl.d.ts.map +1 -1
  42. package/dist/backends/local-jsonl.js +348 -227
  43. package/dist/backends/local-jsonl.js.map +1 -1
  44. package/dist/backends/local-jsonl.test.js +290 -21
  45. package/dist/backends/local-jsonl.test.js.map +1 -1
  46. package/dist/backends/signoz-api-circuit-breaker.test.d.ts +6 -0
  47. package/dist/backends/signoz-api-circuit-breaker.test.d.ts.map +1 -0
  48. package/dist/backends/signoz-api-circuit-breaker.test.js +548 -0
  49. package/dist/backends/signoz-api-circuit-breaker.test.js.map +1 -0
  50. package/dist/backends/signoz-api-rate-limiter.test.d.ts +6 -0
  51. package/dist/backends/signoz-api-rate-limiter.test.d.ts.map +1 -0
  52. package/dist/backends/signoz-api-rate-limiter.test.js +389 -0
  53. package/dist/backends/signoz-api-rate-limiter.test.js.map +1 -0
  54. package/dist/backends/signoz-api-ssrf.test.d.ts +6 -0
  55. package/dist/backends/signoz-api-ssrf.test.d.ts.map +1 -0
  56. package/dist/backends/signoz-api-ssrf.test.js +216 -0
  57. package/dist/backends/signoz-api-ssrf.test.js.map +1 -0
  58. package/dist/backends/signoz-api-test-helpers.d.ts +80 -0
  59. package/dist/backends/signoz-api-test-helpers.d.ts.map +1 -0
  60. package/dist/backends/signoz-api-test-helpers.js +79 -0
  61. package/dist/backends/signoz-api-test-helpers.js.map +1 -0
  62. package/dist/backends/signoz-api.d.ts +16 -0
  63. package/dist/backends/signoz-api.d.ts.map +1 -1
  64. package/dist/backends/signoz-api.js +71 -9
  65. package/dist/backends/signoz-api.js.map +1 -1
  66. package/dist/backends/signoz-api.test.d.ts +9 -0
  67. package/dist/backends/signoz-api.test.d.ts.map +1 -1
  68. package/dist/backends/signoz-api.test.js +14 -1027
  69. package/dist/backends/signoz-api.test.js.map +1 -1
  70. package/dist/lib/cache.d.ts +47 -1
  71. package/dist/lib/cache.d.ts.map +1 -1
  72. package/dist/lib/cache.js +40 -3
  73. package/dist/lib/cache.js.map +1 -1
  74. package/dist/lib/circuit-breaker.d.ts +83 -0
  75. package/dist/lib/circuit-breaker.d.ts.map +1 -0
  76. package/dist/lib/circuit-breaker.js +125 -0
  77. package/dist/lib/circuit-breaker.js.map +1 -0
  78. package/dist/lib/circuit-breaker.test.d.ts +2 -0
  79. package/dist/lib/circuit-breaker.test.d.ts.map +1 -0
  80. package/dist/lib/circuit-breaker.test.js +263 -0
  81. package/dist/lib/circuit-breaker.test.js.map +1 -0
  82. package/dist/lib/constants-symlink.test.d.ts +12 -0
  83. package/dist/lib/constants-symlink.test.d.ts.map +1 -0
  84. package/dist/lib/constants-symlink.test.js +357 -0
  85. package/dist/lib/constants-symlink.test.js.map +1 -0
  86. package/dist/lib/constants.d.ts +43 -0
  87. package/dist/lib/constants.d.ts.map +1 -1
  88. package/dist/lib/constants.js +154 -24
  89. package/dist/lib/constants.js.map +1 -1
  90. package/dist/lib/constants.test.js +156 -7
  91. package/dist/lib/constants.test.js.map +1 -1
  92. package/dist/lib/edge-cases.test.d.ts +11 -0
  93. package/dist/lib/edge-cases.test.d.ts.map +1 -0
  94. package/dist/lib/edge-cases.test.js +634 -0
  95. package/dist/lib/edge-cases.test.js.map +1 -0
  96. package/dist/lib/error-sanitizer.d.ts.map +1 -1
  97. package/dist/lib/error-sanitizer.js +62 -26
  98. package/dist/lib/error-sanitizer.js.map +1 -1
  99. package/dist/lib/error-sanitizer.test.js +186 -0
  100. package/dist/lib/error-sanitizer.test.js.map +1 -1
  101. package/dist/lib/error-types.d.ts +54 -0
  102. package/dist/lib/error-types.d.ts.map +1 -0
  103. package/dist/lib/error-types.js +154 -0
  104. package/dist/lib/error-types.js.map +1 -0
  105. package/dist/lib/error-types.test.d.ts +2 -0
  106. package/dist/lib/error-types.test.d.ts.map +1 -0
  107. package/dist/lib/error-types.test.js +196 -0
  108. package/dist/lib/error-types.test.js.map +1 -0
  109. package/dist/lib/file-utils.test.js +3 -3
  110. package/dist/lib/file-utils.test.js.map +1 -1
  111. package/dist/lib/indexer.test.js +157 -24
  112. package/dist/lib/indexer.test.js.map +1 -1
  113. package/dist/lib/input-validator.d.ts +17 -0
  114. package/dist/lib/input-validator.d.ts.map +1 -1
  115. package/dist/lib/input-validator.fuzz.test.d.ts +12 -0
  116. package/dist/lib/input-validator.fuzz.test.d.ts.map +1 -0
  117. package/dist/lib/input-validator.fuzz.test.js +290 -0
  118. package/dist/lib/input-validator.fuzz.test.js.map +1 -0
  119. package/dist/lib/input-validator.js +62 -3
  120. package/dist/lib/input-validator.js.map +1 -1
  121. package/dist/lib/input-validator.test.js +129 -1
  122. package/dist/lib/input-validator.test.js.map +1 -1
  123. package/dist/lib/logger.d.ts +46 -0
  124. package/dist/lib/logger.d.ts.map +1 -0
  125. package/dist/lib/logger.js +81 -0
  126. package/dist/lib/logger.js.map +1 -0
  127. package/dist/lib/logger.test.d.ts +2 -0
  128. package/dist/lib/logger.test.d.ts.map +1 -0
  129. package/dist/lib/logger.test.js +122 -0
  130. package/dist/lib/logger.test.js.map +1 -0
  131. package/dist/lib/query-sanitizer.d.ts +51 -3
  132. package/dist/lib/query-sanitizer.d.ts.map +1 -1
  133. package/dist/lib/query-sanitizer.js +105 -31
  134. package/dist/lib/query-sanitizer.js.map +1 -1
  135. package/dist/lib/query-sanitizer.test.js +102 -1
  136. package/dist/lib/query-sanitizer.test.js.map +1 -1
  137. package/dist/lib/server-utils.d.ts +88 -0
  138. package/dist/lib/server-utils.d.ts.map +1 -0
  139. package/dist/lib/server-utils.js +173 -0
  140. package/dist/lib/server-utils.js.map +1 -0
  141. package/dist/lib/shared-schemas.d.ts +81 -0
  142. package/dist/lib/shared-schemas.d.ts.map +1 -0
  143. package/dist/lib/shared-schemas.js +80 -0
  144. package/dist/lib/shared-schemas.js.map +1 -0
  145. package/dist/lib/shared-schemas.test.d.ts +5 -0
  146. package/dist/lib/shared-schemas.test.d.ts.map +1 -0
  147. package/dist/lib/shared-schemas.test.js +106 -0
  148. package/dist/lib/shared-schemas.test.js.map +1 -0
  149. package/dist/lib/toon-encoder.d.ts +26 -0
  150. package/dist/lib/toon-encoder.d.ts.map +1 -0
  151. package/dist/lib/toon-encoder.js +61 -0
  152. package/dist/lib/toon-encoder.js.map +1 -0
  153. package/dist/lib/toon-encoder.test.d.ts +5 -0
  154. package/dist/lib/toon-encoder.test.d.ts.map +1 -0
  155. package/dist/lib/toon-encoder.test.js +85 -0
  156. package/dist/lib/toon-encoder.test.js.map +1 -0
  157. package/dist/server.d.ts +1 -49
  158. package/dist/server.d.ts.map +1 -1
  159. package/dist/server.js +154 -162
  160. package/dist/server.js.map +1 -1
  161. package/dist/server.test.js +198 -7
  162. package/dist/server.test.js.map +1 -1
  163. package/dist/test-helpers/env-utils.d.ts +87 -0
  164. package/dist/test-helpers/env-utils.d.ts.map +1 -0
  165. package/dist/test-helpers/env-utils.js +132 -0
  166. package/dist/test-helpers/env-utils.js.map +1 -0
  167. package/dist/test-helpers/file-utils.d.ts +67 -0
  168. package/dist/test-helpers/file-utils.d.ts.map +1 -1
  169. package/dist/test-helpers/file-utils.js +165 -2
  170. package/dist/test-helpers/file-utils.js.map +1 -1
  171. package/dist/test-helpers/fuzz-generators.d.ts +58 -0
  172. package/dist/test-helpers/fuzz-generators.d.ts.map +1 -0
  173. package/dist/test-helpers/fuzz-generators.js +216 -0
  174. package/dist/test-helpers/fuzz-generators.js.map +1 -0
  175. package/dist/test-helpers/index.d.ts +11 -0
  176. package/dist/test-helpers/index.d.ts.map +1 -0
  177. package/dist/test-helpers/index.js +30 -0
  178. package/dist/test-helpers/index.js.map +1 -0
  179. package/dist/test-helpers/memfs-utils.d.ts +181 -0
  180. package/dist/test-helpers/memfs-utils.d.ts.map +1 -0
  181. package/dist/test-helpers/memfs-utils.js +292 -0
  182. package/dist/test-helpers/memfs-utils.js.map +1 -0
  183. package/dist/test-helpers/memfs-utils.test.d.ts +5 -0
  184. package/dist/test-helpers/memfs-utils.test.d.ts.map +1 -0
  185. package/dist/test-helpers/memfs-utils.test.js +338 -0
  186. package/dist/test-helpers/memfs-utils.test.js.map +1 -0
  187. package/dist/test-helpers/mock-backends.d.ts +113 -2
  188. package/dist/test-helpers/mock-backends.d.ts.map +1 -1
  189. package/dist/test-helpers/mock-backends.js +199 -3
  190. package/dist/test-helpers/mock-backends.js.map +1 -1
  191. package/dist/test-helpers/mock-backends.test.d.ts +5 -0
  192. package/dist/test-helpers/mock-backends.test.d.ts.map +1 -0
  193. package/dist/test-helpers/mock-backends.test.js +368 -0
  194. package/dist/test-helpers/mock-backends.test.js.map +1 -0
  195. package/dist/test-helpers/race-condition-helpers.d.ts +85 -0
  196. package/dist/test-helpers/race-condition-helpers.d.ts.map +1 -0
  197. package/dist/test-helpers/race-condition-helpers.js +279 -0
  198. package/dist/test-helpers/race-condition-helpers.js.map +1 -0
  199. package/dist/test-helpers/schema-validators.d.ts +32 -0
  200. package/dist/test-helpers/schema-validators.d.ts.map +1 -0
  201. package/dist/test-helpers/schema-validators.js +125 -0
  202. package/dist/test-helpers/schema-validators.js.map +1 -0
  203. package/dist/test-helpers/test-data-builders.d.ts +260 -0
  204. package/dist/test-helpers/test-data-builders.d.ts.map +1 -0
  205. package/dist/test-helpers/test-data-builders.js +337 -0
  206. package/dist/test-helpers/test-data-builders.js.map +1 -0
  207. package/dist/test-helpers/test-data-builders.test.d.ts +2 -0
  208. package/dist/test-helpers/test-data-builders.test.d.ts.map +1 -0
  209. package/dist/test-helpers/test-data-builders.test.js +306 -0
  210. package/dist/test-helpers/test-data-builders.test.js.map +1 -0
  211. package/dist/test-helpers/tool-validators.d.ts +28 -0
  212. package/dist/test-helpers/tool-validators.d.ts.map +1 -0
  213. package/dist/test-helpers/tool-validators.js +71 -0
  214. package/dist/test-helpers/tool-validators.js.map +1 -0
  215. package/dist/tools/context-stats.d.ts +1 -0
  216. package/dist/tools/context-stats.d.ts.map +1 -1
  217. package/dist/tools/context-stats.js +9 -5
  218. package/dist/tools/context-stats.js.map +1 -1
  219. package/dist/tools/context-stats.test.js +24 -10
  220. package/dist/tools/context-stats.test.js.map +1 -1
  221. package/dist/tools/get-trace-url.js +2 -2
  222. package/dist/tools/get-trace-url.js.map +1 -1
  223. package/dist/tools/health-check.js +2 -2
  224. package/dist/tools/health-check.js.map +1 -1
  225. package/dist/tools/query-evaluations.d.ts +21 -18
  226. package/dist/tools/query-evaluations.d.ts.map +1 -1
  227. package/dist/tools/query-evaluations.js +33 -19
  228. package/dist/tools/query-evaluations.js.map +1 -1
  229. package/dist/tools/query-evaluations.test.js +60 -63
  230. package/dist/tools/query-evaluations.test.js.map +1 -1
  231. package/dist/tools/query-llm-events.d.ts +19 -15
  232. package/dist/tools/query-llm-events.d.ts.map +1 -1
  233. package/dist/tools/query-llm-events.js +31 -15
  234. package/dist/tools/query-llm-events.js.map +1 -1
  235. package/dist/tools/query-llm-events.test.js +277 -12
  236. package/dist/tools/query-llm-events.test.js.map +1 -1
  237. package/dist/tools/query-logs.d.ts +22 -22
  238. package/dist/tools/query-logs.d.ts.map +1 -1
  239. package/dist/tools/query-logs.js +9 -9
  240. package/dist/tools/query-logs.js.map +1 -1
  241. package/dist/tools/query-logs.test.js +19 -72
  242. package/dist/tools/query-logs.test.js.map +1 -1
  243. package/dist/tools/query-metrics.d.ts +14 -14
  244. package/dist/tools/query-metrics.d.ts.map +1 -1
  245. package/dist/tools/query-metrics.js +9 -9
  246. package/dist/tools/query-metrics.js.map +1 -1
  247. package/dist/tools/query-metrics.test.js +12 -25
  248. package/dist/tools/query-metrics.test.js.map +1 -1
  249. package/dist/tools/query-traces.d.ts +28 -28
  250. package/dist/tools/query-traces.d.ts.map +1 -1
  251. package/dist/tools/query-traces.js +18 -18
  252. package/dist/tools/query-traces.js.map +1 -1
  253. package/dist/tools/query-traces.test.js +58 -54
  254. package/dist/tools/query-traces.test.js.map +1 -1
  255. package/dist/tools/setup-claudeignore.js +7 -7
  256. package/dist/tools/setup-claudeignore.js.map +1 -1
  257. package/dist/tools/setup-claudeignore.test.js +4 -25
  258. package/dist/tools/setup-claudeignore.test.js.map +1 -1
  259. package/package.json +4 -2
@@ -0,0 +1,389 @@
1
+ /**
2
+ * Rate limiter tests for SigNoz API backend
3
+ * L-STYLE-2: Extracted from signoz-api.test.ts for maintainability
4
+ */
5
+ import { describe, it } from 'node:test';
6
+ import assert from 'node:assert';
7
+ import { SigNozApiBackend, TokenBucketRateLimiter } from './signoz-api.js';
8
+ import { setupMock, createV5TraceResponse } from './signoz-api-test-helpers.js';
9
+ describe('SigNozApiBackend rate limiter', () => {
10
+ it('should allow requests up to max tokens', async () => {
11
+ let callCount = 0;
12
+ globalThis.fetch = setupMock(async () => {
13
+ callCount++;
14
+ return {
15
+ ok: true,
16
+ json: async () => createV5TraceResponse([]),
17
+ text: async () => '',
18
+ };
19
+ });
20
+ // Create backend - default is 60 tokens
21
+ const backend = new SigNozApiBackend('https://signoz.example.com', 'test-key');
22
+ // Make multiple rapid requests - should all succeed within token limit
23
+ for (let i = 0; i < 10; i++) {
24
+ await backend.queryTraces({});
25
+ }
26
+ assert.strictEqual(callCount, 10, 'All requests within limit should succeed');
27
+ });
28
+ it('should block requests when rate limit exceeded', async () => {
29
+ const originalDateNow = Date.now;
30
+ let currentTime = 1000000;
31
+ Date.now = () => currentTime;
32
+ // Create limiter with very low rate limit (3 tokens) for testing
33
+ const limiter = new TokenBucketRateLimiter(3, 1); // 3 tokens, 1/sec refill
34
+ // Use all tokens
35
+ assert.strictEqual(limiter.tryConsume(), true, '1st token should be available');
36
+ assert.strictEqual(limiter.tryConsume(), true, '2nd token should be available');
37
+ assert.strictEqual(limiter.tryConsume(), true, '3rd token should be available');
38
+ assert.strictEqual(limiter.tryConsume(), false, '4th token should be blocked');
39
+ Date.now = originalDateNow;
40
+ });
41
+ it('should refill tokens over time', async () => {
42
+ const originalDateNow = Date.now;
43
+ let currentTime = 1000000;
44
+ Date.now = () => currentTime;
45
+ const limiter = new TokenBucketRateLimiter(3, 1); // 3 tokens, 1/sec refill
46
+ // Use all tokens
47
+ limiter.tryConsume();
48
+ limiter.tryConsume();
49
+ limiter.tryConsume();
50
+ assert.strictEqual(limiter.getAvailableTokens(), 0, 'All tokens should be used');
51
+ // Advance time by 2 seconds
52
+ currentTime += 2000;
53
+ // Should have refilled 2 tokens
54
+ assert.strictEqual(limiter.getAvailableTokens(), 2, 'Should have refilled 2 tokens');
55
+ // Advance time by 5 more seconds (past max)
56
+ currentTime += 5000;
57
+ // Should cap at max tokens
58
+ assert.strictEqual(limiter.getAvailableTokens(), 3, 'Should cap at max tokens');
59
+ Date.now = originalDateNow;
60
+ });
61
+ it('should return empty array when rate limited', async () => {
62
+ const originalDateNow = Date.now;
63
+ let currentTime = 1000000;
64
+ Date.now = () => currentTime;
65
+ let callCount = 0;
66
+ globalThis.fetch = setupMock(async () => {
67
+ callCount++;
68
+ return {
69
+ ok: true,
70
+ json: async () => createV5TraceResponse([]),
71
+ text: async () => '',
72
+ };
73
+ });
74
+ // We can't easily set custom rate limits on the backend, so we'll test
75
+ // the behavior by making 60+ requests (default limit)
76
+ const backend = new SigNozApiBackend('https://signoz.example.com', 'test-key');
77
+ // Make 60 requests (default token limit)
78
+ for (let i = 0; i < 60; i++) {
79
+ await backend.queryTraces({});
80
+ }
81
+ assert.strictEqual(callCount, 60);
82
+ // 61st request should be rate limited
83
+ const result = await backend.queryTraces({});
84
+ assert.deepStrictEqual(result, [], 'Should return empty array when rate limited');
85
+ assert.strictEqual(callCount, 60, 'Should not have made fetch call when rate limited');
86
+ Date.now = originalDateNow;
87
+ });
88
+ it('should log warning when rate limit exceeded', async () => {
89
+ const originalDateNow = Date.now;
90
+ let currentTime = 1000000;
91
+ Date.now = () => currentTime;
92
+ const warnLogs = [];
93
+ const originalWarn = console.warn;
94
+ console.warn = (msg) => { warnLogs.push(msg); };
95
+ const limiter = new TokenBucketRateLimiter(2, 1);
96
+ // Use all tokens
97
+ limiter.tryConsume();
98
+ limiter.tryConsume();
99
+ // This should trigger the warning
100
+ limiter.tryConsume();
101
+ console.warn = originalWarn;
102
+ Date.now = originalDateNow;
103
+ assert(warnLogs.some(log => log.includes('[obs-toolkit] Rate limit exceeded')));
104
+ });
105
+ it('should report rate limit status in health check', async () => {
106
+ const originalDateNow = Date.now;
107
+ let currentTime = 1000000;
108
+ Date.now = () => currentTime;
109
+ globalThis.fetch = setupMock(async () => {
110
+ return {
111
+ ok: true,
112
+ json: async () => ({ status: 'success' }),
113
+ text: async () => '',
114
+ };
115
+ });
116
+ const backend = new SigNozApiBackend('https://signoz.example.com', 'test-key');
117
+ // Use all 60 tokens
118
+ for (let i = 0; i < 60; i++) {
119
+ await backend.queryTraces({});
120
+ }
121
+ // Health check should report rate limit status
122
+ const health = await backend.healthCheck();
123
+ assert.strictEqual(health.status, 'error');
124
+ assert(health.message?.includes('Rate limit'));
125
+ Date.now = originalDateNow;
126
+ });
127
+ it('should reset tokens with reset method', async () => {
128
+ const limiter = new TokenBucketRateLimiter(3, 1);
129
+ // Use all tokens
130
+ limiter.tryConsume();
131
+ limiter.tryConsume();
132
+ limiter.tryConsume();
133
+ assert.strictEqual(limiter.getAvailableTokens(), 0);
134
+ // Reset
135
+ limiter.reset();
136
+ assert.strictEqual(limiter.getAvailableTokens(), 3, 'Should have all tokens after reset');
137
+ });
138
+ it('should handle logs query when rate limited', async () => {
139
+ const originalDateNow = Date.now;
140
+ let currentTime = 1000000;
141
+ Date.now = () => currentTime;
142
+ let callCount = 0;
143
+ globalThis.fetch = setupMock(async () => {
144
+ callCount++;
145
+ return {
146
+ ok: true,
147
+ json: async () => ({ data: { data: { results: [] } } }),
148
+ text: async () => '',
149
+ };
150
+ });
151
+ const backend = new SigNozApiBackend('https://signoz.example.com', 'test-key');
152
+ // Use all 60 tokens
153
+ for (let i = 0; i < 60; i++) {
154
+ await backend.queryTraces({});
155
+ }
156
+ assert.strictEqual(callCount, 60);
157
+ // Logs query should also be rate limited
158
+ const result = await backend.queryLogs({});
159
+ assert.deepStrictEqual(result, []);
160
+ assert.strictEqual(callCount, 60, 'Should not make fetch call for logs when rate limited');
161
+ Date.now = originalDateNow;
162
+ });
163
+ it('should handle metrics query when rate limited', async () => {
164
+ const originalDateNow = Date.now;
165
+ let currentTime = 1000000;
166
+ Date.now = () => currentTime;
167
+ let callCount = 0;
168
+ globalThis.fetch = setupMock(async () => {
169
+ callCount++;
170
+ return {
171
+ ok: true,
172
+ json: async () => ({ data: { data: { results: [] } } }),
173
+ text: async () => '',
174
+ };
175
+ });
176
+ const backend = new SigNozApiBackend('https://signoz.example.com', 'test-key');
177
+ // Use all 60 tokens
178
+ for (let i = 0; i < 60; i++) {
179
+ await backend.queryTraces({});
180
+ }
181
+ assert.strictEqual(callCount, 60);
182
+ // Metrics query should also be rate limited
183
+ const result = await backend.queryMetrics({ metricName: 'test' });
184
+ assert.deepStrictEqual(result, []);
185
+ assert.strictEqual(callCount, 60, 'Should not make fetch call for metrics when rate limited');
186
+ Date.now = originalDateNow;
187
+ });
188
+ it('should refund single token correctly', async () => {
189
+ const originalDateNow = Date.now;
190
+ let currentTime = 1000000;
191
+ Date.now = () => currentTime;
192
+ const limiter = new TokenBucketRateLimiter(3, 1); // 3 tokens, 1/sec refill
193
+ // Use all tokens
194
+ limiter.tryConsume();
195
+ limiter.tryConsume();
196
+ limiter.tryConsume();
197
+ assert.strictEqual(limiter.getAvailableTokens(), 0, 'All tokens should be used');
198
+ // Refund one token
199
+ limiter.refund();
200
+ assert.strictEqual(limiter.getAvailableTokens(), 1, 'Should have 1 token after refund');
201
+ // Refund again
202
+ limiter.refund();
203
+ assert.strictEqual(limiter.getAvailableTokens(), 2, 'Should have 2 tokens after second refund');
204
+ // Refund at max should not exceed max
205
+ limiter.refund();
206
+ limiter.refund(); // This should cap at max
207
+ assert.strictEqual(limiter.getAvailableTokens(), 3, 'Should cap at max tokens');
208
+ Date.now = originalDateNow;
209
+ });
210
+ it('should refund token when circuit breaker rejects request', async () => {
211
+ const originalDateNow = Date.now;
212
+ let currentTime = 1000000;
213
+ Date.now = () => currentTime;
214
+ let callCount = 0;
215
+ globalThis.fetch = setupMock(async () => {
216
+ callCount++;
217
+ // Simulate failures to open circuit breaker
218
+ const response = {
219
+ ok: false,
220
+ status: 500,
221
+ json: async () => ({}),
222
+ text: async () => 'Internal Server Error',
223
+ };
224
+ return response;
225
+ });
226
+ const backend = new SigNozApiBackend('https://signoz.example.com', 'test-key');
227
+ // Cause circuit breaker to open (default is 3 failures per constants.ts)
228
+ for (let i = 0; i < 3; i++) {
229
+ try {
230
+ await backend.queryTraces({});
231
+ }
232
+ catch {
233
+ // Expected to throw
234
+ }
235
+ }
236
+ assert.strictEqual(callCount, 3, 'Should have made 3 failing requests');
237
+ // Get initial token count - 60 tokens - 3 consumed = 57
238
+ // Now circuit breaker is open, subsequent requests should refund tokens
239
+ const result1 = await backend.queryTraces({});
240
+ assert.deepStrictEqual(result1, [], 'Should return empty when circuit breaker open');
241
+ // This should not consume more tokens because the token was refunded
242
+ const result2 = await backend.queryTraces({});
243
+ assert.deepStrictEqual(result2, [], 'Should return empty when circuit breaker open');
244
+ // No additional fetch calls should have been made
245
+ assert.strictEqual(callCount, 3, 'Should not make more fetch calls when circuit breaker open');
246
+ Date.now = originalDateNow;
247
+ });
248
+ // A2 Category 1: Rate Limiter Boundary Tests
249
+ describe('rate limiter boundary tests (A2)', () => {
250
+ it('should allow exactly MAX_TOKENS requests', async () => {
251
+ const MAX_TOKENS = 60;
252
+ const limiter = new TokenBucketRateLimiter(MAX_TOKENS, 1);
253
+ // Consume exactly MAX_TOKENS
254
+ for (let i = 0; i < MAX_TOKENS; i++) {
255
+ const allowed = limiter.tryConsume();
256
+ assert.strictEqual(allowed, true, `Request ${i + 1} should be allowed`);
257
+ }
258
+ assert.strictEqual(limiter.getAvailableTokens(), 0, '60th request should leave 0 tokens');
259
+ });
260
+ it('should block the 61st request', async () => {
261
+ const MAX_TOKENS = 60;
262
+ const limiter = new TokenBucketRateLimiter(MAX_TOKENS, 1);
263
+ // Consume all tokens
264
+ for (let i = 0; i < MAX_TOKENS; i++) {
265
+ limiter.tryConsume();
266
+ }
267
+ // 61st request should fail
268
+ const allowed = limiter.tryConsume();
269
+ assert.strictEqual(allowed, false, '61st request should be blocked');
270
+ });
271
+ it('should refill tokens after elapsed time', async () => {
272
+ const originalDateNow = Date.now;
273
+ let currentTime = 1000000;
274
+ Date.now = () => currentTime;
275
+ try {
276
+ const limiter = new TokenBucketRateLimiter(60, 1); // 1 token per second
277
+ // Use all tokens
278
+ for (let i = 0; i < 60; i++) {
279
+ limiter.tryConsume();
280
+ }
281
+ assert.strictEqual(limiter.getAvailableTokens(), 0);
282
+ // Advance time by 1 second - should get 1 token back
283
+ currentTime += 1000;
284
+ assert.strictEqual(limiter.getAvailableTokens(), 1, 'Should have 1 token after 1 second');
285
+ // Advance time by 5 more seconds - should have 6 tokens
286
+ currentTime += 5000;
287
+ assert.strictEqual(limiter.getAvailableTokens(), 6, 'Should have 6 tokens after 6 seconds');
288
+ }
289
+ finally {
290
+ Date.now = originalDateNow;
291
+ }
292
+ });
293
+ it('should not add partial tokens for fractional seconds', async () => {
294
+ const originalDateNow = Date.now;
295
+ let currentTime = 1000000;
296
+ Date.now = () => currentTime;
297
+ try {
298
+ const limiter = new TokenBucketRateLimiter(60, 1);
299
+ // Use all tokens
300
+ for (let i = 0; i < 60; i++) {
301
+ limiter.tryConsume();
302
+ }
303
+ // Advance by 500ms - should NOT add a partial token
304
+ currentTime += 500;
305
+ assert.strictEqual(limiter.getAvailableTokens(), 0, 'Should not have partial tokens at 500ms');
306
+ // Advance to 999ms - still no token
307
+ currentTime += 499;
308
+ assert.strictEqual(limiter.getAvailableTokens(), 0, 'Should not have token at 999ms');
309
+ // Advance to 1000ms - now should have 1 token
310
+ currentTime += 1;
311
+ assert.strictEqual(limiter.getAvailableTokens(), 1, 'Should have 1 token at 1000ms');
312
+ }
313
+ finally {
314
+ Date.now = originalDateNow;
315
+ }
316
+ });
317
+ it('should cap refilled tokens at MAX_TOKENS', async () => {
318
+ const originalDateNow = Date.now;
319
+ let currentTime = 1000000;
320
+ Date.now = () => currentTime;
321
+ try {
322
+ const limiter = new TokenBucketRateLimiter(60, 1);
323
+ // Use 30 tokens (leave 30)
324
+ for (let i = 0; i < 30; i++) {
325
+ limiter.tryConsume();
326
+ }
327
+ assert.strictEqual(limiter.getAvailableTokens(), 30);
328
+ // Advance by 60 seconds - should cap at 60, not 90
329
+ currentTime += 60000;
330
+ assert.strictEqual(limiter.getAvailableTokens(), 60, 'Should cap at MAX_TOKENS');
331
+ }
332
+ finally {
333
+ Date.now = originalDateNow;
334
+ }
335
+ });
336
+ it('should handle exact exhaustion correctly', async () => {
337
+ const limiter = new TokenBucketRateLimiter(3, 1);
338
+ limiter.tryConsume();
339
+ limiter.tryConsume();
340
+ assert.strictEqual(limiter.getAvailableTokens(), 1, 'Should have 1 token left');
341
+ limiter.tryConsume();
342
+ assert.strictEqual(limiter.getAvailableTokens(), 0, 'Should have exactly 0 tokens');
343
+ const allowed = limiter.tryConsume();
344
+ assert.strictEqual(allowed, false, 'Next request should fail');
345
+ });
346
+ it('should not overflow on large refill periods', async () => {
347
+ const originalDateNow = Date.now;
348
+ let currentTime = 1000000;
349
+ Date.now = () => currentTime;
350
+ try {
351
+ const limiter = new TokenBucketRateLimiter(60, 1);
352
+ // Use all tokens
353
+ for (let i = 0; i < 60; i++) {
354
+ limiter.tryConsume();
355
+ }
356
+ // Advance by a very long time (1 day)
357
+ currentTime += 86400000;
358
+ const tokens = limiter.getAvailableTokens();
359
+ assert.strictEqual(tokens, 60, 'Should cap at MAX_TOKENS, not overflow');
360
+ }
361
+ finally {
362
+ Date.now = originalDateNow;
363
+ }
364
+ });
365
+ it('should cap elapsed time to prevent precision loss (H1)', async () => {
366
+ const originalDateNow = Date.now;
367
+ let currentTime = 1000000;
368
+ Date.now = () => currentTime;
369
+ try {
370
+ // 10 tokens, 1/sec refill = max useful elapsed is 10 seconds
371
+ const limiter = new TokenBucketRateLimiter(10, 1);
372
+ // Use all tokens
373
+ for (let i = 0; i < 10; i++) {
374
+ limiter.tryConsume();
375
+ }
376
+ assert.strictEqual(limiter.getAvailableTokens(), 0);
377
+ // Advance by 30 days (extreme case for long-running process)
378
+ currentTime += 30 * 24 * 60 * 60 * 1000;
379
+ // Should still correctly cap at maxTokens, not have precision issues
380
+ const tokens = limiter.getAvailableTokens();
381
+ assert.strictEqual(tokens, 10, 'Should handle extreme elapsed times without precision loss');
382
+ }
383
+ finally {
384
+ Date.now = originalDateNow;
385
+ }
386
+ });
387
+ });
388
+ });
389
+ //# sourceMappingURL=signoz-api-rate-limiter.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"signoz-api-rate-limiter.test.js","sourceRoot":"","sources":["../../src/backends/signoz-api-rate-limiter.test.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,MAAM,MAAM,aAAa,CAAC;AACjC,OAAO,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,MAAM,iBAAiB,CAAC;AAC3E,OAAO,EAAE,SAAS,EAAE,qBAAqB,EAAE,MAAM,8BAA8B,CAAC;AAEhF,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;IAC7C,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,UAAU,CAAC,KAAK,GAAG,SAAS,CAAC,KAAK,IAAI,EAAE;YACtC,SAAS,EAAE,CAAC;YACZ,OAAO;gBACL,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,qBAAqB,CAAC,EAAE,CAAC;gBAC3C,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,EAAE;aACrB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,wCAAwC;QACxC,MAAM,OAAO,GAAG,IAAI,gBAAgB,CAAC,4BAA4B,EAAE,UAAU,CAAC,CAAC;QAE/E,uEAAuE;QACvE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAChC,CAAC;QACD,MAAM,CAAC,WAAW,CAAC,SAAS,EAAE,EAAE,EAAE,0CAA0C,CAAC,CAAC;IAChF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC;QACjC,IAAI,WAAW,GAAG,OAAO,CAAC;QAC1B,IAAI,CAAC,GAAG,GAAG,GAAG,EAAE,CAAC,WAAW,CAAC;QAE7B,iEAAiE;QACjE,MAAM,OAAO,GAAG,IAAI,sBAAsB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,yBAAyB;QAE3E,iBAAiB;QACjB,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,+BAA+B,CAAC,CAAC;QAChF,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,+BAA+B,CAAC,CAAC;QAChF,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,+BAA+B,CAAC,CAAC;QAChF,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,KAAK,EAAE,6BAA6B,CAAC,CAAC;QAE/E,IAAI,CAAC,GAAG,GAAG,eAAe,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC9C,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC;QACjC,IAAI,WAAW,GAAG,OAAO,CAAC;QAC1B,IAAI,CAAC,GAAG,GAAG,GAAG,EAAE,CAAC,WAAW,CAAC;QAE7B,MAAM,OAAO,GAAG,IAAI,sBAAsB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,yBAAyB;QAE3E,iBAAiB;QACjB,OAAO,CAAC,UAAU,EAAE,CAAC;QACrB,OAAO,CAAC,UAAU,EAAE,CAAC;QACrB,OAAO,CAAC,UAAU,EAAE,CAAC;QACrB,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,EAAE,2BAA2B,CAAC,CAAC;QAEjF,4BAA4B;QAC5B,WAAW,IAAI,IAAI,CAAC;QAEpB,gCAAgC;QAChC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,EAAE,+BAA+B,CAAC,CAAC;QAErF,4CAA4C;QAC5C,WAAW,IAAI,IAAI,CAAC;QAEpB,2BAA2B;QAC3B,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,EAAE,0BAA0B,CAAC,CAAC;QAEhF,IAAI,CAAC,GAAG,GAAG,eAAe,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC;QACjC,IAAI,WAAW,GAAG,OAAO,CAAC;QAC1B,IAAI,CAAC,GAAG,GAAG,GAAG,EAAE,CAAC,WAAW,CAAC;QAE7B,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,UAAU,CAAC,KAAK,GAAG,SAAS,CAAC,KAAK,IAAI,EAAE;YACtC,SAAS,EAAE,CAAC;YACZ,OAAO;gBACL,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,qBAAqB,CAAC,EAAE,CAAC;gBAC3C,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,EAAE;aACrB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,uEAAuE;QACvE,sDAAsD;QACtD,MAAM,OAAO,GAAG,IAAI,gBAAgB,CAAC,4BAA4B,EAAE,UAAU,CAAC,CAAC;QAE/E,yCAAyC;QACzC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAChC,CAAC;QACD,MAAM,CAAC,WAAW,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAElC,sCAAsC;QACtC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAC7C,MAAM,CAAC,eAAe,CAAC,MAAM,EAAE,EAAE,EAAE,6CAA6C,CAAC,CAAC;QAClF,MAAM,CAAC,WAAW,CAAC,SAAS,EAAE,EAAE,EAAE,mDAAmD,CAAC,CAAC;QAEvF,IAAI,CAAC,GAAG,GAAG,eAAe,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC;QACjC,IAAI,WAAW,GAAG,OAAO,CAAC;QAC1B,IAAI,CAAC,GAAG,GAAG,GAAG,EAAE,CAAC,WAAW,CAAC;QAE7B,MAAM,QAAQ,GAAa,EAAE,CAAC;QAC9B,MAAM,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC;QAClC,OAAO,CAAC,IAAI,GAAG,CAAC,GAAW,EAAE,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAExD,MAAM,OAAO,GAAG,IAAI,sBAAsB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAEjD,iBAAiB;QACjB,OAAO,CAAC,UAAU,EAAE,CAAC;QACrB,OAAO,CAAC,UAAU,EAAE,CAAC;QAErB,kCAAkC;QAClC,OAAO,CAAC,UAAU,EAAE,CAAC;QAErB,OAAO,CAAC,IAAI,GAAG,YAAY,CAAC;QAC5B,IAAI,CAAC,GAAG,GAAG,eAAe,CAAC;QAE3B,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,mCAAmC,CAAC,CAAC,CAAC,CAAC;IAClF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC;QACjC,IAAI,WAAW,GAAG,OAAO,CAAC;QAC1B,IAAI,CAAC,GAAG,GAAG,GAAG,EAAE,CAAC,WAAW,CAAC;QAE7B,UAAU,CAAC,KAAK,GAAG,SAAS,CAAC,KAAK,IAAI,EAAE;YACtC,OAAO;gBACL,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;gBACzC,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,EAAE;aACrB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,IAAI,gBAAgB,CAAC,4BAA4B,EAAE,UAAU,CAAC,CAAC;QAE/E,oBAAoB;QACpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAChC,CAAC;QAED,+CAA+C;QAC/C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,WAAW,EAAE,CAAC;QAC3C,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAC3C,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC;QAE/C,IAAI,CAAC,GAAG,GAAG,eAAe,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,OAAO,GAAG,IAAI,sBAAsB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAEjD,iBAAiB;QACjB,OAAO,CAAC,UAAU,EAAE,CAAC;QACrB,OAAO,CAAC,UAAU,EAAE,CAAC;QACrB,OAAO,CAAC,UAAU,EAAE,CAAC;QACrB,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAC,CAAC;QAEpD,QAAQ;QACR,OAAO,CAAC,KAAK,EAAE,CAAC;QAChB,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,EAAE,oCAAoC,CAAC,CAAC;IAC5F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC;QACjC,IAAI,WAAW,GAAG,OAAO,CAAC;QAC1B,IAAI,CAAC,GAAG,GAAG,GAAG,EAAE,CAAC,WAAW,CAAC;QAE7B,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,UAAU,CAAC,KAAK,GAAG,SAAS,CAAC,KAAK,IAAI,EAAE;YACtC,SAAS,EAAE,CAAC;YACZ,OAAO;gBACL,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;gBACvD,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,EAAE;aACrB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,IAAI,gBAAgB,CAAC,4BAA4B,EAAE,UAAU,CAAC,CAAC;QAE/E,oBAAoB;QACpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAChC,CAAC;QACD,MAAM,CAAC,WAAW,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAElC,yCAAyC;QACzC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QAC3C,MAAM,CAAC,eAAe,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QACnC,MAAM,CAAC,WAAW,CAAC,SAAS,EAAE,EAAE,EAAE,uDAAuD,CAAC,CAAC;QAE3F,IAAI,CAAC,GAAG,GAAG,eAAe,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC;QACjC,IAAI,WAAW,GAAG,OAAO,CAAC;QAC1B,IAAI,CAAC,GAAG,GAAG,GAAG,EAAE,CAAC,WAAW,CAAC;QAE7B,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,UAAU,CAAC,KAAK,GAAG,SAAS,CAAC,KAAK,IAAI,EAAE;YACtC,SAAS,EAAE,CAAC;YACZ,OAAO;gBACL,EAAE,EAAE,IAAI;gBACR,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;gBACvD,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,EAAE;aACrB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,IAAI,gBAAgB,CAAC,4BAA4B,EAAE,UAAU,CAAC,CAAC;QAE/E,oBAAoB;QACpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAChC,CAAC;QACD,MAAM,CAAC,WAAW,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAElC,4CAA4C;QAC5C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,YAAY,CAAC,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC,CAAC;QAClE,MAAM,CAAC,eAAe,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QACnC,MAAM,CAAC,WAAW,CAAC,SAAS,EAAE,EAAE,EAAE,0DAA0D,CAAC,CAAC;QAE9F,IAAI,CAAC,GAAG,GAAG,eAAe,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC;QACjC,IAAI,WAAW,GAAG,OAAO,CAAC;QAC1B,IAAI,CAAC,GAAG,GAAG,GAAG,EAAE,CAAC,WAAW,CAAC;QAE7B,MAAM,OAAO,GAAG,IAAI,sBAAsB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,yBAAyB;QAE3E,iBAAiB;QACjB,OAAO,CAAC,UAAU,EAAE,CAAC;QACrB,OAAO,CAAC,UAAU,EAAE,CAAC;QACrB,OAAO,CAAC,UAAU,EAAE,CAAC;QACrB,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,EAAE,2BAA2B,CAAC,CAAC;QAEjF,mBAAmB;QACnB,OAAO,CAAC,MAAM,EAAE,CAAC;QACjB,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,EAAE,kCAAkC,CAAC,CAAC;QAExF,eAAe;QACf,OAAO,CAAC,MAAM,EAAE,CAAC;QACjB,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,EAAE,0CAA0C,CAAC,CAAC;QAEhG,sCAAsC;QACtC,OAAO,CAAC,MAAM,EAAE,CAAC;QACjB,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,yBAAyB;QAC3C,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,EAAE,0BAA0B,CAAC,CAAC;QAEhF,IAAI,CAAC,GAAG,GAAG,eAAe,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC;QACjC,IAAI,WAAW,GAAG,OAAO,CAAC;QAC1B,IAAI,CAAC,GAAG,GAAG,GAAG,EAAE,CAAC,WAAW,CAAC;QAE7B,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,UAAU,CAAC,KAAK,GAAG,SAAS,CAAC,KAAK,IAAI,EAAE;YACtC,SAAS,EAAE,CAAC;YACZ,4CAA4C;YAC5C,MAAM,QAAQ,GAAG;gBACf,EAAE,EAAE,KAAK;gBACT,MAAM,EAAE,GAAG;gBACX,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;gBACtB,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,uBAAuB;aAC1C,CAAC;YACF,OAAO,QAAQ,CAAC;QAClB,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,IAAI,gBAAgB,CAAC,4BAA4B,EAAE,UAAU,CAAC,CAAC;QAE/E,yEAAyE;QACzE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3B,IAAI,CAAC;gBACH,MAAM,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;YAChC,CAAC;YAAC,MAAM,CAAC;gBACP,oBAAoB;YACtB,CAAC;QACH,CAAC;QACD,MAAM,CAAC,WAAW,CAAC,SAAS,EAAE,CAAC,EAAE,qCAAqC,CAAC,CAAC;QAExE,wDAAwD;QACxD,wEAAwE;QACxE,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAC9C,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,EAAE,EAAE,+CAA+C,CAAC,CAAC;QAErF,qEAAqE;QACrE,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAC9C,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,EAAE,EAAE,+CAA+C,CAAC,CAAC;QAErF,kDAAkD;QAClD,MAAM,CAAC,WAAW,CAAC,SAAS,EAAE,CAAC,EAAE,4DAA4D,CAAC,CAAC;QAE/F,IAAI,CAAC,GAAG,GAAG,eAAe,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,6CAA6C;IAC7C,QAAQ,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAChD,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,UAAU,GAAG,EAAE,CAAC;YACtB,MAAM,OAAO,GAAG,IAAI,sBAAsB,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;YAE1D,6BAA6B;YAC7B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpC,MAAM,OAAO,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC;gBACrC,MAAM,CAAC,WAAW,CAAC,OAAO,EAAE,IAAI,EAAE,WAAW,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;YAC1E,CAAC;YAED,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,EAAE,oCAAoC,CAAC,CAAC;QAC5F,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,MAAM,UAAU,GAAG,EAAE,CAAC;YACtB,MAAM,OAAO,GAAG,IAAI,sBAAsB,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;YAE1D,qBAAqB;YACrB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpC,OAAO,CAAC,UAAU,EAAE,CAAC;YACvB,CAAC;YAED,2BAA2B;YAC3B,MAAM,OAAO,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC;YACrC,MAAM,CAAC,WAAW,CAAC,OAAO,EAAE,KAAK,EAAE,gCAAgC,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC;YACjC,IAAI,WAAW,GAAG,OAAO,CAAC;YAC1B,IAAI,CAAC,GAAG,GAAG,GAAG,EAAE,CAAC,WAAW,CAAC;YAE7B,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,IAAI,sBAAsB,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,qBAAqB;gBAExE,iBAAiB;gBACjB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;oBAC5B,OAAO,CAAC,UAAU,EAAE,CAAC;gBACvB,CAAC;gBACD,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAC,CAAC;gBAEpD,qDAAqD;gBACrD,WAAW,IAAI,IAAI,CAAC;gBACpB,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,EAAE,oCAAoC,CAAC,CAAC;gBAE1F,wDAAwD;gBACxD,WAAW,IAAI,IAAI,CAAC;gBACpB,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,EAAE,sCAAsC,CAAC,CAAC;YAC9F,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,GAAG,GAAG,eAAe,CAAC;YAC7B,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC;YACjC,IAAI,WAAW,GAAG,OAAO,CAAC;YAC1B,IAAI,CAAC,GAAG,GAAG,GAAG,EAAE,CAAC,WAAW,CAAC;YAE7B,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,IAAI,sBAAsB,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;gBAElD,iBAAiB;gBACjB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;oBAC5B,OAAO,CAAC,UAAU,EAAE,CAAC;gBACvB,CAAC;gBAED,oDAAoD;gBACpD,WAAW,IAAI,GAAG,CAAC;gBACnB,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,EAAE,yCAAyC,CAAC,CAAC;gBAE/F,oCAAoC;gBACpC,WAAW,IAAI,GAAG,CAAC;gBACnB,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,EAAE,gCAAgC,CAAC,CAAC;gBAEtF,8CAA8C;gBAC9C,WAAW,IAAI,CAAC,CAAC;gBACjB,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,EAAE,+BAA+B,CAAC,CAAC;YACvF,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,GAAG,GAAG,eAAe,CAAC;YAC7B,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC;YACjC,IAAI,WAAW,GAAG,OAAO,CAAC;YAC1B,IAAI,CAAC,GAAG,GAAG,GAAG,EAAE,CAAC,WAAW,CAAC;YAE7B,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,IAAI,sBAAsB,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;gBAElD,2BAA2B;gBAC3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;oBAC5B,OAAO,CAAC,UAAU,EAAE,CAAC;gBACvB,CAAC;gBACD,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,EAAE,CAAC,CAAC;gBAErD,mDAAmD;gBACnD,WAAW,IAAI,KAAK,CAAC;gBACrB,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,EAAE,EAAE,0BAA0B,CAAC,CAAC;YACnF,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,GAAG,GAAG,eAAe,CAAC;YAC7B,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,OAAO,GAAG,IAAI,sBAAsB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAEjD,OAAO,CAAC,UAAU,EAAE,CAAC;YACrB,OAAO,CAAC,UAAU,EAAE,CAAC;YACrB,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,EAAE,0BAA0B,CAAC,CAAC;YAEhF,OAAO,CAAC,UAAU,EAAE,CAAC;YACrB,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,EAAE,8BAA8B,CAAC,CAAC;YAEpF,MAAM,OAAO,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC;YACrC,MAAM,CAAC,WAAW,CAAC,OAAO,EAAE,KAAK,EAAE,0BAA0B,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC;YACjC,IAAI,WAAW,GAAG,OAAO,CAAC;YAC1B,IAAI,CAAC,GAAG,GAAG,GAAG,EAAE,CAAC,WAAW,CAAC;YAE7B,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,IAAI,sBAAsB,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;gBAElD,iBAAiB;gBACjB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;oBAC5B,OAAO,CAAC,UAAU,EAAE,CAAC;gBACvB,CAAC;gBAED,sCAAsC;gBACtC,WAAW,IAAI,QAAQ,CAAC;gBACxB,MAAM,MAAM,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;gBAC5C,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,EAAE,wCAAwC,CAAC,CAAC;YAC3E,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,GAAG,GAAG,eAAe,CAAC;YAC7B,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC;YACjC,IAAI,WAAW,GAAG,OAAO,CAAC;YAC1B,IAAI,CAAC,GAAG,GAAG,GAAG,EAAE,CAAC,WAAW,CAAC;YAE7B,IAAI,CAAC;gBACH,6DAA6D;gBAC7D,MAAM,OAAO,GAAG,IAAI,sBAAsB,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;gBAElD,iBAAiB;gBACjB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;oBAC5B,OAAO,CAAC,UAAU,EAAE,CAAC;gBACvB,CAAC;gBACD,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAC,CAAC;gBAEpD,6DAA6D;gBAC7D,WAAW,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;gBAExC,qEAAqE;gBACrE,MAAM,MAAM,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;gBAC5C,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,EAAE,4DAA4D,CAAC,CAAC;YAC/F,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,GAAG,GAAG,eAAe,CAAC;YAC7B,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -0,0 +1,6 @@
1
+ /**
2
+ * SSRF protection tests for SigNoz API backend
3
+ * L-STYLE-2: Extracted from signoz-api.test.ts for maintainability
4
+ */
5
+ export {};
6
+ //# sourceMappingURL=signoz-api-ssrf.test.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"signoz-api-ssrf.test.d.ts","sourceRoot":"","sources":["../../src/backends/signoz-api-ssrf.test.ts"],"names":[],"mappings":"AAAA;;;GAGG"}
@@ -0,0 +1,216 @@
1
+ /**
2
+ * SSRF protection tests for SigNoz API backend
3
+ * L-STYLE-2: Extracted from signoz-api.test.ts for maintainability
4
+ */
5
+ import { describe, it } from 'node:test';
6
+ import assert from 'node:assert';
7
+ import { SigNozApiBackend } from './signoz-api.js';
8
+ describe('SigNozApiBackend SSRF protection', () => {
9
+ it('should block localhost URL', () => {
10
+ const backend = new SigNozApiBackend('https://localhost/api', 'test-key');
11
+ // Backend with blocked URL will have empty baseUrl
12
+ const url = backend.getTraceUrl('trace-123');
13
+ assert.strictEqual(url, '/trace/trace-123', 'Should have empty base for blocked localhost');
14
+ });
15
+ it('should block 127.0.0.1', () => {
16
+ const backend = new SigNozApiBackend('https://127.0.0.1/api', 'test-key');
17
+ const url = backend.getTraceUrl('trace-123');
18
+ assert.strictEqual(url, '/trace/trace-123', 'Should have empty base for blocked 127.0.0.1');
19
+ });
20
+ it('should block IPv6 localhost ::1', () => {
21
+ const backend = new SigNozApiBackend('https://[::1]/api', 'test-key');
22
+ const url = backend.getTraceUrl('trace-123');
23
+ assert.strictEqual(url, '/trace/trace-123', 'Should block IPv6 localhost');
24
+ });
25
+ it('should block long-form IPv6 localhost', () => {
26
+ const backend = new SigNozApiBackend('https://[0:0:0:0:0:0:0:1]/api', 'test-key');
27
+ const url = backend.getTraceUrl('trace-123');
28
+ assert.strictEqual(url, '/trace/trace-123', 'Should block long-form IPv6 localhost');
29
+ });
30
+ it('should block IPv4-mapped IPv6 localhost', () => {
31
+ const backend = new SigNozApiBackend('https://[::ffff:127.0.0.1]/api', 'test-key');
32
+ const url = backend.getTraceUrl('trace-123');
33
+ assert.strictEqual(url, '/trace/trace-123', 'Should block IPv4-mapped localhost');
34
+ });
35
+ it('should block .localhost TLD', () => {
36
+ const backend = new SigNozApiBackend('https://myapp.localhost/api', 'test-key');
37
+ const url = backend.getTraceUrl('trace-123');
38
+ assert.strictEqual(url, '/trace/trace-123', 'Should block .localhost TLD');
39
+ });
40
+ it('should block private 192.168.x.x ranges', () => {
41
+ const backend = new SigNozApiBackend('https://192.168.1.1/api', 'test-key');
42
+ const url = backend.getTraceUrl('trace-123');
43
+ assert.strictEqual(url, '/trace/trace-123', 'Should block 192.168.x.x');
44
+ });
45
+ it('should block private 10.x.x.x ranges', () => {
46
+ const backend = new SigNozApiBackend('https://10.0.0.1/api', 'test-key');
47
+ const url = backend.getTraceUrl('trace-123');
48
+ assert.strictEqual(url, '/trace/trace-123', 'Should block 10.x.x.x');
49
+ });
50
+ it('should block private 172.16-31.x.x ranges', () => {
51
+ const backend = new SigNozApiBackend('https://172.16.0.1/api', 'test-key');
52
+ const url = backend.getTraceUrl('trace-123');
53
+ assert.strictEqual(url, '/trace/trace-123', 'Should block 172.16.x.x');
54
+ });
55
+ it('should block IPv6 unique local addresses (fc00::/7)', () => {
56
+ const backend = new SigNozApiBackend('https://[fc00::1]/api', 'test-key');
57
+ const url = backend.getTraceUrl('trace-123');
58
+ assert.strictEqual(url, '/trace/trace-123', 'Should block fc00:: ULA');
59
+ });
60
+ it('should block IPv6 link-local addresses (fe80::)', () => {
61
+ const backend = new SigNozApiBackend('https://[fe80::1]/api', 'test-key');
62
+ const url = backend.getTraceUrl('trace-123');
63
+ assert.strictEqual(url, '/trace/trace-123', 'Should block fe80:: link-local');
64
+ });
65
+ it('should block .local domain', () => {
66
+ const backend = new SigNozApiBackend('https://myhost.local/api', 'test-key');
67
+ const url = backend.getTraceUrl('trace-123');
68
+ assert.strictEqual(url, '/trace/trace-123', 'Should block .local domain');
69
+ });
70
+ it('should block .internal domain', () => {
71
+ const backend = new SigNozApiBackend('https://signoz.internal/api', 'test-key');
72
+ const url = backend.getTraceUrl('trace-123');
73
+ assert.strictEqual(url, '/trace/trace-123', 'Should block .internal domain');
74
+ });
75
+ it('should block .home.arpa domain', () => {
76
+ const backend = new SigNozApiBackend('https://router.home.arpa/api', 'test-key');
77
+ const url = backend.getTraceUrl('trace-123');
78
+ assert.strictEqual(url, '/trace/trace-123', 'Should block .home.arpa domain');
79
+ });
80
+ it('should allow valid external HTTPS URL', () => {
81
+ const backend = new SigNozApiBackend('https://signoz.example.com/api/', 'test-key');
82
+ const url = backend.getTraceUrl('trace-123');
83
+ assert.ok(url.includes('signoz.example.com'), 'Should allow valid external URL');
84
+ });
85
+ it('should block HTTP protocol', () => {
86
+ const backend = new SigNozApiBackend('http://signoz.example.com/api', 'test-key');
87
+ const url = backend.getTraceUrl('trace-123');
88
+ assert.strictEqual(url, '/trace/trace-123', 'Should block HTTP protocol');
89
+ });
90
+ // A2 Category 6: SSRF Advanced Tests
91
+ describe('advanced SSRF protection (A2)', () => {
92
+ it('should block URL-encoded localhost', () => {
93
+ // %6c%6f%63%61%6c%68%6f%73%74 = localhost
94
+ const backend = new SigNozApiBackend('https://%6c%6f%63%61%6c%68%6f%73%74/api', 'test-key');
95
+ const url = backend.getTraceUrl('trace-123');
96
+ assert.strictEqual(url, '/trace/trace-123', 'Should block URL-encoded localhost');
97
+ });
98
+ it('should block double-encoded localhost', () => {
99
+ // %256c = %l (double-encoded l)
100
+ const backend = new SigNozApiBackend('https://%256c%256f%256c%256f%256c%2568%256f%2573%2574/api', 'test-key');
101
+ const url = backend.getTraceUrl('trace-123');
102
+ // URL constructor may fail or return blocked URL
103
+ assert.strictEqual(url, '/trace/trace-123', 'Should block double-encoded localhost');
104
+ });
105
+ it('should block IPv6 compressed format for localhost', () => {
106
+ const backend = new SigNozApiBackend('https://[::ffff:7f00:1]/api', 'test-key');
107
+ const url = backend.getTraceUrl('trace-123');
108
+ assert.strictEqual(url, '/trace/trace-123', 'Should block ::ffff:7f00:1');
109
+ });
110
+ it('should block IPv6 with zone ID', () => {
111
+ // fe80::1%eth0 - link-local with zone ID
112
+ const backend = new SigNozApiBackend('https://[fe80::1%25eth0]/api', 'test-key');
113
+ const url = backend.getTraceUrl('trace-123');
114
+ assert.strictEqual(url, '/trace/trace-123', 'Should block IPv6 with zone ID');
115
+ });
116
+ it('should block decimal IP representation of localhost', () => {
117
+ // 2130706433 = 127.0.0.1 in decimal
118
+ const backend = new SigNozApiBackend('https://2130706433/api', 'test-key');
119
+ const url = backend.getTraceUrl('trace-123');
120
+ // May be parsed as hostname (not blocked by URL constructor)
121
+ // But our SSRF check should catch it if it resolves
122
+ assert.strictEqual(typeof url, 'string');
123
+ });
124
+ it('should block octal IP representation', () => {
125
+ // 0177.0.0.1 = 127.0.0.1 in octal
126
+ const backend = new SigNozApiBackend('https://0177.0.0.1/api', 'test-key');
127
+ const url = backend.getTraceUrl('trace-123');
128
+ // Octal notation may or may not be parsed by URL
129
+ assert.strictEqual(typeof url, 'string');
130
+ });
131
+ it('should block AWS metadata endpoint (169.254.169.254)', () => {
132
+ // M-INFO-1: Cloud metadata endpoints now blocked
133
+ const backend = new SigNozApiBackend('https://169.254.169.254/api', 'test-key');
134
+ const url = backend.getTraceUrl('trace-123');
135
+ assert.strictEqual(url, '/trace/trace-123', 'Should block AWS metadata endpoint');
136
+ });
137
+ it('should block GCP metadata endpoint', () => {
138
+ // .internal TLD is blocked
139
+ const backend = new SigNozApiBackend('https://metadata.google.internal/api', 'test-key');
140
+ const url = backend.getTraceUrl('trace-123');
141
+ assert.strictEqual(url, '/trace/trace-123', 'Should block GCP metadata');
142
+ });
143
+ it('should handle IPv4-mapped private ranges', () => {
144
+ // ::ffff:192.168.1.1 gets normalized by URL constructor to ::ffff:c0a8:101
145
+ // Current SSRF check uses string pattern that doesn't match this format
146
+ // Security note: Consider enhancing IPv4-mapped detection
147
+ const backend = new SigNozApiBackend('https://[::ffff:192.168.1.1]/api', 'test-key');
148
+ const url = backend.getTraceUrl('trace-123');
149
+ // Document current behavior - may not be blocked due to hex normalization
150
+ assert.strictEqual(typeof url, 'string');
151
+ });
152
+ it('should block link-local IPv4 addresses (169.254.0.0/16)', () => {
153
+ // M-INFO-1: Full link-local range now blocked
154
+ const backend = new SigNozApiBackend('https://169.254.1.1/api', 'test-key');
155
+ const url = backend.getTraceUrl('trace-123');
156
+ assert.strictEqual(url, '/trace/trace-123', 'Should block link-local addresses');
157
+ });
158
+ it('should block fd00:: IPv6 unique local', () => {
159
+ const backend = new SigNozApiBackend('https://[fd00::1]/api', 'test-key');
160
+ const url = backend.getTraceUrl('trace-123');
161
+ assert.strictEqual(url, '/trace/trace-123', 'Should block fd00:: ULA');
162
+ });
163
+ it('should handle empty hostname', () => {
164
+ // Empty hostname is invalid
165
+ try {
166
+ const backend = new SigNozApiBackend('https:///api', 'test-key');
167
+ const url = backend.getTraceUrl('trace-123');
168
+ assert.strictEqual(url, '/trace/trace-123');
169
+ }
170
+ catch {
171
+ // URL constructor may throw for invalid URL
172
+ }
173
+ });
174
+ it('should handle username:password@ in URL', () => {
175
+ // Credentials in URL should not bypass SSRF check
176
+ const backend = new SigNozApiBackend('https://user:pass@localhost/api', 'test-key');
177
+ const url = backend.getTraceUrl('trace-123');
178
+ assert.strictEqual(url, '/trace/trace-123', 'Should block localhost with credentials');
179
+ });
180
+ it('should handle localhost with port', () => {
181
+ const backend = new SigNozApiBackend('https://localhost:8080/api', 'test-key');
182
+ const url = backend.getTraceUrl('trace-123');
183
+ assert.strictEqual(url, '/trace/trace-123', 'Should block localhost with port');
184
+ });
185
+ it('should block full 127.0.0.0/8 loopback range', () => {
186
+ // M-INFO-1: Full loopback range now blocked
187
+ const addresses = ['127.0.0.2', '127.1.1.1', '127.255.255.254'];
188
+ for (const addr of addresses) {
189
+ const backend = new SigNozApiBackend(`https://${addr}/api`, 'test-key');
190
+ const url = backend.getTraceUrl('trace-123');
191
+ assert.strictEqual(url, '/trace/trace-123', `Should block ${addr}`);
192
+ }
193
+ });
194
+ it('should handle 0.0.0.0', () => {
195
+ const backend = new SigNozApiBackend('https://0.0.0.0/api', 'test-key');
196
+ const url = backend.getTraceUrl('trace-123');
197
+ assert.strictEqual(url, '/trace/trace-123', 'Should block 0.0.0.0');
198
+ });
199
+ it('should block Kubernetes internal domains', () => {
200
+ const internalDomains = [
201
+ 'https://kubernetes.default.svc/api',
202
+ 'https://kubernetes.default/api',
203
+ 'https://service.namespace.svc.cluster.local/api',
204
+ ];
205
+ for (const domain of internalDomains) {
206
+ const backend = new SigNozApiBackend(domain, 'test-key');
207
+ const url = backend.getTraceUrl('trace-123');
208
+ // Should block internal/local domains
209
+ if (domain.includes('.local')) {
210
+ assert.strictEqual(url, '/trace/trace-123', `Should block ${domain}`);
211
+ }
212
+ }
213
+ });
214
+ });
215
+ });
216
+ //# sourceMappingURL=signoz-api-ssrf.test.js.map