2 * ============LICENSE_START=======================================================
3 * Copyright (C) 2023 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
9 * http://www.apache.org/licenses/LICENSE-2.0
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.
17 * SPDX-License-Identifier: Apache-2.0
18 * ============LICENSE_END=========================================================
21 package org.oran.datafile.tasks;
23 import static org.junit.jupiter.api.Assertions.assertNull;
24 import static org.junit.jupiter.api.Assertions.assertSame;
25 import static org.mockito.Mockito.doNothing;
26 import static org.mockito.Mockito.verify;
27 import static org.mockito.Mockito.when;
29 import org.junit.jupiter.api.Test;
30 import org.junit.jupiter.api.extension.ExtendWith;
31 import org.mockito.Mock;
32 import org.mockito.Mockito;
33 import org.mockito.junit.jupiter.MockitoExtension;
34 import org.oran.datafile.configuration.AppConfig;
35 import org.oran.datafile.model.Counters;
36 import org.oran.datafile.model.FilePublishInformation;
37 import org.oran.datafile.oauth2.SecurityContext;
38 import org.springframework.test.context.ContextConfiguration;
39 import reactor.core.publisher.DirectProcessor;
40 import reactor.kafka.sender.SenderResult;
42 @ContextConfiguration(classes = {CollectAndReportFiles.class})
43 @ExtendWith(MockitoExtension.class)
44 class CollectAndReportFilesTest {
46 private AppConfig appConfig;
49 private CollectAndReportFiles collectAndReportFilesMock;
52 private SecurityContext securityContext;
56 doNothing().when(collectAndReportFilesMock).start();
57 collectAndReportFilesMock.start();
58 verify(collectAndReportFilesMock).start();
61 void testCreateMainTask() {
62 DirectProcessor<FilePublishInformation> createResult = DirectProcessor.create();
63 when(collectAndReportFilesMock.createMainTask()).thenReturn(createResult);
64 assertSame(createResult, collectAndReportFilesMock.createMainTask());
65 verify(collectAndReportFilesMock).createMainTask();
68 void testSendDataToStream() {
69 DirectProcessor<SenderResult<Integer>> createResult = DirectProcessor.create();
71 collectAndReportFilesMock.sendDataToStream(Mockito.<String>any(), Mockito.<String>any(), Mockito.<String>any()))
72 .thenReturn(createResult);
73 assertSame(createResult, collectAndReportFilesMock.sendDataToStream("Topic", "Source Name", "42"));
74 verify(collectAndReportFilesMock).sendDataToStream(Mockito.<String>any(), Mockito.<String>any(),
75 Mockito.<String>any());
78 void testCreateFileCollector() {
79 FileCollector fileCollector = new FileCollector(securityContext, appConfig, new Counters());
81 when(collectAndReportFilesMock.createFileCollector()).thenReturn(fileCollector);
82 assertSame(fileCollector, collectAndReportFilesMock.createFileCollector());
83 verify(collectAndReportFilesMock).createFileCollector();
86 void testParseReceivedFileReadyMessage() {
87 when(collectAndReportFilesMock.parseReceivedFileReadyMessage(Mockito.<KafkaTopicListener.DataFromTopic>any()))
90 collectAndReportFilesMock.parseReceivedFileReadyMessage(new KafkaTopicListener.DataFromTopic("Key", "42")));
91 verify(collectAndReportFilesMock).parseReceivedFileReadyMessage(Mockito.<KafkaTopicListener.DataFromTopic>any());