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
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.onap.dmaap.datarouter.node;
23 import static org.mockito.BDDMockito.given;
24 import static org.mockito.Mockito.mock;
25 import static org.mockito.Mockito.when;
27 import java.io.ByteArrayInputStream;
29 import java.io.IOException;
30 import java.io.InputStream;
31 import java.net.InetAddress;
33 import java.net.URLConnection;
34 import java.net.URLStreamHandler;
35 import java.net.URLStreamHandlerFactory;
36 import java.nio.file.Files;
37 import java.nio.file.Paths;
38 import java.util.HashMap;
40 import org.apache.commons.io.FileUtils;
41 import org.junit.After;
42 import org.junit.AfterClass;
43 import org.junit.Assert;
44 import org.junit.Before;
45 import org.junit.BeforeClass;
46 import org.junit.Test;
47 import org.mockito.Mock;
48 import org.powermock.core.classloader.annotations.PowerMockIgnore;
49 import org.powermock.core.classloader.annotations.PrepareForTest;
51 @PowerMockIgnore({"javax.net.ssl.*", "javax.security.auth.x500.X500Principal", "javax.crypto.*"})
52 @PrepareForTest({InetAddress.class})
53 public class NodeConfigManagerTest {
55 private static HttpUrlStreamHandler httpUrlStreamHandler;
58 InputStream inputStream;
61 NodeConfig nodeConfig;
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);
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);
84 public void resetHandler() {
85 httpUrlStreamHandler.resetConnections();
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"));
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"));
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.
146 public static class HttpUrlStreamHandler extends URLStreamHandler {
148 private Map<URL, URLConnection> connections = new HashMap();
151 protected URLConnection openConnection(URL url) {
152 return connections.get(url);
155 void resetConnections() {
156 connections = new HashMap();
159 HttpUrlStreamHandler addConnection(URL url, URLConnection urlConnection) {
160 connections.put(url, urlConnection);