source.dussan.org Git - sonarqube.git/blob

   1 /*
   2  * SonarQube
   3  * Copyright (C) 2009-2023 SonarSource SA
   4  * mailto:info AT sonarsource DOT com
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 3 of the License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public License
  17  * along with this program; if not, write to the Free Software Foundation,
  18  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  19  */
  20 import { setImmediate } from 'timers';
  21 import { getIndexationStatus } from '../../../../api/ce';
  22 import { get, remove, save } from '../../../../helpers/storage';
  23 import { IndexationStatus } from '../../../../types/indexation';
  24 import IndexationNotificationHelper from '../IndexationNotificationHelper';
  25
  26 beforeEach(() => {
  27   jest.clearAllMocks();
  28   jest.useFakeTimers();
  29 });
  30
  31 afterEach(() => {
  32   jest.runOnlyPendingTimers();
  33   jest.useRealTimers();
  34 });
  35
  36 jest.mock('../../../../api/ce', () => ({
  37   getIndexationStatus: jest.fn().mockResolvedValue({}),
  38 }));
  39
  40 jest.mock('../../../../helpers/storage', () => ({
  41   get: jest.fn(),
  42   remove: jest.fn(),
  43   save: jest.fn(),
  44 }));
  45
  46 it('should properly start & stop polling for indexation status', async () => {
  47   const onNewStatus = jest.fn();
  48   const newStatus: IndexationStatus = {
  49     isCompleted: false,
  50     percentCompleted: 100,
  51     hasFailures: false,
  52   };
  53   (getIndexationStatus as jest.Mock).mockResolvedValueOnce(newStatus);
  54
  55   IndexationNotificationHelper.startPolling(onNewStatus);
  56   expect(getIndexationStatus).toHaveBeenCalled();
  57
  58   await new Promise(setImmediate);
  59   expect(onNewStatus).toHaveBeenCalledWith(newStatus);
  60
  61   jest.runOnlyPendingTimers();
  62   expect(getIndexationStatus).toHaveBeenCalledTimes(2);
  63
  64   (getIndexationStatus as jest.Mock).mockClear();
  65
  66   IndexationNotificationHelper.stopPolling();
  67   jest.runAllTimers();
  68
  69   expect(getIndexationStatus).not.toHaveBeenCalled();
  70 });
  71
  72 it('should properly handle the flag to show the completed banner', () => {
  73   IndexationNotificationHelper.markCompletedNotificationAsToDisplay();
  74
  75   expect(save).toHaveBeenCalledWith(expect.any(String), 'true');
  76
  77   (get as jest.Mock).mockReturnValueOnce('true');
  78   let shouldDisplay = IndexationNotificationHelper.shouldDisplayCompletedNotification();
  79
  80   expect(shouldDisplay).toBe(true);
  81   expect(get).toHaveBeenCalled();
  82
  83   IndexationNotificationHelper.markCompletedNotificationAsDisplayed();
  84
  85   expect(remove).toHaveBeenCalled();
  86
  87   shouldDisplay = IndexationNotificationHelper.shouldDisplayCompletedNotification();
  88
  89   expect(shouldDisplay).toBe(false);
  90 });