search-data-service/src/test/java/org/onap/aai/sa/searchdbabstraction/elasticsearch/dao/ElasticSearchResultItemTest.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.sa.searchdbabstraction.elasticsearch.dao;
  22
  23 import org.junit.Assert;
  24 import org.junit.Test;
  25
  26 public class ElasticSearchResultItemTest {
  27
  28     @Test
  29     public void testAllMethods() {
  30         ElasticSearchShardStatus shardStatus = new ElasticSearchShardStatus();
  31         shardStatus.setTotal(10);
  32         shardStatus.setSuccessful(0);
  33         shardStatus.setFailed(1);
  34         Assert.assertEquals(shardStatus.getTotal(), 10);
  35         Assert.assertEquals(shardStatus.getSuccessful(), 0);
  36         Assert.assertEquals(shardStatus.getFailed(), 1);
  37
  38         ElasticSearchCause cause = new ElasticSearchCause();
  39         cause.setType("type-1");
  40         cause.setReason("reason-1");
  41         Assert.assertEquals(cause.getType(), "type-1");
  42         Assert.assertEquals(cause.getReason(), "reason-1");
  43
  44         ElasticSearchError error = new ElasticSearchError();
  45         error.setType("type-1");
  46         error.setReason("reason-1");
  47         error.setCausedBy(cause);
  48         Assert.assertEquals(error.getType(), "type-1");
  49         Assert.assertEquals(error.getReason(), "reason-1");
  50         Assert.assertNotNull(error.getCausedBy());
  51         error.setAdditionalProperties("name-1", "value-1");
  52         Assert.assertNotNull(error.getAdditionalProperties());
  53
  54         // Create Status
  55         ElasticSearchResultItem resultItem1 = new ElasticSearchResultItem();
  56         resultItem1.setCreate(getStatus(shardStatus, error));
  57         Assert.assertNotNull(resultItem1.getCreate());
  58         Assert.assertEquals(resultItem1.operationType(), "create");
  59         Assert.assertEquals(resultItem1.operationStatus(), resultItem1.getCreate());
  60         Assert.assertTrue(resultItem1.toString().contains("create"));
  61         Assert.assertNotNull(resultItem1.toJson());
  62
  63         // Index Status
  64         ElasticSearchResultItem resultItem2 = new ElasticSearchResultItem();
  65         resultItem2.setIndex(getStatus(shardStatus, error));
  66         Assert.assertNotNull(resultItem2.getIndex());
  67         Assert.assertEquals(resultItem2.operationType(), "update");
  68         Assert.assertEquals(resultItem2.operationStatus(), resultItem2.getIndex());
  69         Assert.assertTrue(resultItem2.toString().contains("index"));
  70         Assert.assertNotNull(resultItem2.toJson());
  71
  72         // Delete Status
  73         ElasticSearchResultItem resultItem3 = new ElasticSearchResultItem();
  74         resultItem3.setDelete(getStatus(shardStatus, error));
  75         Assert.assertNotNull(resultItem3.getDelete());
  76         Assert.assertEquals(resultItem3.operationType(), "delete");
  77         Assert.assertEquals(resultItem3.operationStatus(), resultItem3.getDelete());
  78         Assert.assertTrue(resultItem3.toString().contains("delete"));
  79         Assert.assertNotNull(resultItem3.toJson());
  80
  81         // Unknown Status
  82         ElasticSearchResultItem resultItem4 = new ElasticSearchResultItem();
  83         Assert.assertEquals(resultItem4.operationType(), "unknown");
  84         Assert.assertNull(resultItem4.operationStatus());
  85
  86         // ElasticSearchBulkOperationResult
  87         ElasticSearchResultItem[] resultItems = {resultItem1, resultItem2, resultItem3};
  88         ElasticSearchBulkOperationResult result = new ElasticSearchBulkOperationResult();
  89         result.setErrors(true);
  90         result.setTook(new Integer(10));
  91         result.setItems(resultItems);
  92         Assert.assertTrue(result.getErrors());
  93         Assert.assertEquals(result.getTook(), new Integer(10));
  94         Assert.assertNotNull(result.getItems());
  95         Assert.assertNotNull(result.toString());
  96     }
  97
  98     private ElasticSearchOperationStatus getStatus(ElasticSearchShardStatus shardStatus, ElasticSearchError error) {
  99         ElasticSearchOperationStatus status = new ElasticSearchOperationStatus();
 100         status.setIndex("index-1");
 101         status.setType("type-1");
 102         status.setId("id-1");
 103         status.setVersion("1.0");
 104         status.setShards(shardStatus);
 105         status.setStatus(new Integer(1));
 106         status.setError(error);
 107         status.setAdditionalProperties("REQUEST_URL", "http://127.0.0.1");
 108         Assert.assertEquals(status.getIndex(), "index-1");
 109         Assert.assertEquals(status.getType(), "type-1");
 110         Assert.assertEquals(status.getId(), "id-1");
 111         Assert.assertEquals(status.getVersion(), "1.0");
 112         Assert.assertEquals(status.getStatus(), new Integer(1));
 113         Assert.assertNotNull(status.getShards());
 114         Assert.assertNotNull(status.getError());
 115         Assert.assertNotNull(status.getAdditionalProperties());
 116         return status;
 117     }
 118
 119 }