datarouter-node/src/test/java/org/onap/dmaap/datarouter/node/DeliveryTest.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 package org.onap.dmaap.datarouter.node;
  24
  25 import static org.mockito.Mockito.mock;
  26 import static org.mockito.Mockito.times;
  27 import static org.mockito.Mockito.verify;
  28
  29 import java.io.File;
  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;
  44
  45 @RunWith(PowerMockRunner.class)
  46 @SuppressStaticInitializationFor("org.onap.dmaap.datarouter.node.NodeConfigManager")
  47 public class DeliveryTest {
  48
  49     @Mock
  50     private DeliveryQueue deliveryQueue;
  51     @Mock
  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");
  60
  61     @Before
  62     public void setUp() throws IOException {
  63         nDir.mkdirs();
  64         sDir.mkdirs();
  65         newNDir.mkdirs();
  66         newNFile.createNewFile();
  67         newSDir.mkdirs();
  68         newSpoolFile.createNewFile();
  69         spoolFileMeta.createNewFile();
  70         config = mockNodeConfigManager();
  71     }
  72
  73     @Test
  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();
  81     }
  82
  83     @Test
  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");
  91     }
  92
  93     @Test
  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));
  99     }
 100
 101     @Test
 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));
 107     }
 108
 109     @After
 110     public void tearDown() {
 111         newSpoolFile.delete();
 112         spoolFileMeta.delete();
 113         newNFile.delete();
 114         newNDir.delete();
 115         newSDir.delete();
 116         new File("tmp/s/0").delete();
 117         nDir.delete();
 118         sDir.delete();
 119         File tmpDir = new File("tmp");
 120         tmpDir.delete();
 121     }
 122
 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");
 131         return config;
 132     }
 133
 134     private DestInfo[] createDestInfoObjects() {
 135         DestInfo[] destInfos = new DestInfo[1];
 136         DestInfo destInfo = new DestInfoBuilder().setName("node.datarouternew.com").setSpool("tmp/s/0/1")
 137                 .setSubid("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;
 143         return destInfos;
 144     }
 145 }