resource-assignment/provider/src/test/java/jtest/org/openecomp/sdnc/rm/DataSetup.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * openECOMP : SDN-C
   4  * ================================================================================
   5  * Copyright (C) 2017 ONAP Intellectual Property. All rights
   6  *                                              reserved.
   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
  22 package jtest.org.openecomp.sdnc.rm;
  23
  24 import java.util.Date;
  25
  26 import jtest.util.org.openecomp.sdnc.ra.TestDb;
  27 import jtest.util.org.openecomp.sdnc.ra.TestTable;
  28
  29 public class DataSetup {
  30
  31         private TestDb testDb;
  32
  33         private TestTable resource = null;
  34         private TestTable allocationItem = null;
  35
  36         private static final String[] RESOURCE_COLUMNS =
  37                 { "asset_id", "resource_name", "resource_type", "lt_used", "ll_label", "ll_reference_count", "rr_used" };
  38
  39         private static final String[] ALLOCATION_ITEM_COLUMNS = {
  40                 "resource_id", "application_id", "resource_set_id", "resource_union_id", "resource_share_group_list",
  41                 "lt_used", "ll_label", "rr_used", "allocation_time" };
  42
  43         private void initTables() {
  44                 if (resource == null)
  45                         resource = testDb.table("RESOURCE", "resource_id", RESOURCE_COLUMNS);
  46                 if (allocationItem == null)
  47                         allocationItem = testDb.table("ALLOCATION_ITEM", "allocation_item_id", ALLOCATION_ITEM_COLUMNS);
  48         }
  49
  50         public void cleanup() {
  51                 allocationItem.delete("true");
  52                 resource.delete("true");
  53         }
  54
  55         public void setupLimitItem(
  56                 String resourceName,
  57                 String assetId,
  58                 String resourceSetId,
  59                 String resourceUnionId,
  60                 long used) {
  61                 initTables();
  62
  63                 Long rid = resource.getId("asset_id = '" + assetId + "' AND resource_name = '" + resourceName + "'");
  64                 if (rid == null) {
  65                         resource.add(assetId, resourceName, "Limit", 1, null, null, null);
  66                         rid = resource.getLastId();
  67                 }
  68                 allocationItem.add(rid, "SDNC", resourceSetId, resourceUnionId, null, used, null, null, new Date());
  69         }
  70
  71         public void setupRangeItem(
  72                 String resourceName,
  73                 String assetId,
  74                 String resourceSetId,
  75                 String resourceUnionId,
  76                 String used) {
  77                 initTables();
  78
  79                 Long rid = resource.getId("asset_id = '" + assetId + "' AND resource_name = '" + resourceName + "'");
  80                 if (rid == null) {
  81                         resource.add(assetId, resourceName, "Range", null, null, null, used);
  82                         rid = resource.getLastId();
  83                 }
  84                 allocationItem.add(rid, "SDNC", resourceSetId, resourceUnionId, null, null, null, used, new Date());
  85         }
  86
  87         public void setupLabelItem(
  88                 String resourceName,
  89                 String assetId,
  90                 String resourceSetId,
  91                 String resourceUnionId,
  92                 String label) {
  93                 initTables();
  94
  95                 Long rid = resource.getId("asset_id = '" + assetId + "' AND resource_name = '" + resourceName + "'");
  96                 if (rid == null) {
  97                         resource.add(assetId, resourceName, "Label", null, label, 1, null);
  98                         rid = resource.getLastId();
  99                 }
 100                 allocationItem.add(rid, "SDNC", resourceSetId, resourceUnionId, null, null, label, null, new Date());
 101         }
 102
 103         public void setTestDb(TestDb testDb) {
 104                 this.testDb = testDb;
 105         }
 106 }