1 files changed, 247 insertions, 0 deletions
diff --git a/src/pybind/mgr/dashboard/frontend/src/app/shared/api/prometheus.service.spec.ts b/src/pybind/mgr/dashboard/frontend/src/app/shared/api/prometheus.service.spec.ts
new file mode 100644
index 000000000..c42f6e7ac
--- /dev/null
+++ b/src/pybind/mgr/dashboard/frontend/src/app/shared/api/prometheus.service.spec.ts
@@ -0,0 +1,247 @@
+import { HttpClientTestingModule, HttpTestingController } from '@angular/common/http/testing';
+import { TestBed } from '@angular/core/testing';
+
+import { configureTestBed } from '~/testing/unit-test-helper';
+import { AlertmanagerNotification } from '../models/prometheus-alerts';
+import { PrometheusService } from './prometheus.service';
+import { SettingsService } from './settings.service';
+
+describe('PrometheusService', () => {
+  let service: PrometheusService;
+  let httpTesting: HttpTestingController;
+
+  configureTestBed({
+    providers: [PrometheusService, SettingsService],
+    imports: [HttpClientTestingModule]
+  });
+
+  beforeEach(() => {
+    service = TestBed.inject(PrometheusService);
+    httpTesting = TestBed.inject(HttpTestingController);
+  });
+
+  afterEach(() => {
+    httpTesting.verify();
+  });
+
+  it('should be created', () => {
+    expect(service).toBeTruthy();
+  });
+
+  it('should get alerts', () => {
+    service.getAlerts().subscribe();
+    const req = httpTesting.expectOne('api/prometheus');
+    expect(req.request.method).toBe('GET');
+  });
+
+  it('should get silences', () => {
+    service.getSilences().subscribe();
+    const req = httpTesting.expectOne('api/prometheus/silences');
+    expect(req.request.method).toBe('GET');
+  });
+
+  it('should set a silence', () => {
+    const silence = {
+      id: 'someId',
+      matchers: [
+        {
+          name: 'getZero',
+          value: 0,
+          isRegex: false
+        }
+      ],
+      startsAt: '2019-01-25T14:32:46.646300974Z',
+      endsAt: '2019-01-25T18:32:46.646300974Z',
+      createdBy: 'someCreator',
+      comment: 'for testing purpose'
+    };
+    service.setSilence(silence).subscribe();
+    const req = httpTesting.expectOne('api/prometheus/silence');
+    expect(req.request.method).toBe('POST');
+    expect(req.request.body).toEqual(silence);
+  });
+
+  it('should expire a silence', () => {
+    service.expireSilence('someId').subscribe();
+    const req = httpTesting.expectOne('api/prometheus/silence/someId');
+    expect(req.request.method).toBe('DELETE');
+  });
+
+  it('should call getNotificationSince without a notification', () => {
+    service.getNotifications().subscribe();
+    const req = httpTesting.expectOne('api/prometheus/notifications?from=last');
+    expect(req.request.method).toBe('GET');
+  });
+
+  it('should call getNotificationSince with notification', () => {
+    service.getNotifications({ id: '42' } as AlertmanagerNotification).subscribe();
+    const req = httpTesting.expectOne('api/prometheus/notifications?from=42');
+    expect(req.request.method).toBe('GET');
+  });
+
+  describe('test getRules()', () => {
+    let data: {}; // Subset of PrometheusRuleGroup to keep the tests concise.
+
+    beforeEach(() => {
+      data = {
+        groups: [
+          {
+            name: 'test',
+            rules: [
+              {
+                name: 'load_0',
+                type: 'alerting'
+              },
+              {
+                name: 'load_1',
+                type: 'alerting'
+              },
+              {
+                name: 'load_2',
+                type: 'alerting'
+              }
+            ]
+          },
+          {
+            name: 'recording_rule',
+            rules: [
+              {
+                name: 'node_memory_MemUsed_percent',
+                type: 'recording'
+              }
+            ]
+          }
+        ]
+      };
+    });
+
+    it('should get rules without applying filters', () => {
+      service.getRules().subscribe((rules) => {
+        expect(rules).toEqual(data);
+      });
+
+      const req = httpTesting.expectOne('api/prometheus/rules');
+      expect(req.request.method).toBe('GET');
+      req.flush(data);
+    });
+
+    it('should get rewrite rules only', () => {
+      service.getRules('rewrites').subscribe((rules) => {
+        expect(rules).toEqual({
+          groups: [
+            { name: 'test', rules: [] },
+            { name: 'recording_rule', rules: [] }
+          ]
+        });
+      });
+
+      const req = httpTesting.expectOne('api/prometheus/rules');
+      expect(req.request.method).toBe('GET');
+      req.flush(data);
+    });
+
+    it('should get alerting rules only', () => {
+      service.getRules('alerting').subscribe((rules) => {
+        expect(rules).toEqual({
+          groups: [
+            {
+              name: 'test',
+              rules: [
+                { name: 'load_0', type: 'alerting' },
+                { name: 'load_1', type: 'alerting' },
+                { name: 'load_2', type: 'alerting' }
+              ]
+            },
+            { name: 'recording_rule', rules: [] }
+          ]
+        });
+      });
+
+      const req = httpTesting.expectOne('api/prometheus/rules');
+      expect(req.request.method).toBe('GET');
+      req.flush(data);
+    });
+  });
+
+  describe('ifAlertmanagerConfigured', () => {
+    let x: any;
+    let host: string;
+
+    const receiveConfig = () => {
+      const req = httpTesting.expectOne('api/settings/alertmanager-api-host');
+      expect(req.request.method).toBe('GET');
+      req.flush({ value: host });
+    };
+
+    beforeEach(() => {
+      x = false;
+      TestBed.inject(SettingsService)['settings'] = {};
+      service.ifAlertmanagerConfigured(
+        (v) => (x = v),
+        () => (x = [])
+      );
+      host = 'http://localhost:9093';
+    });
+
+    it('changes x in a valid case', () => {
+      expect(x).toBe(false);
+      receiveConfig();
+      expect(x).toBe(host);
+    });
+
+    it('does changes x an empty array in a invalid case', () => {
+      host = '';
+      receiveConfig();
+      expect(x).toEqual([]);
+    });
+
+    it('disables the set setting', () => {
+      receiveConfig();
+      service.disableAlertmanagerConfig();
+      x = false;
+      service.ifAlertmanagerConfigured((v) => (x = v));
+      expect(x).toBe(false);
+    });
+  });
+
+  describe('ifPrometheusConfigured', () => {
+    let x: any;
+    let host: string;
+
+    const receiveConfig = () => {
+      const req = httpTesting.expectOne('api/settings/prometheus-api-host');
+      expect(req.request.method).toBe('GET');
+      req.flush({ value: host });
+    };
+
+    beforeEach(() => {
+      x = false;
+      TestBed.inject(SettingsService)['settings'] = {};
+      service.ifPrometheusConfigured(
+        (v) => (x = v),
+        () => (x = [])
+      );
+      host = 'http://localhost:9090';
+    });
+
+    it('changes x in a valid case', () => {
+      expect(x).toBe(false);
+      receiveConfig();
+      expect(x).toBe(host);
+    });
+
+    it('does changes x an empty array in a invalid case', () => {
+      host = '';
+      receiveConfig();
+      expect(x).toEqual([]);
+    });
+
+    it('disables the set setting', () => {
+      receiveConfig();
+      service.disablePrometheusConfig();
+      x = false;
+      service.ifPrometheusConfigured((v) => (x = v));
+      expect(x).toBe(false);
+    });
+  });
+});