test add post review

scality · Jul 4, 2024 · 1c03a2f · 1c03a2f
1 parent ccea8fd
commit 1c03a2f
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/tests/unit/CircuitBreaker.spec.ts b/tests/unit/CircuitBreaker.spec.ts
@@ -2,6 +2,7 @@ import { afterEach, beforeEach, describe, expect, test, jest } from '@jest/globa
 
 import { CircuitBreaker, BreakerState } from '../../src/CircuitBreaker';
 import { defaultProbeEvaluateInterval } from '../../src/Configuration';
+import { PrometheusClient } from '../../src/probes/PrometheusClient';
 
 describe('CircuitBreaker', () => {
     let b: CircuitBreaker;
@@ -135,6 +136,29 @@ describe('CircuitBreaker', () => {
             expect(s).toBeCalled();
         });
 
+        test('should set _failedProbes to true when a probe fails', async () => {
+            const spy = jest.spyOn(PrometheusClient.prototype, 'instantQuery').mockRejectedValue(new Error('Failed to query Prometheus'));
+            const config = {
+                probes: [
+                    {
+                        type: 'prometheusQuery',
+                        prometheus: {
+                            endpoint: 'http://localhost:9090',
+                            timeout: 3000,
+                        },
+                        query: 'up',
+                        threshold: 1,
+                        averagedOverInterval: '1m',
+                    },
+                ],
+            };
+            const circuitBreaker = new CircuitBreaker(config);
+            await circuitBreaker._evaluate();
+            expect(circuitBreaker._failedProbes).toBe(true);
+
+            spy.mockRestore();
+        });
+
         test('should not schedule next evaluation if not started', async () => {
             const config = {
                 probes: [