2 * ============LICENSE_START=======================================================
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
12 * http://www.apache.org/licenses/LICENSE-2.0
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=========================================================
21 package org.onap.aai.datarouter.policy;
23 import static org.junit.Assert.*;
24 import static org.mockito.Matchers.anyObject;
25 import static org.mockito.Matchers.anyString;
28 import java.io.FileInputStream;
30 import org.apache.camel.Exchange;
31 import org.apache.camel.Message;
32 import org.apache.commons.io.IOUtils;
33 import org.junit.Before;
34 import org.junit.Test;
35 import org.onap.aai.datarouter.policy.EntityEventPolicy;
36 import org.onap.aai.datarouter.policy.EntityEventPolicyConfig;
37 import org.onap.aai.datarouter.util.NodeUtils;
38 import org.onap.aai.datarouter.util.SearchServiceAgent;
39 import org.powermock.api.mockito.PowerMockito;
43 public class EntityEventPolicyTest {
44 private EntityEventPolicy policy;
45 private String eventJson;
46 private InMemorySearchDatastore searchDb;
48 @SuppressWarnings("unchecked")
50 public void init() throws Exception {
51 EntityEventPolicyConfig config = PowerMockito.mock(EntityEventPolicyConfig.class);
52 PowerMockito.when(config.getSearchKeystorePwd()).thenReturn("password");
53 PowerMockito.when(config.getSourceDomain()).thenReturn("JUNIT");
55 searchDb = new InMemorySearchDatastore();
56 policy = new EntityEventPolicyStubbed(config).withSearchDb(searchDb);
58 FileInputStream event = new FileInputStream( new File("src/test/resources/aai_event.json"));
59 eventJson = IOUtils.toString(event, "UTF-8");
64 public void testProcess() throws Exception {
65 policy.process(getExchangeEvent("event1","create"));
66 policy.process(getExchangeEvent("event2","create"));
68 assertNotNull(searchDb.get(NodeUtils.generateUniqueShaDigest("event1")));
69 assertNotNull(searchDb.get(NodeUtils.generateUniqueShaDigest("event2")));
71 policy.process(getExchangeEvent("event1","update"));
72 policy.process(getExchangeEvent("event2","update"));
73 assertNotNull(searchDb.get(NodeUtils.generateUniqueShaDigest("event1")));
74 assertNotNull(searchDb.get(NodeUtils.generateUniqueShaDigest("event2")));
76 policy.process(getExchangeEvent("event2","delete"));
77 assertNull(searchDb.get(NodeUtils.generateUniqueShaDigest("event2")));
80 private Exchange getExchangeEvent(String link,String action){
81 Object obj = eventJson.replace("$LINK",link ).replace("$ACTION",action) ;
82 Exchange exchange = PowerMockito.mock(Exchange.class);
83 Message inMessage = PowerMockito.mock(Message.class);
84 Message outMessage = PowerMockito.mock(Message.class);
85 PowerMockito.when(exchange.getIn()).thenReturn(inMessage);
86 PowerMockito.when(inMessage.getBody()).thenReturn(obj);
88 PowerMockito.when(exchange.getOut()).thenReturn(outMessage);
89 PowerMockito.doNothing().when(outMessage).setBody(anyObject());
90 PowerMockito.doNothing().when(outMessage).setHeader(anyString(), anyObject());