Merge "Added support for using oauth token for Kafka"
[nonrtric/plt/ranpm.git] / datafilecollector / src / test / java / org / onap / dcaegen2 / collectors / datafile / utils / LoggingUtils.java
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.utils;
22
23 import ch.qos.logback.classic.Level;
24 import ch.qos.logback.classic.Logger;
25 import ch.qos.logback.classic.spi.ILoggingEvent;
26 import ch.qos.logback.core.read.ListAppender;
27
28 import org.slf4j.LoggerFactory;
29
30 public class LoggingUtils {
31
32     /**
33      * Returns a ListAppender that contains all logging events. Call this method at the very beginning of the test
34      */
35     public static ListAppender<ILoggingEvent> getLogListAppender(Class<?> logClass) {
36         return getLogListAppender(logClass, false);
37     }
38
39     /**
40      * Returns a ListAppender that contains all logging events. Call this method at the very beginning of the test
41      *
42      * @param logClass class whose appender is wanted.
43      * @param allLevels true if all log levels should be activated.
44      */
45     public static ListAppender<ILoggingEvent> getLogListAppender(Class<?> logClass, boolean allLevels) {
46         Logger logger = (Logger) LoggerFactory.getLogger(logClass);
47         if (allLevels) {
48             logger.setLevel(Level.ALL);
49         }
50         ListAppender<ILoggingEvent> listAppender = new ListAppender<>();
51         listAppender.start();
52         logger.addAppender(listAppender);
53
54         return listAppender;
55     }
56 }