aboutsummaryrefslogtreecommitdiffstats
path: root/src/test/java/org/onap/aai/datarouter/policy/SpikeAggregateGenericVnfProcessorTest.java
blob: 03970bd64565e731af8682bbf7f87c494408affa (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
/**
 * ============LICENSE_START=======================================================
 * org.onap.aai
 * ================================================================================
 * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
 * Copyright © 2017-2018 Amdocs
 * ================================================================================
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * ============LICENSE_END=========================================================
 */
package org.onap.aai.datarouter.policy;

import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import static org.mockito.Matchers.anyObject;
import static org.mockito.Matchers.anyString;

import java.io.File;
import java.io.FileInputStream;

import org.apache.camel.Exchange;
import org.apache.camel.Message;
import org.apache.commons.io.IOUtils;
import org.junit.Before;
import org.junit.Test;
import org.onap.aai.datarouter.util.NodeUtils;
import org.powermock.api.mockito.PowerMockito;

public class SpikeAggregateGenericVnfProcessorTest {
  private SpikeAggregateGenericVnfProcessor policy;
  private InMemorySearchDatastore searchDb;

  @Before
  public void init() throws Exception {
    SpikeEventPolicyConfig config = PowerMockito.mock(SpikeEventPolicyConfig.class);
    PowerMockito.when(config.getSearchKeystorePwd()).thenReturn("password");
    PowerMockito.when(config.getSourceDomain()).thenReturn("JUNIT");

    searchDb = new InMemorySearchDatastore();
    policy = new SpikeAggregateGenericVnfProcessorStubbed(config).withSearchDb(searchDb);
  }

  @Test
  public void testProcess_success() throws Exception {

    String genericVnfEventJsonTemplate = IOUtils.toString(
        new FileInputStream(new File("src/test/resources/generic-vnf-spike-event.json")), "UTF-8");

    policy.process(
        getExchangeEvent(genericVnfEventJsonTemplate, "update-notification", "CREATE", "gvnf123"));

    assertNotNull(searchDb.get(NodeUtils.generateUniqueShaDigest("generic-vnf/gvnf123")));

    policy.process(
        getExchangeEvent(genericVnfEventJsonTemplate, "update-notification", "DELETE", "gvnf123"));

    assertNull(searchDb.get(NodeUtils.generateUniqueShaDigest("generic-vnf/gvnf123")));

    
  }
  /*
   * Failure test cases - no searchable attribute for type
   */

  @Test
  public void testProcess_failure_unknownOxmEntityType() throws Exception {

    String pserverEventJsonTemplate = IOUtils.toString(
        new FileInputStream(new File("src/test/resources/optical-router-spike-event.json")),
        "UTF-8");

    policy.process(
        getExchangeEvent(pserverEventJsonTemplate, "update-notification", "CREATE", "optronic123"));

    assertNull(searchDb.get(NodeUtils.generateUniqueShaDigest("optical-router/optronic123")));
  }

  @Test
  public void testProcess_failure_missingMandatoryFieldsFromBodyObject() throws Exception {

    String pserverEventJsonTemplate = IOUtils.toString(
        new FileInputStream(
            new File("src/test/resources/pserver-missing-mandtory-field-spike-event.json")),
        "UTF-8");

    policy.process(
        getExchangeEvent(pserverEventJsonTemplate, "update-notification", "CREATE", "pserver123"));

    assertNull(searchDb.get(NodeUtils.generateUniqueShaDigest("pserver/pserver123")));
  }

  private Exchange getExchangeEvent(String payloadTemplate, String eventType, String operationType,
      String entityKey) {
    Object obj = payloadTemplate.replace("$EVENT_TYPE", eventType)
        .replace("$OPERATION_TYPE", operationType).replace("$ENTITY_KEY", entityKey);

    Exchange exchange = PowerMockito.mock(Exchange.class);
    Message inMessage = PowerMockito.mock(Message.class);
    Message outMessage = PowerMockito.mock(Message.class);
    PowerMockito.when(exchange.getIn()).thenReturn(inMessage);
    PowerMockito.when(inMessage.getBody()).thenReturn(obj);

    PowerMockito.when(exchange.getOut()).thenReturn(outMessage);
    PowerMockito.doNothing().when(outMessage).setBody(anyObject());
    PowerMockito.doNothing().when(outMessage).setHeader(anyString(), anyObject());

    return exchange;

  }

}