2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2020 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
11 * http://www.apache.org/licenses/LICENSE-2.0
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=========================================================
21 package org.onap.policy.drools.apps.controller.usecases;
23 import static org.assertj.core.api.Assertions.assertThat;
24 import static org.assertj.core.api.Assertions.assertThatIllegalStateException;
25 import static org.junit.Assert.assertEquals;
26 import static org.junit.Assert.assertNull;
27 import static org.junit.Assert.assertSame;
28 import static org.mockito.ArgumentMatchers.anyString;
29 import static org.mockito.Mockito.when;
31 import java.util.List;
33 import java.util.concurrent.CompletableFuture;
34 import org.junit.Before;
35 import org.junit.Test;
36 import org.mockito.Mock;
37 import org.mockito.MockitoAnnotations;
38 import org.onap.aai.domain.yang.GenericVnf;
39 import org.onap.policy.controlloop.VirtualControlLoopEvent;
40 import org.onap.policy.controlloop.actorserviceprovider.OperationOutcome;
41 import org.onap.policy.controlloop.actorserviceprovider.OperationProperties;
42 import org.onap.policy.controlloop.actorserviceprovider.parameters.ControlLoopOperationParams;
43 import org.onap.policy.controlloop.eventmanager.StepContext;
44 import org.onap.policy.controlloop.policy.Target;
45 import org.onap.policy.controlloop.policy.TargetType;
47 public class LockOperation2Test {
48 private static final String MY_PNF = "my-pnf";
49 private static final String MY_VNF = "my-vnf";
50 private static final String MY_ACTOR = "my-actor";
51 private static final String MY_OPERATION = "my-operation";
54 private StepContext stepContext;
56 private ControlLoopOperationParams params;
58 private VirtualControlLoopEvent event;
59 private Target target;
60 private CompletableFuture<OperationOutcome> future;
61 private GenericVnf vnf;
62 private LockOperation2 oper;
69 MockitoAnnotations.initMocks(this);
71 event = new VirtualControlLoopEvent();
72 event.setTarget("pnf.pnf-name");
73 event.setAai(Map.of("pnf.pnf-name", MY_PNF));
75 target = new Target();
76 target.setType(TargetType.PNF);
78 future = new CompletableFuture<>();
80 vnf = new GenericVnf();
83 when(stepContext.requestLock(anyString())).thenReturn(future);
85 when(params.getTarget()).thenReturn(target);
86 when(params.getActor()).thenReturn(MY_ACTOR);
87 when(params.getOperation()).thenReturn(MY_OPERATION);
89 oper = new LockOperation2(stepContext, params);
93 public void testGetPropertyNames() {
94 assertThat(oper.getPropertyNames()).isEqualTo(List.of(OperationProperties.AAI_TARGET_ENTITY));
98 public void testStart() {
100 assertThatIllegalStateException().isThrownBy(() -> oper.start())
101 .withMessage("target lock entity has not been determined yet");
103 oper.setProperty(OperationProperties.AAI_TARGET_ENTITY, "some-target");
104 assertSame(future, oper.start());
108 public void testSetProperty() {
109 oper.setProperty("unknown-property", "some data");
110 assertNull(oper.getTargetEntity());
112 oper.setProperty(OperationProperties.AAI_TARGET_ENTITY, "other data");
113 assertEquals("other data", oper.getTargetEntity());
117 public void testGetActorName_testGetName() {
118 assertEquals(MY_ACTOR, oper.getActorName());
119 assertEquals(MY_OPERATION, oper.getName());