Update license date and text
[aai/data-router.git] / src / test / java / org / onap / aai / datarouter / policy / SpikeEntityEventPolicyTest.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.onap.aai.datarouter.util.NodeUtils;
37 import org.onap.aai.datarouter.util.SearchServiceAgent;
38 import org.powermock.api.mockito.PowerMockito;
39
40
41
42 public class SpikeEntityEventPolicyTest {
43   SpikeEntityEventPolicy policy;
44   String eventJson;
45
46   @SuppressWarnings("unchecked")
47   @Before
48   public void init() throws Exception {
49     SpikeEntityEventPolicyConfig config = PowerMockito.mock(SpikeEntityEventPolicyConfig.class);
50     PowerMockito.when(config.getSearchKeystorePwd()).thenReturn("password");
51     PowerMockito.when(config.getSourceDomain()).thenReturn("JUNIT");
52
53
54     SearchServiceAgent searchServiceAgent = PowerMockito.mock(SearchServiceAgent.class);
55     PowerMockito.whenNew(SearchServiceAgent.class).withAnyArguments()
56         .thenReturn(searchServiceAgent);
57
58
59     policy = new SpikeEntityEventPolicyStubbed(config);
60     FileInputStream event = new FileInputStream(new File("src/test/resources/spike_event.json"));
61     eventJson = IOUtils.toString(event, "UTF-8");  
62
63   }
64
65   @Test
66   public void testProcess_success() throws Exception {
67     policy.process(getExchangeEvent("12345", "create", "generic-vnf"));
68     policy.process(getExchangeEvent("23456", "create", "generic-vnf"));
69
70     assertNotNull(
71         InMemorySearchDatastore.get(NodeUtils.generateUniqueShaDigest("generic-vnf/12345")));
72     assertNotNull(
73         InMemorySearchDatastore.get(NodeUtils.generateUniqueShaDigest("generic-vnf/23456")));
74
75    
76     policy.process(getExchangeEvent("23456", "delete", "generic-vnf"));
77     assertNull(InMemorySearchDatastore.get(NodeUtils.generateUniqueShaDigest("23456")));
78   }
79   @Test
80   public void testProcess_fail() throws Exception {
81     policy.process(getExchangeEvent("12345", "create", "NotValid"));
82     assertNull(
83         InMemorySearchDatastore.get(NodeUtils.generateUniqueShaDigest("NotValid/12345")));
84     
85     policy.process(getExchangeEvent("", "create", "generic-vnf"));
86     assertNull(
87         InMemorySearchDatastore.get(NodeUtils.generateUniqueShaDigest("generic-vnf/")));
88
89   }
90
91   
92   private Exchange getExchangeEvent(String key, String action, String type) {
93     Object obj = eventJson.replace("$KEY", key).replace("$ACTION", action).replace("$TYPE", type);
94     Exchange exchange = PowerMockito.mock(Exchange.class);
95     Message inMessage = PowerMockito.mock(Message.class);
96     Message outMessage = PowerMockito.mock(Message.class);
97     PowerMockito.when(exchange.getIn()).thenReturn(inMessage); 
98     PowerMockito.when(inMessage.getBody()).thenReturn(obj);
99
100     PowerMockito.when(exchange.getOut()).thenReturn(outMessage);
101     PowerMockito.doNothing().when(outMessage).setBody(anyObject());
102     PowerMockito.doNothing().when(outMessage).setHeader(anyString(), anyObject());
103
104     return exchange;
105
106   }
107
108
109
110 }