core/core-engine/src/test/java/org/onap/policy/apex/core/engine/executor/DummyTaskSelectExecutor.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2018 Ericsson. All rights reserved.
   4  *  Modifications Copyright (C) 2019 Nordix Foundation.
   5  * ================================================================================
   6  * Licensed under the Apache License, Version 2.0 (the "License");
   7  * you may not use this file except in compliance with the License.
   8  * You may obtain a copy of the License at
   9  *
  10  *      http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  *
  18  * SPDX-License-Identifier: Apache-2.0
  19  * ============LICENSE_END=========================================================
  20  */
  21
  22 package org.onap.policy.apex.core.engine.executor;
  23
  24 import java.util.Properties;
  25
  26 import org.onap.policy.apex.context.ContextException;
  27 import org.onap.policy.apex.core.engine.event.EnEvent;
  28 import org.onap.policy.apex.core.engine.executor.exception.StateMachineException;
  29 import org.onap.policy.apex.model.basicmodel.concepts.AxArtifactKey;
  30
  31 /**
  32  * Dummy task selection executor for testing.
  33  */
  34 public class DummyTaskSelectExecutor extends TaskSelectExecutor {
  35     private boolean override;
  36
  37     private static int taskNo;
  38
  39     public DummyTaskSelectExecutor() {
  40         this(false);
  41     }
  42
  43     public DummyTaskSelectExecutor(final boolean override) {
  44         this.override = override;
  45     }
  46
  47     @Override
  48     public void prepare() throws StateMachineException {
  49         if (!override) {
  50             super.prepare();
  51         }
  52     }
  53
  54     /**
  55      * {@inheritDoc}.
  56      */
  57     @Override
  58     public AxArtifactKey execute(final long executionId, final Properties executionProperties,
  59             final EnEvent newIncomingEvent) throws StateMachineException, ContextException {
  60         if (!override) {
  61             return super.execute(executionId, executionProperties, newIncomingEvent);
  62         }
  63
  64         return new AxArtifactKey("task" + (taskNo++) + ":0.0.1");
  65     }
  66
  67     public void setTaskNo(int incomingTaskNo) {
  68         taskNo = incomingTaskNo;
  69     }
  70
  71     /**
  72      * {@inheritDoc}.
  73      */
  74     @Override
  75     public void cleanUp() throws StateMachineException {
  76         if (!override) {
  77             super.cleanUp();
  78         }
  79     }
  80 }