1 /*******************************************************************************
2 * ============LICENSE_START==================================================
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
11 * * http://www.apache.org/licenses/LICENSE-2.0
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====================================================
20 * * ECOMP is a trademark and service mark of AT&T Intellectual Property.
22 ******************************************************************************/
23 package org.onap.dmaap.datarouter.node;
25 import static org.mockito.Mockito.mock;
26 import static org.mockito.Mockito.times;
27 import static org.mockito.Mockito.verify;
30 import java.io.IOException;
31 import java.util.HashMap;
32 import java.util.Hashtable;
33 import org.apache.commons.lang3.reflect.FieldUtils;
34 import org.junit.After;
35 import org.junit.Assert;
36 import org.junit.Before;
37 import org.junit.Test;
38 import org.junit.runner.RunWith;
39 import org.mockito.Mock;
40 import org.onap.dmaap.datarouter.node.Delivery.DelItem;
41 import org.powermock.api.mockito.PowerMockito;
42 import org.powermock.core.classloader.annotations.SuppressStaticInitializationFor;
43 import org.powermock.modules.junit4.PowerMockRunner;
45 @RunWith(PowerMockRunner.class)
46 @SuppressStaticInitializationFor("org.onap.dmaap.datarouter.node.NodeConfigManager")
47 public class DeliveryTest {
50 private DeliveryQueue deliveryQueue;
52 private NodeConfigManager config;
53 private File nDir = new File("tmp/n");
54 private File newNDir = new File("tmp/n/0");
55 private File newNFile = new File("tmp/n/0/testN.txt");
56 private File sDir = new File("tmp/s");
57 private File newSDir = new File("tmp/s/0/1");
58 private File newSpoolFile = new File("tmp/s/0/1/123456789.dmaap-dr-node");
59 private File spoolFileMeta = new File("tmp/s/0/1/123456789.dmaap-dr-node.M");
62 public void setUp() throws IOException {
66 newNFile.createNewFile();
68 newSpoolFile.createNewFile();
69 spoolFileMeta.createNewFile();
70 config = mockNodeConfigManager();
74 public void Validate_Reset_Queue_Calls_Reset_Queue_On_Delivery_Queue_Object() throws IllegalAccessException {
75 Delivery delivery = new Delivery(config);
76 HashMap<String, DeliveryQueue> dqs = new HashMap<>();
77 dqs.put("tmp/s/0/1", deliveryQueue);
78 FieldUtils.writeDeclaredField(delivery, "dqs", dqs, true);
79 delivery.resetQueue("tmp/s/0/1");
80 verify(deliveryQueue, times(1)).resetQueue();
84 public void Validate_Mark_Success_Calls_Mark_Success_On_Delivery_Queue_Object() throws IllegalAccessException {
85 Delivery delivery = new Delivery(config);
86 HashMap<String, DeliveryQueue> dqs = new HashMap<>();
87 dqs.put("tmp/s/0/1", deliveryQueue);
88 FieldUtils.writeDeclaredField(delivery, "dqs", dqs, true);
89 delivery.markTaskSuccess("tmp/s/0/1", "123456789.dmaap-dr-node");
90 verify(deliveryQueue, times(1)).markTaskSuccess("123456789.dmaap-dr-node");
94 public void Validate_DelItem_With_Equal_Spool_And_PubId_Are_Equal() {
95 DelItem delItem1 = new DelItem("123456789.dmaap-dr-node", "tmp/s/0/1");
96 DelItem delItem2 = new DelItem("123456789.dmaap-dr-node", "tmp/s/0/1");
97 Assert.assertEquals(delItem1, delItem2);
98 Assert.assertEquals(0, delItem1.compareTo(delItem2));
102 public void Validate_DelItem_With_Unequal_Spool_And_PubId_Are_Not_Equal() {
103 DelItem delItem1 = new DelItem("123456789.dmaap-dr-node", "tmp/s/0/1");
104 DelItem delItem2 = new DelItem("000000000.dmaap-dr-node", "tmp/s/0/2");
105 Assert.assertNotEquals(delItem1, delItem2);
106 Assert.assertNotEquals(0, delItem1.compareTo(delItem2));
110 public void tearDown() {
111 newSpoolFile.delete();
112 spoolFileMeta.delete();
116 new File("tmp/s/0").delete();
119 File tmpDir = new File("tmp");
123 private NodeConfigManager mockNodeConfigManager() {
124 NodeConfigManager config = mock(NodeConfigManager.class);
125 PowerMockito.when(config.isConfigured()).thenReturn(true);
126 PowerMockito.when(config.getAllDests()).thenReturn(createDestInfoObjects());
127 PowerMockito.when(config.getFreeDiskStart()).thenReturn(0.9);
128 PowerMockito.when(config.getFreeDiskStop()).thenReturn(0.2);
129 PowerMockito.when(config.getDeliveryThreads()).thenReturn(0);
130 PowerMockito.when(config.getSpoolBase()).thenReturn("tmp");
134 private DestInfo[] createDestInfoObjects() {
135 DestInfo[] destInfos = new DestInfo[1];
136 DestInfo destInfo = new DestInfoBuilder().setName("node.datarouternew.com").setSpool("tmp/s/0/1")
138 .setLogdata("logs/").setUrl("/subs/1").setAuthuser("user1")
139 .setAuthentication("Basic dXNlcjE6cGFzc3dvcmQx")
140 .setMetaonly(false).setUse100(true).setPrivilegedSubscriber(false).setFollowRedirects(false)
141 .setDecompress(false).createDestInfo();
142 destInfos[0] = destInfo;