gerrit.onap Code Review - dcaegen2/collectors/datafile.git/blob

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2019 Nordix Foundation.
   4  * ================================================================================
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *      http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  *
  17  * SPDX-License-Identifier: Apache-2.0
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.onap.dcaegen2.collectors.datafile.configuration;
  22
  23 import static org.junit.Assert.assertFalse;
  24 import static org.junit.jupiter.api.Assertions.assertEquals;
  25 import static org.junit.jupiter.api.Assertions.assertTrue;
  26 import static org.mockito.ArgumentMatchers.any;
  27 import static org.mockito.ArgumentMatchers.eq;
  28 import static org.mockito.Mockito.mock;
  29 import static org.mockito.Mockito.spy;
  30 import static org.mockito.Mockito.verify;
  31 import static org.mockito.Mockito.verifyNoMoreInteractions;
  32
  33 import java.time.Duration;
  34 import java.time.Instant;
  35 import java.util.ArrayList;
  36 import java.util.List;
  37 import java.util.concurrent.ScheduledFuture;
  38 import org.junit.jupiter.api.Test;
  39 import org.mockito.ArgumentCaptor;
  40 import org.onap.dcaegen2.collectors.datafile.tasks.ScheduledTasks;
  41 import org.springframework.http.HttpStatus;
  42 import org.springframework.http.ResponseEntity;
  43 import org.springframework.scheduling.TaskScheduler;
  44 import reactor.test.StepVerifier;
  45
  46 public class SchedulerConfigTest {
  47
  48     @Test
  49     public void getResponseFromCancellationOfTasks_success() {
  50         List<ScheduledFuture<?>> scheduledFutureList = new ArrayList<>();
  51         ScheduledFuture<?> scheduledFutureMock = mock(ScheduledFuture.class);
  52         scheduledFutureList.add(scheduledFutureMock);
  53
  54         SchedulerConfig.setScheduledFutureList(scheduledFutureList);
  55
  56         SchedulerConfig schedulerUnderTest = new SchedulerConfig(null, null, null);
  57
  58         String msg = "Datafile Service has already been stopped!";
  59         StepVerifier.create(schedulerUnderTest.getResponseFromCancellationOfTasks())
  60                 .expectNext(new ResponseEntity<String>(msg, HttpStatus.CREATED)) //
  61                 .verifyComplete();
  62
  63         verify(scheduledFutureMock).cancel(false);
  64         verifyNoMoreInteractions(scheduledFutureMock);
  65
  66         assertEquals(0, scheduledFutureList.size());
  67     }
  68
  69     @Test
  70     public void tryToStartTaskWhenNotStarted_success() {
  71         TaskScheduler taskSchedulerMock = mock(TaskScheduler.class);
  72         ScheduledTasks scheduledTasksMock = mock(ScheduledTasks.class);
  73         CloudConfiguration cloudConfigurationMock = mock(CloudConfiguration.class);
  74         List<ScheduledFuture<?>> scheduledFutureList = new ArrayList<>();
  75
  76         SchedulerConfig.setScheduledFutureList(scheduledFutureList);
  77
  78         SchedulerConfig schedulerUnderTestSpy =
  79                 spy(new SchedulerConfig(taskSchedulerMock, scheduledTasksMock, cloudConfigurationMock));
  80
  81         boolean actualResult = schedulerUnderTestSpy.tryToStartTask();
  82
  83         assertTrue(actualResult);
  84
  85         ArgumentCaptor<Runnable> runTaskRunnableCaptor = ArgumentCaptor.forClass(Runnable.class);
  86         verify(taskSchedulerMock).scheduleAtFixedRate(runTaskRunnableCaptor.capture(), any(Instant.class),
  87                 eq(Duration.ofMinutes(5)));
  88
  89         ArgumentCaptor<Runnable> scheduleMainDatafileEventTaskCaptor = ArgumentCaptor.forClass(Runnable.class);
  90         verify(taskSchedulerMock).scheduleWithFixedDelay(scheduleMainDatafileEventTaskCaptor.capture(),
  91                 eq(Duration.ofSeconds(15)));
  92         ArgumentCaptor<Runnable> purgeCachedInformationCaptor = ArgumentCaptor.forClass(Runnable.class);
  93         verify(taskSchedulerMock).scheduleWithFixedDelay(purgeCachedInformationCaptor.capture(),
  94                 eq(Duration.ofHours(1)));
  95         verifyNoMoreInteractions(taskSchedulerMock);
  96
  97         scheduleMainDatafileEventTaskCaptor.getValue().run();
  98         purgeCachedInformationCaptor.getValue().run();
  99         verify(scheduledTasksMock).purgeCachedInformation(any(Instant.class));
 100         verify(scheduledTasksMock).executeDatafileMainTask();
 101         verifyNoMoreInteractions(scheduledTasksMock);
 102
 103         runTaskRunnableCaptor.getValue().run();
 104         verify(cloudConfigurationMock).runTask();
 105         verifyNoMoreInteractions(cloudConfigurationMock);
 106
 107         assertEquals(3, scheduledFutureList.size());
 108     }
 109
 110     @Test
 111     public void tryToStartTaskWhenAlreadyStarted_shouldReturnFalse() {
 112         List<ScheduledFuture<?>> scheduledFutureList = new ArrayList<>();
 113         ScheduledFuture<?> scheduledFutureMock = mock(ScheduledFuture.class);
 114         scheduledFutureList.add(scheduledFutureMock);
 115
 116         SchedulerConfig.setScheduledFutureList(scheduledFutureList);
 117
 118         SchedulerConfig schedulerUnderTest = new SchedulerConfig(null, null, null);
 119
 120         boolean actualResult = schedulerUnderTest.tryToStartTask();
 121
 122         assertFalse(actualResult);
 123     }
 124 }