metron-interface/metron-alerts/src/app/alerts/configure-rows/show-hide/show-hide.service.spec.ts

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
import { TestBed, inject, getTestBed } from '@angular/core/testing';

import { ShowHideService } from './show-hide.service';
import { QueryBuilder } from 'app/alerts/alerts-list/query-builder';

import { Spy } from 'jasmine-core';
import { Filter } from 'app/model/filter';

class QueryBuilderMock {
  addOrUpdateFilter = () => {};
  removeFilter = () => {};
}

describe('ShowHideService', () => {
  let queryBuilderMock: QueryBuilderMock;

  beforeEach(() => {
    spyOn(localStorage, 'getItem').and.returnValues('true', 'false');
    spyOn(localStorage, 'setItem');

    spyOn(ShowHideService.prototype, 'setFilterFor').and.callThrough();

    TestBed.configureTestingModule({
      providers: [
        ShowHideService,
        { provide: QueryBuilder, useClass: QueryBuilderMock },
      ]
    });

    queryBuilderMock = getTestBed().get(QueryBuilder);
  });

  it('should be created', inject([ShowHideService], (service: ShowHideService) => {
    expect(service).toBeTruthy();
  }));

  it('should have QueryBuilder injected', inject([ShowHideService], (service: ShowHideService) => {
    expect(service.queryBuilder).toBeTruthy();
  }));

  it('should get persisted state from localStorage', inject([ShowHideService], (service: ShowHideService) => {
    expect(localStorage.getItem).toHaveBeenCalledWith(service.HIDE_RESOLVE_STORAGE_KEY);
    expect(localStorage.getItem).toHaveBeenCalledWith(service.HIDE_DISMISS_STORAGE_KEY);
  }));

  it('should set initial filter state', inject([ShowHideService], (service: ShowHideService) => {
    expect((service.setFilterFor as Spy).calls.argsFor(0)[1]).toBe(true);
    expect((service.setFilterFor as Spy).calls.argsFor(0)[0]).toBe('RESOLVE');
    expect((service.setFilterFor as Spy).calls.argsFor(1)[0]).toBe('DISMISS');
    expect((service.setFilterFor as Spy).calls.argsFor(1)[1]).toBe(false);
  }));

  it('should set value loaded from localStorage to hideDismissed ', inject([ShowHideService], (service: ShowHideService) => {
    expect(service.hideDismissed).toBe(false);
  }));

  it('should set value loaded from localStorage to hideResolved', inject([ShowHideService], (service: ShowHideService) => {
    expect(service.hideResolved).toBe(true);
  }));

  it('should save state to localStorage on change for RESOLVE', inject([ShowHideService], (service: ShowHideService) => {
    service.setFilterFor('RESOLVE', true);

    expect(localStorage.setItem).toHaveBeenCalledWith(service.HIDE_RESOLVE_STORAGE_KEY, true);
  }));

  it('should save state to localStorage on change for DISMISS', inject([ShowHideService], (service: ShowHideService) => {
    service.setFilterFor('DISMISS', true);

    expect(localStorage.setItem).toHaveBeenCalledWith(service.HIDE_DISMISS_STORAGE_KEY, true);
  }));

  it('should be able to add RESOLVE filter to QueryBuilder', inject([ShowHideService], (service: ShowHideService) => {
    spyOn(queryBuilderMock, 'addOrUpdateFilter');
    spyOn(queryBuilderMock, 'removeFilter');

    service.setFilterFor('RESOLVE', true);
    expect(queryBuilderMock.addOrUpdateFilter).toHaveBeenCalledWith(new Filter('-alert_status', 'RESOLVE', false));
    expect(queryBuilderMock.removeFilter).not.toHaveBeenCalled();
  }));

  it('should be able to remove RESOLVE filter to QueryBuilder', inject([ShowHideService], (service: ShowHideService) => {
    spyOn(queryBuilderMock, 'addOrUpdateFilter');
    spyOn(queryBuilderMock, 'removeFilter');

    service.setFilterFor('RESOLVE', false);
    expect(queryBuilderMock.removeFilter).toHaveBeenCalledWith(new Filter('-alert_status', 'RESOLVE', false));
    expect(queryBuilderMock.addOrUpdateFilter).not.toHaveBeenCalled();
  }));

  it('should be able to add DISMISS filter to QueryBuilder', inject([ShowHideService], (service: ShowHideService) => {
    spyOn(queryBuilderMock, 'addOrUpdateFilter');
    spyOn(queryBuilderMock, 'removeFilter');

    service.setFilterFor('DISMISS', true);
    expect(queryBuilderMock.addOrUpdateFilter).toHaveBeenCalledWith(new Filter('-alert_status', 'DISMISS', false));
    expect(queryBuilderMock.removeFilter).not.toHaveBeenCalled();
  }));

  it('should be able to remove DISMISS filter to QueryBuilder', inject([ShowHideService], (service: ShowHideService) => {
    spyOn(queryBuilderMock, 'addOrUpdateFilter');
    spyOn(queryBuilderMock, 'removeFilter');

    service.setFilterFor('DISMISS', false);
    expect(queryBuilderMock.removeFilter).toHaveBeenCalledWith(new Filter('-alert_status', 'DISMISS', false));
    expect(queryBuilderMock.addOrUpdateFilter).not.toHaveBeenCalled();
  }));
});