datarouter-node/src/test/java/org/onap/dmaap/datarouter/node/NodeConfigManagerTest.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.dmaap.datarouter.node;
  22
  23 import static org.mockito.BDDMockito.given;
  24 import static org.mockito.Mockito.mock;
  25 import static org.mockito.Mockito.when;
  26
  27 import java.io.ByteArrayInputStream;
  28 import java.io.File;
  29 import java.io.IOException;
  30 import java.io.InputStream;
  31 import java.net.InetAddress;
  32 import java.net.URL;
  33 import java.net.URLStreamHandler;
  34 import java.net.URLStreamHandlerFactory;
  35 import java.nio.file.Files;
  36 import java.nio.file.Paths;
  37 import java.util.HashMap;
  38 import java.util.Map;
  39 import org.apache.commons.io.FileUtils;
  40 import org.junit.After;
  41 import org.junit.AfterClass;
  42 import org.junit.Assert;
  43 import org.junit.Before;
  44 import org.junit.BeforeClass;
  45 import org.junit.Test;
  46 import org.mockito.Mock;
  47 import org.powermock.core.classloader.annotations.PowerMockIgnore;
  48 import org.powermock.core.classloader.annotations.PrepareForTest;
  49 import sun.net.www.URLConnection;
  50
  51 @PowerMockIgnore({"javax.net.ssl.*", "javax.security.auth.x500.X500Principal", "javax.crypto.*"})
  52 @PrepareForTest({InetAddress.class})
  53 public class NodeConfigManagerTest {
  54
  55     private static HttpUrlStreamHandler httpUrlStreamHandler;
  56
  57     @Mock
  58     InputStream inputStream;
  59
  60     @Mock
  61     NodeConfig nodeConfig;
  62
  63     @BeforeClass
  64     public static void init() {
  65         System.setProperty("org.onap.dmaap.datarouter.node.properties", "src/test/resources/node_test.properties");
  66         // Allows for mocking URL connections
  67         URLStreamHandlerFactory urlStreamHandlerFactory = mock(URLStreamHandlerFactory.class);
  68         URL.setURLStreamHandlerFactory(urlStreamHandlerFactory);
  69         httpUrlStreamHandler = new HttpUrlStreamHandler();
  70         given(urlStreamHandlerFactory.createURLStreamHandler("https")).willReturn(httpUrlStreamHandler);
  71     }
  72
  73     @Before
  74     public void reset() throws IOException {
  75         String href = "https://dmaap-dr-prov:8443/internal/prov";
  76         URLConnection urlConnection = mock(URLConnection.class);
  77         httpUrlStreamHandler.addConnection(new URL(href), urlConnection);
  78         //File prov = new File("src/test/resources/prov_data.json");
  79         InputStream anyInputStream = new ByteArrayInputStream(Files.readAllBytes(Paths.get("src/test/resources/prov_data.json")));
  80         when(urlConnection.getInputStream()).thenReturn(anyInputStream);
  81     }
  82
  83     @After
  84     public void resetHandler() {
  85         httpUrlStreamHandler.resetConnections();
  86     }
  87
  88     @AfterClass
  89     public static void tearDownClass() throws IOException {
  90         FileUtils.deleteDirectory(new File(System.getProperty("user.dir") + "/src/test/resources/spool"));
  91         FileUtils.deleteDirectory(new File(System.getProperty("user.dir") + "/src/test/resources/logs"));
  92     }
  93
  94     @Test
  95     public void Verify_NodeConfigMan_Getters() {
  96         NodeConfigManager nodeConfigManager = NodeConfigManager.getInstance();
  97         Assert.assertEquals("legacy", nodeConfigManager.getAafInstance());
  98         Assert.assertEquals("src/test/resources/spool/f", nodeConfigManager.getSpoolDir());
  99         Assert.assertEquals("src/test/resources/spool", nodeConfigManager.getSpoolBase());
 100         Assert.assertEquals("PKCS12", nodeConfigManager.getKSType());
 101         Assert.assertEquals(8080, nodeConfigManager.getHttpPort());
 102         Assert.assertEquals(8443, nodeConfigManager.getHttpsPort());
 103         Assert.assertEquals(443, nodeConfigManager.getExtHttpsPort());
 104         Assert.assertEquals("dmaap-dr-node", nodeConfigManager.getMyName());
 105         Assert.assertEquals("https://dmaap-dr-prov:8443/internal/logs", nodeConfigManager.getEventLogUrl());
 106         Assert.assertEquals("src/test/resources/logs/events", nodeConfigManager.getEventLogPrefix());
 107         Assert.assertEquals(".log", nodeConfigManager.getEventLogSuffix());
 108         Assert.assertEquals("src/test/resources/logs", nodeConfigManager.getLogDir());
 109         Assert.assertEquals((86400000L * 30), nodeConfigManager.getLogRetention());
 110         Assert.assertEquals(new String[] {"TLSv1.1", "TLSv1.2"}, nodeConfigManager.getEnabledprotocols());
 111         Assert.assertEquals("org.onap.dmaap-dr.feed", nodeConfigManager.getAafType());
 112         Assert.assertEquals("publish", nodeConfigManager.getAafAction());
 113         Assert.assertFalse(nodeConfigManager.getCadiEnabled());
 114         Assert.assertFalse(nodeConfigManager.isShutdown());
 115         Assert.assertTrue(nodeConfigManager.isConfigured());
 116         Assert.assertEquals("legacy", nodeConfigManager.getAafInstance("1"));
 117         Assert.assertNotNull(nodeConfigManager.getPublishId());
 118         Assert.assertNotNull(nodeConfigManager.getAllDests());
 119         Assert.assertEquals(10000, nodeConfigManager.getInitFailureTimer());
 120         Assert.assertEquals(600000, nodeConfigManager.getWaitForFileProcessFailureTimer());
 121         Assert.assertEquals(3600000, nodeConfigManager.getMaxFailureTimer());
 122         Assert.assertEquals(2.0, nodeConfigManager.getFailureBackoff(),0.0);
 123         Assert.assertEquals(86400000, nodeConfigManager.getExpirationTimer());
 124         Assert.assertEquals(100, nodeConfigManager.getFairFileLimit());
 125         Assert.assertEquals(60000, nodeConfigManager.getFairTimeLimit());
 126         Assert.assertNotNull(nodeConfigManager.getTargets("1"));
 127         Assert.assertEquals("src/test/resources/spool/f", nodeConfigManager.getSpoolDir());
 128         Assert.assertEquals("src/test/resources/aaf/org.onap.dmaap-dr.p12", nodeConfigManager.getKSFile());
 129         Assert.assertEquals("jks", nodeConfigManager.getTstype());
 130         Assert.assertEquals("src/test/resources/aaf/org.onap.dmaap-dr.trust.jks", nodeConfigManager.getTsfile());
 131         Assert.assertEquals(40, nodeConfigManager.getDeliveryThreads());
 132         Assert.assertEquals("30", nodeConfigManager.getEventLogInterval());
 133         Assert.assertFalse(nodeConfigManager.isFollowRedirects());
 134         Assert.assertNotNull(nodeConfigManager.getTimer());
 135         Assert.assertEquals("1", nodeConfigManager.getFeedId("1"));
 136         Assert.assertEquals("Basic ZG1hYXAtZHItbm9kZTpsaEFUNHY2N3F3blY3QVFxV3ByMm84WXNuVjg9", nodeConfigManager.getMyAuth());
 137         Assert.assertEquals(0.05, nodeConfigManager.getFreeDiskStart(), 0.0);
 138         Assert.assertEquals(0.2, nodeConfigManager.getFreeDiskStop(), 0.0);
 139         Assert.assertEquals("org.onap.dmaap-dr.feed|legacy|publish", nodeConfigManager.getPermission("legacy"));
 140     }
 141
 142     /**
 143      * {@link URLStreamHandler} that allows us to control the {@link URLConnection URLConnections} that are returned
 144      * by {@link URL URLs} in the code under test.
 145      */
 146     public static class HttpUrlStreamHandler extends URLStreamHandler {
 147
 148         private Map<URL, URLConnection> connections = new HashMap();
 149
 150         @Override
 151         protected URLConnection openConnection(URL url) {
 152             return connections.get(url);
 153         }
 154
 155         void resetConnections() {
 156             connections = new HashMap();
 157         }
 158
 159         HttpUrlStreamHandler addConnection(URL url, URLConnection urlConnection) {
 160             connections.put(url, urlConnection);
 161             return this;
 162         }
 163     }
 164 }