models-interactions/model-actors/actor.so/src/test/java/org/onap/policy/controlloop/actor/so/BasicSoOperation.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP
   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
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  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=========================================================
  19  */
  20
  21 package org.onap.policy.controlloop.actor.so;
  22
  23 import static org.mockito.ArgumentMatchers.any;
  24 import static org.mockito.Mockito.when;
  25
  26 import java.util.Collections;
  27 import java.util.HashMap;
  28 import java.util.LinkedList;
  29 import java.util.List;
  30 import java.util.Map;
  31 import org.mockito.Mock;
  32 import org.onap.policy.aai.AaiCqResponse;
  33 import org.onap.policy.controlloop.actor.test.BasicHttpOperation;
  34 import org.onap.policy.controlloop.actorserviceprovider.Util;
  35 import org.onap.policy.controlloop.policy.Target;
  36 import org.onap.policy.so.SoRequest;
  37 import org.onap.policy.so.SoRequestParameters;
  38 import org.onap.policy.so.SoRequestReferences;
  39 import org.onap.policy.so.SoRequestStatus;
  40 import org.onap.policy.so.SoResponse;
  41
  42 /**
  43  * Superclass for various operator tests.
  44  */
  45 public abstract class BasicSoOperation extends BasicHttpOperation<SoRequest> {
  46     protected static final String[] IGNORE_FIELDS = {"RequestID", "subRequestID", "seconds", "nanos"};
  47
  48     public static final String MODEL_CUSTOM_ID = "my-model-customization-id";
  49     public static final String MODEL_INVAR_ID = "my-model-invariant-id";
  50     public static final String MODEL_NAME = "my-model-name";
  51     public static final String MODEL_VERSION = "my-model-version";
  52     public static final String MODEL_VERS_ID = "my-model-version-id";
  53     public static final String SUBSCRIPTION_SVC_TYPE = "my-subscription-service-type";
  54     public static final String PATH_GET = "my-path-get/";
  55     public static final int MAX_GETS = 3;
  56     public static final int WAIT_SEC_GETS = 20;
  57     public static final Integer VF_COUNT = 10;
  58
  59     @Mock
  60     protected SoConfig config;
  61
  62     protected Target target;
  63     protected SoResponse response;
  64
  65     /**
  66      * Constructs the object using a default actor and operation name.
  67      */
  68     public BasicSoOperation() {
  69         super();
  70     }
  71
  72     /**
  73      * Constructs the object.
  74      *
  75      * @param actor actor name
  76      * @param operation operation name
  77      */
  78     public BasicSoOperation(String actor, String operation) {
  79         super(actor, operation);
  80     }
  81
  82     /**
  83      * Initializes mocks and sets up.
  84      */
  85     public void setUp() throws Exception {
  86         super.setUpBasic();
  87
  88         response = new SoResponse();
  89
  90         SoRequest request = new SoRequest();
  91         response.setRequest(request);
  92
  93         SoRequestStatus status = new SoRequestStatus();
  94         request.setRequestStatus(status);
  95         status.setRequestState(SoOperation.COMPLETE);
  96
  97         SoRequestReferences ref = new SoRequestReferences();
  98         response.setRequestReferences(ref);
  99         ref.setRequestId(REQ_ID.toString());
 100
 101         when(rawResponse.getStatus()).thenReturn(200);
 102         when(rawResponse.readEntity(String.class)).thenReturn(coder.encode(response));
 103
 104         initConfig();
 105     }
 106
 107     @Override
 108     protected void initConfig() {
 109         super.initConfig();
 110         when(config.getClient()).thenReturn(client);
 111         when(config.getMaxGets()).thenReturn(MAX_GETS);
 112         when(config.getPathGet()).thenReturn(PATH_GET);
 113         when(config.getWaitSecGet()).thenReturn(WAIT_SEC_GETS);
 114     }
 115
 116     @Override
 117     protected void makeContext() {
 118         super.makeContext();
 119
 120         target = new Target();
 121         target.setModelCustomizationId(MODEL_CUSTOM_ID);
 122         target.setModelInvariantId(MODEL_INVAR_ID);
 123         target.setModelName(MODEL_NAME);
 124         target.setModelVersion(MODEL_VERSION);
 125         target.setModelVersionId(MODEL_VERS_ID);
 126
 127         params = params.toBuilder().target(target).build();
 128     }
 129
 130     @Override
 131     protected Map<String, Object> makePayload() {
 132         Map<String, Object> payload = new HashMap<>();
 133
 134         // request parameters
 135         SoRequestParameters reqParams = new SoRequestParameters();
 136         reqParams.setSubscriptionServiceType(SUBSCRIPTION_SVC_TYPE);
 137         payload.put(SoOperation.REQ_PARAM_NM, Util.translate("", reqParams, String.class));
 138
 139         // config parameters
 140         List<Map<String, String>> config = new LinkedList<>();
 141         config.add(Collections.emptyMap());
 142         payload.put(SoOperation.CONFIG_PARAM_NM, Util.translate("", config, String.class));
 143
 144         return payload;
 145     }
 146
 147     protected AaiCqResponse makeCqResponse() {
 148         when(cqResponse.getVfModuleCount(any(), any(), any())).thenReturn(VF_COUNT);
 149         return cqResponse;
 150     }
 151 }