datarouter-prov/src/test/java/org/onap/dmaap/datarouter/provisioning/DrServletTestBase.java

   1 /*******************************************************************************
   2  * ============LICENSE_START==================================================
   3  * * org.onap.dmaap
   4  * * ===========================================================================
   5  * * Copyright © 2017 AT&T Intellectual Property. All rights reserved.
   6  * * ===========================================================================
   7  * * Licensed under the Apache License, Version 2.0 (the "License");
   8  * * you may not use this file except in compliance with the License.
   9  * * You may obtain a copy of the License at
  10  * *
  11  *  *      http://www.apache.org/licenses/LICENSE-2.0
  12  * *
  13  *  * Unless required by applicable law or agreed to in writing, software
  14  * * distributed under the License is distributed on an "AS IS" BASIS,
  15  * * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * * See the License for the specific language governing permissions and
  17  * * limitations under the License.
  18  * * ============LICENSE_END====================================================
  19  * *
  20  * * ECOMP is a trademark and service mark of AT&T Intellectual Property.
  21  * *
  22  ******************************************************************************/
  23
  24 package org.onap.dmaap.datarouter.provisioning;
  25
  26 import ch.qos.logback.classic.Logger;
  27 import ch.qos.logback.classic.spi.ILoggingEvent;
  28 import ch.qos.logback.core.read.ListAppender;
  29 import org.apache.commons.lang3.reflect.FieldUtils;
  30 import org.junit.After;
  31 import org.junit.AfterClass;
  32 import org.junit.Assert;
  33 import org.junit.Before;
  34 import org.onap.dmaap.datarouter.provisioning.utils.DB;
  35 import org.slf4j.LoggerFactory;
  36
  37 import java.io.File;
  38 import java.io.FileNotFoundException;
  39 import java.io.PrintWriter;
  40 import java.util.Properties;
  41 import java.util.Scanner;
  42
  43 import static org.junit.Assert.assertEquals;
  44 import static org.mockito.Mockito.mock;
  45 import static org.mockito.Mockito.when;
  46
  47 public class DrServletTestBase {
  48
  49
  50     @Before
  51     public void setUp() throws Exception {
  52         Properties props = new Properties();
  53         props.setProperty("org.onap.dmaap.datarouter.provserver.isaddressauthenabled", "false");
  54         props.setProperty("org.onap.dmaap.datarouter.provserver.accesslog.dir", "unit-test-logs");
  55         props.setProperty("org.onap.dmaap.datarouter.provserver.spooldir", "unit-test-logs/spool");
  56         props.setProperty("org.onap.dmaap.datarouter.provserver.https.relaxation", "false");
  57         FieldUtils.writeDeclaredStaticField(DB.class, "props", props, true);
  58         FieldUtils.writeDeclaredStaticField(BaseServlet.class, "startmsgFlag", false, true);
  59         SynchronizerTask synchronizerTask = mock(SynchronizerTask.class);
  60         when(synchronizerTask.getState()).thenReturn(SynchronizerTask.UNKNOWN);
  61         FieldUtils.writeDeclaredStaticField(BaseServlet.class, "synctask", synchronizerTask, true);
  62     }
  63
  64     public ListAppender<ILoggingEvent> setTestLogger(Class c) {
  65         Logger logger = (Logger) LoggerFactory.getLogger(c);
  66         ListAppender<ILoggingEvent> listAppender = new ListAppender<>();
  67         listAppender.start();
  68         logger.addAppender(listAppender);
  69         return listAppender;
  70     }
  71
  72     public void verifyEnteringExitCalled(ListAppender<ILoggingEvent> listAppender) {
  73         assertEquals("EELF0004I  Entering data router provisioning component with RequestId and InvocationId", listAppender.list.get(0).getMessage());
  74         assertEquals("EELF0005I  Exiting data router provisioning component with RequestId and InvocationId", listAppender.list.get(2).getMessage());
  75         assertEquals(3, listAppender.list.size());
  76     }
  77
  78     @After
  79     public void tearDown() throws Exception {
  80     }
  81 }