model/basic-model/src/test/java/org/onap/policy/apex/model/basicmodel/handling/DummyAxModelWithReferences.java

   1 /*
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2016-2018 Ericsson. All rights reserved.
   4  * ================================================================================
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *      http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  *
  17  * SPDX-License-Identifier: Apache-2.0
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.onap.policy.apex.model.basicmodel.handling;
  22
  23 import java.util.ArrayList;
  24 import java.util.List;
  25 import org.onap.policy.apex.model.basicmodel.concepts.AxArtifactKey;
  26 import org.onap.policy.apex.model.basicmodel.concepts.AxKey;
  27 import org.onap.policy.apex.model.basicmodel.concepts.AxModel;
  28 import org.onap.policy.apex.model.basicmodel.concepts.AxReferenceKey;
  29
  30 public class DummyAxModelWithReferences extends AxModel {
  31     private static final long serialVersionUID = -8194956638511120008L;
  32
  33     private List<AxKey> extrakeyList = new ArrayList<>();
  34
  35     public DummyAxModelWithReferences(final AxArtifactKey key) {
  36         super(key);
  37     }
  38
  39     @Override
  40     public List<AxKey> getKeys() {
  41         List<AxKey> keys = super.getKeys();
  42         keys.addAll(extrakeyList);
  43
  44         return keys;
  45     }
  46
  47     public List<AxKey> getExtrakeyList() {
  48         return extrakeyList;
  49     }
  50
  51     /**
  52      * Set the reference key list.
  53      */
  54     public void setReferenceKeyList() {
  55         List<AxKey> keys = super.getKeys();
  56
  57         for (AxKey key: keys) {
  58             AxArtifactKey akey = (AxArtifactKey) key;
  59             AxReferenceKey keyRef = new AxReferenceKey(akey, akey.getName());
  60             extrakeyList.add(keyRef);
  61         }
  62     }
  63
  64     public void addKey(final AxKey akey) {
  65         extrakeyList.add(akey);
  66     }
  67
  68     public void removeKey(final AxKey akey) {
  69         extrakeyList.remove(akey);
  70     }
  71 }