src/test/java/org/onap/aai/datarouter/policy/SpikeAutosuggestProcessorTest.java

   1 /**
   2  * ============LICENSE_START=======================================================
   3  * org.onap.aai
   4  * ================================================================================
   5  * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
   6  * Copyright © 2017-2018 Amdocs
   7  * ================================================================================
   8  * Licensed under the Apache License, Version 2.0 (the "License");
   9  * you may not use this file except in compliance with the License.
  10  * You may obtain a copy of the License at
  11  *
  12  *       http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing, software
  15  * distributed under the License is distributed on an "AS IS" BASIS,
  16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  17  * See the License for the specific language governing permissions and
  18  * limitations under the License.
  19  * ============LICENSE_END=========================================================
  20  */
  21 package org.onap.aai.datarouter.policy;
  22
  23 import static org.junit.Assert.assertNotNull;
  24 import static org.junit.Assert.assertNull;
  25 import static org.mockito.Matchers.anyObject;
  26 import static org.mockito.Matchers.anyString;
  27
  28 import java.io.File;
  29 import java.io.FileInputStream;
  30
  31 import org.apache.camel.Exchange;
  32 import org.apache.camel.Message;
  33 import org.apache.commons.io.IOUtils;
  34 import org.junit.Before;
  35 import org.junit.Test;
  36 import org.junit.runner.RunWith;
  37 import org.onap.aai.datarouter.util.NodeUtils;
  38 import org.onap.aai.setup.SchemaLocationsBean;
  39 import org.onap.aai.setup.SchemaVersions;
  40 import org.powermock.api.mockito.PowerMockito;
  41 import org.springframework.beans.factory.annotation.Autowired;
  42 import org.springframework.test.context.ContextConfiguration;
  43 import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
  44
  45 @RunWith(SpringJUnit4ClassRunner.class)
  46 @ContextConfiguration("file:src/test/resources/spring-beans/data-router-oxm.xml")
  47 public class SpikeAutosuggestProcessorTest {
  48   private SpikeEventPolicyConfig eventPolicyConfig;
  49   private SpikeAutosuggestIndexProcessor policy;
  50   private InMemorySearchDatastore searchDb;
  51
  52   @Autowired
  53   private SchemaVersions schemaVersions;
  54   @Autowired
  55   private SchemaLocationsBean schemaLocationsBean;
  56
  57   @Before
  58   public void init() throws Exception {
  59
  60     eventPolicyConfig = new SpikeEventPolicyConfig();
  61     eventPolicyConfig.setSearchKeystorePwd("password");
  62     eventPolicyConfig.setSourceDomain("JUNIT");
  63
  64     eventPolicyConfig.setSchemaVersions(schemaVersions);
  65     eventPolicyConfig.setSchemaLocationsBean(schemaLocationsBean);
  66
  67     searchDb = new InMemorySearchDatastore();
  68     policy = new SpikeAutosuggestProcessorStubbed(eventPolicyConfig).withSearchDb(searchDb);
  69
  70   }
  71
  72   @Test
  73   public void testProcess_success() throws Exception {
  74
  75     String genericVnfEventJsonTemplate = IOUtils.toString(
  76         new FileInputStream(new File("src/test/resources/generic-vnf-spike-event.json")), "UTF-8");
  77
  78     policy.process(getExchangeEvent(genericVnfEventJsonTemplate, "update-notification", "CREATE", "vserver123"));
  79
  80     assertNotNull(
  81         searchDb.get(NodeUtils.generateUniqueShaDigest("junk and Running VNFs")));
  82     assertNotNull(
  83         searchDb.get(NodeUtils.generateUniqueShaDigest("junk VNFs")));
  84     assertNotNull(
  85         searchDb.get(NodeUtils.generateUniqueShaDigest("Running VNFs")));
  86
  87   }
  88
  89   /*
  90    * Failure test cases
  91    * - no searchable attribute for type
  92    */
  93
  94   @Test
  95   public void testProcess_failure_unknownOxmEntityType() throws Exception {
  96
  97     String pserverEventJsonTemplate = IOUtils.toString(
  98         new FileInputStream(new File("src/test/resources/optical-router-spike-event.json")), "UTF-8");
  99
 100     policy.process(getExchangeEvent(pserverEventJsonTemplate, "update-notification", "CREATE", "optronic123"));
 101
 102     assertNull(
 103         searchDb.get(NodeUtils.generateUniqueShaDigest("optical-router/optronic123")));
 104   }
 105
 106   @Test
 107   public void testProcess_failure_missingMandatoryFieldsFromBodyObject() throws Exception {
 108
 109     String pserverEventJsonTemplate = IOUtils.toString(
 110         new FileInputStream(new File("src/test/resources/pserver-missing-mandtory-field-spike-event.json")), "UTF-8");
 111
 112     policy.process(getExchangeEvent(pserverEventJsonTemplate, "update-notification", "CREATE", "pserver123"));
 113
 114     assertNull(
 115         searchDb.get(NodeUtils.generateUniqueShaDigest("pserver/pserver123")));
 116   }
 117
 118   @Test
 119   public void testProcess_failure_missingMandatoryVertexProperties() throws Exception {
 120
 121     String pserverEventJsonTemplate = IOUtils.toString(
 122         new FileInputStream(new File("src/test/resources/pserver-missing-primary-key-spike-event.json")), "UTF-8");
 123
 124     policy.process(getExchangeEvent(pserverEventJsonTemplate, "update-notification", "CREATE", "pserver123"));
 125
 126     assertNull(
 127         searchDb.get(NodeUtils.generateUniqueShaDigest("pserver/pserver123")));
 128   }
 129
 130   @Test
 131   public void testProcess_failure_noSuggestibleAttributesForEntityType() throws Exception {
 132
 133     String pserverEventJsonTemplate = IOUtils.toString(
 134         new FileInputStream(new File("src/test/resources/vserver-spike-event.json")), "UTF-8");
 135
 136     policy.process(getExchangeEvent(pserverEventJsonTemplate, "update-notification", "CREATE", "vserver123"));
 137
 138     assertNull(
 139         searchDb.get(NodeUtils.generateUniqueShaDigest("vserver/vserver123")));
 140   }
 141
 142   private Exchange getExchangeEvent(String payloadTemplate, String eventType, String operationType,
 143       String entityKey) {
 144     Object obj = payloadTemplate.replace("$EVENT_TYPE", eventType)
 145         .replace("$OPERATION_TYPE", operationType).replace("$ENTITY_KEY", entityKey);
 146
 147     Exchange exchange = PowerMockito.mock(Exchange.class);
 148     Message inMessage = PowerMockito.mock(Message.class);
 149     Message outMessage = PowerMockito.mock(Message.class);
 150     PowerMockito.when(exchange.getIn()).thenReturn(inMessage);
 151     PowerMockito.when(inMessage.getBody()).thenReturn(obj);
 152
 153     PowerMockito.when(exchange.getOut()).thenReturn(outMessage);
 154     PowerMockito.doNothing().when(outMessage).setBody(anyObject());
 155     PowerMockito.doNothing().when(outMessage).setHeader(anyString(), anyObject());
 156
 157     return exchange;
 158
 159   }
 160
 161
 162 }