src/test/java/org/onap/aai/datarouter/policy/EntityEventPolicyStubbed.java

   1 /**
   2  * ============LICENSE_START=======================================================
   3  * org.onap.aai
   4  * ================================================================================
   5  * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
   6  * Copyright © 2017-2018 Amdocs
   7  * ================================================================================
   8  * Licensed under the Apache License, Version 2.0 (the "License");
   9  * you may not use this file except in compliance with the License.
  10  * You may obtain a copy of the License at
  11  *
  12  *       http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing, software
  15  * distributed under the License is distributed on an "AS IS" BASIS,
  16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  17  * See the License for the specific language governing permissions and
  18  * limitations under the License.
  19  * ============LICENSE_END=========================================================
  20  */
  21 package org.onap.aai.datarouter.policy;
  22
  23 import java.io.FileNotFoundException;
  24
  25 import org.onap.aai.datarouter.entity.DocumentStoreDataEntity;
  26 import org.onap.aai.datarouter.policy.EntityEventPolicy;
  27 import org.onap.aai.datarouter.policy.EntityEventPolicyConfig;
  28
  29 public class EntityEventPolicyStubbed extends EntityEventPolicy {
  30
  31   private InMemorySearchDatastore searchDb;
  32
  33   public EntityEventPolicyStubbed(EntityEventPolicyConfig config) throws FileNotFoundException {
  34     super(config);
  35
  36   }
  37
  38   public InMemorySearchDatastore getSearchDb() {
  39     return searchDb;
  40   }
  41
  42   public EntityEventPolicyStubbed withSearchDb(InMemorySearchDatastore searchDb) {
  43     this.searchDb = searchDb;
  44     return this;
  45   }
  46
  47   protected void handleSearchServiceOperation(DocumentStoreDataEntity eventEntity, String action,
  48       String index) {
  49     // Stub out the actual call to Search Data service and instead store/update documents in memory
  50     try {
  51       switch (action.toLowerCase()) {
  52         case "create":
  53           searchDb.put(eventEntity.getId(), eventEntity.getAsJson()); // they are executed if
  54                                                                       // variable == c1
  55           break;
  56         case "update":
  57           searchDb.put(eventEntity.getId(), eventEntity.getAsJson()); // they are executed if
  58                                                                       // variable == c1
  59           break;
  60         case "delete":
  61           searchDb.remove(eventEntity.getId()); // they are executed if variable == c1
  62           break;
  63         default:
  64           break;
  65       }
  66     } catch (Exception ex) {
  67     }
  68   }
  69 }