model/basic-model/src/test/java/org/onap/policy/apex/model/basicmodel/dao/EntityTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2016-2018 Ericsson. All rights reserved.
   4  *  Modifications Copyright (C) 2020 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.model.basicmodel.dao;
  23
  24 import static org.assertj.core.api.Assertions.assertThatThrownBy;
  25 import static org.junit.Assert.assertEquals;
  26 import static org.junit.Assert.assertNotNull;
  27 import static org.junit.Assert.assertNull;
  28
  29 import java.util.ArrayList;
  30 import java.util.List;
  31 import java.util.Set;
  32 import java.util.TreeSet;
  33 import java.util.UUID;
  34 import org.junit.Test;
  35 import org.onap.policy.apex.model.basicmodel.concepts.ApexException;
  36 import org.onap.policy.apex.model.basicmodel.concepts.AxArtifactKey;
  37 import org.onap.policy.apex.model.basicmodel.concepts.AxKeyInfo;
  38 import org.onap.policy.apex.model.basicmodel.concepts.AxReferenceKey;
  39 import org.onap.policy.apex.model.basicmodel.concepts.DummyEntity;
  40
  41 /**
  42  * JUnit test class.
  43  */
  44 public class EntityTest {
  45     private ApexDao apexDao;
  46
  47     @Test
  48     public void testEntityTestSanity() throws ApexException {
  49         final DaoParameters daoParameters = new DaoParameters();
  50
  51         apexDao = new ApexDaoFactory().createApexDao(daoParameters);
  52
  53         assertThatThrownBy(() -> apexDao.init(null))
  54             .hasMessage("Apex persistence unit parameter not set");
  55         assertThatThrownBy(() -> apexDao.init(daoParameters))
  56             .hasMessage("Apex persistence unit parameter not set");
  57         daoParameters.setPluginClass("somewhere.over.the.rainbow");
  58         daoParameters.setPersistenceUnit("Dorothy");
  59         assertThatThrownBy(() -> apexDao.init(daoParameters))
  60             .hasMessage("Creation of Apex persistence unit \"Dorothy\" failed");
  61         assertThatThrownBy(() -> apexDao.create(new AxArtifactKey()))
  62             .hasMessage("Apex DAO has not been initialized");
  63         apexDao.close();
  64     }
  65
  66     @Test
  67     public void testEntityTestAllOpsJpa() throws ApexException {
  68         final DaoParameters daoParameters = new DaoParameters();
  69         daoParameters.setPluginClass("org.onap.policy.apex.model.basicmodel.dao.impl.DefaultApexDao");
  70         daoParameters.setPersistenceUnit("DaoTest");
  71
  72         apexDao = new ApexDaoFactory().createApexDao(daoParameters);
  73         apexDao.init(daoParameters);
  74
  75         testAllOps();
  76         apexDao.close();
  77     }
  78
  79     @Test
  80     public void testEntityTestBadVals() throws ApexException {
  81         final DaoParameters daoParameters = new DaoParameters();
  82         daoParameters.setPluginClass("org.onap.policy.apex.model.basicmodel.dao.impl.DefaultApexDao");
  83         daoParameters.setPersistenceUnit("DaoTest");
  84
  85         apexDao = new ApexDaoFactory().createApexDao(daoParameters);
  86         apexDao.init(daoParameters);
  87
  88         final AxArtifactKey nullKey = null;
  89         final AxReferenceKey nullRefKey = null;
  90         final List<AxArtifactKey> nullKeyList = null;
  91         final List<AxArtifactKey> emptyKeyList = new ArrayList<>();
  92         final List<AxReferenceKey> nullRKeyList = null;
  93         final List<AxReferenceKey> emptyRKeyList = new ArrayList<>();
  94
  95         apexDao.create(nullKey);
  96         apexDao.createCollection(nullKeyList);
  97         apexDao.createCollection(emptyKeyList);
  98
  99         apexDao.delete(nullKey);
 100         apexDao.deleteCollection(nullKeyList);
 101         apexDao.deleteCollection(emptyKeyList);
 102         apexDao.delete(AxArtifactKey.class, nullKey);
 103         apexDao.delete(AxReferenceKey.class, nullRefKey);
 104         apexDao.deleteByArtifactKey(AxArtifactKey.class, nullKeyList);
 105         apexDao.deleteByArtifactKey(AxArtifactKey.class, emptyKeyList);
 106         apexDao.deleteByReferenceKey(AxReferenceKey.class, nullRKeyList);
 107         apexDao.deleteByReferenceKey(AxReferenceKey.class, emptyRKeyList);
 108
 109         apexDao.get(null, nullKey);
 110         apexDao.get(null, nullRefKey);
 111         apexDao.getAll(null);
 112         apexDao.getAll(null, nullKey);
 113         apexDao.getArtifact(null, nullKey);
 114         apexDao.getArtifact(AxArtifactKey.class, nullKey);
 115         apexDao.getArtifact(null, nullRefKey);
 116         apexDao.getArtifact(AxReferenceKey.class, nullRefKey);
 117         apexDao.size(null);
 118
 119         apexDao.close();
 120     }
 121
 122     private void testAllOps() {
 123         final AxArtifactKey aKey0 = new AxArtifactKey("A-KEY0", "0.0.1");
 124         final AxArtifactKey aKey1 = new AxArtifactKey("A-KEY1", "0.0.1");
 125         final AxArtifactKey aKey2 = new AxArtifactKey("A-KEY2", "0.0.1");
 126         final AxKeyInfo keyInfo0 = new AxKeyInfo(aKey0, UUID.fromString("00000000-0000-0000-0000-000000000000"),
 127                         "key description 0");
 128         final AxKeyInfo keyInfo1 = new AxKeyInfo(aKey1, UUID.fromString("00000000-0000-0000-0000-000000000001"),
 129                         "key description 1");
 130         final AxKeyInfo keyInfo2 = new AxKeyInfo(aKey2, UUID.fromString("00000000-0000-0000-0000-000000000002"),
 131                         "key description 2");
 132
 133         apexDao.create(keyInfo0);
 134
 135         final AxKeyInfo keyInfoBack0 = apexDao.get(AxKeyInfo.class, aKey0);
 136         assertEquals(keyInfo0, keyInfoBack0);
 137
 138         final AxKeyInfo keyInfoBackNull = apexDao.get(AxKeyInfo.class, AxArtifactKey.getNullKey());
 139         assertNull(keyInfoBackNull);
 140
 141         final AxKeyInfo keyInfoBack1 = apexDao.getArtifact(AxKeyInfo.class, aKey0);
 142         assertEquals(keyInfoBack0, keyInfoBack1);
 143
 144         final AxKeyInfo keyInfoBack2 = apexDao.getArtifact(AxKeyInfo.class, new AxArtifactKey("A-KEY3", "0.0.1"));
 145         assertNull(keyInfoBack2);
 146
 147         final Set<AxKeyInfo> keyInfoSetIn = new TreeSet<AxKeyInfo>();
 148         keyInfoSetIn.add(keyInfo1);
 149         keyInfoSetIn.add(keyInfo2);
 150
 151         apexDao.createCollection(keyInfoSetIn);
 152
 153         Set<AxKeyInfo> keyInfoSetOut = new TreeSet<AxKeyInfo>(apexDao.getAll(AxKeyInfo.class));
 154
 155         keyInfoSetIn.add(keyInfo0);
 156         assertEquals(keyInfoSetIn, keyInfoSetOut);
 157
 158         apexDao.delete(keyInfo1);
 159         keyInfoSetIn.remove(keyInfo1);
 160         keyInfoSetOut = new TreeSet<AxKeyInfo>(apexDao.getAll(AxKeyInfo.class));
 161         assertEquals(keyInfoSetIn, keyInfoSetOut);
 162
 163         apexDao.deleteCollection(keyInfoSetIn);
 164         keyInfoSetOut = new TreeSet<AxKeyInfo>(apexDao.getAll(AxKeyInfo.class));
 165         assertEquals(0, keyInfoSetOut.size());
 166
 167         keyInfoSetIn.add(keyInfo0);
 168         keyInfoSetIn.add(keyInfo1);
 169         keyInfoSetIn.add(keyInfo0);
 170         apexDao.createCollection(keyInfoSetIn);
 171         keyInfoSetOut = new TreeSet<AxKeyInfo>(apexDao.getAll(AxKeyInfo.class));
 172         assertEquals(keyInfoSetIn, keyInfoSetOut);
 173
 174         apexDao.delete(AxKeyInfo.class, aKey0);
 175         keyInfoSetOut = new TreeSet<AxKeyInfo>(apexDao.getAll(AxKeyInfo.class));
 176         assertEquals(2, keyInfoSetOut.size());
 177         assertEquals(2, apexDao.size(AxKeyInfo.class));
 178
 179         final Set<AxArtifactKey> keySetIn = new TreeSet<AxArtifactKey>();
 180         keySetIn.add(aKey1);
 181         keySetIn.add(aKey2);
 182
 183         final int deletedCount = apexDao.deleteByArtifactKey(AxKeyInfo.class, keySetIn);
 184         assertEquals(2, deletedCount);
 185
 186         keyInfoSetOut = new TreeSet<AxKeyInfo>(apexDao.getAll(AxKeyInfo.class));
 187         assertEquals(0, keyInfoSetOut.size());
 188
 189         keyInfoSetIn.add(keyInfo0);
 190         keyInfoSetIn.add(keyInfo1);
 191         keyInfoSetIn.add(keyInfo0);
 192         apexDao.createCollection(keyInfoSetIn);
 193         keyInfoSetOut = new TreeSet<AxKeyInfo>(apexDao.getAll(AxKeyInfo.class));
 194         assertEquals(keyInfoSetIn, keyInfoSetOut);
 195
 196         apexDao.deleteAll(AxKeyInfo.class);
 197         assertEquals(0, apexDao.size(AxKeyInfo.class));
 198
 199         final AxArtifactKey owner0Key = new AxArtifactKey("Owner0", "0.0.1");
 200         final AxArtifactKey owner1Key = new AxArtifactKey("Owner1", "0.0.1");
 201         final AxArtifactKey owner2Key = new AxArtifactKey("Owner2", "0.0.1");
 202         final AxArtifactKey owner3Key = new AxArtifactKey("Owner3", "0.0.1");
 203         final AxArtifactKey owner4Key = new AxArtifactKey("Owner4", "0.0.1");
 204         final AxArtifactKey owner5Key = new AxArtifactKey("Owner5", "0.0.1");
 205
 206         apexDao.create(new DummyEntity(new AxReferenceKey(owner0Key, "Entity0"), 100.0));
 207         apexDao.create(new DummyEntity(new AxReferenceKey(owner0Key, "Entity1"), 101.0));
 208         apexDao.create(new DummyEntity(new AxReferenceKey(owner0Key, "Entity2"), 102.0));
 209         apexDao.create(new DummyEntity(new AxReferenceKey(owner0Key, "Entity3"), 103.0));
 210         apexDao.create(new DummyEntity(new AxReferenceKey(owner0Key, "Entity4"), 104.0));
 211         apexDao.create(new DummyEntity(new AxReferenceKey(owner1Key, "Entity5"), 105.0));
 212         apexDao.create(new DummyEntity(new AxReferenceKey(owner1Key, "Entity6"), 106.0));
 213         apexDao.create(new DummyEntity(new AxReferenceKey(owner1Key, "Entity7"), 107.0));
 214         apexDao.create(new DummyEntity(new AxReferenceKey(owner2Key, "Entity8"), 108.0));
 215         apexDao.create(new DummyEntity(new AxReferenceKey(owner2Key, "Entity9"), 109.0));
 216         apexDao.create(new DummyEntity(new AxReferenceKey(owner3Key, "EntityA"), 110.0));
 217         apexDao.create(new DummyEntity(new AxReferenceKey(owner4Key, "EntityB"), 111.0));
 218         apexDao.create(new DummyEntity(new AxReferenceKey(owner5Key, "EntityC"), 112.0));
 219         apexDao.create(new DummyEntity(new AxReferenceKey(owner5Key, "EntityD"), 113.0));
 220         apexDao.create(new DummyEntity(new AxReferenceKey(owner5Key, "EntityE"), 114.0));
 221         apexDao.create(new DummyEntity(new AxReferenceKey(owner5Key, "EntityF"), 115.0));
 222
 223         TreeSet<DummyEntity> testEntitySetOut = new TreeSet<DummyEntity>(apexDao.getAll(DummyEntity.class));
 224         assertEquals(16, testEntitySetOut.size());
 225
 226         testEntitySetOut = new TreeSet<DummyEntity>(apexDao.getAll(DummyEntity.class, owner0Key));
 227         assertEquals(5, testEntitySetOut.size());
 228
 229         testEntitySetOut = new TreeSet<DummyEntity>(apexDao.getAll(DummyEntity.class, owner1Key));
 230         assertEquals(3, testEntitySetOut.size());
 231
 232         testEntitySetOut = new TreeSet<DummyEntity>(apexDao.getAll(DummyEntity.class, owner2Key));
 233         assertEquals(2, testEntitySetOut.size());
 234
 235         testEntitySetOut = new TreeSet<DummyEntity>(apexDao.getAll(DummyEntity.class, owner3Key));
 236         assertEquals(1, testEntitySetOut.size());
 237
 238         testEntitySetOut = new TreeSet<DummyEntity>(apexDao.getAll(DummyEntity.class, owner4Key));
 239         assertEquals(1, testEntitySetOut.size());
 240
 241         testEntitySetOut = new TreeSet<DummyEntity>(apexDao.getAll(DummyEntity.class, owner5Key));
 242         assertEquals(4, testEntitySetOut.size());
 243
 244         assertNotNull(apexDao.get(DummyEntity.class, new AxReferenceKey(owner0Key, "Entity0")));
 245         assertNotNull(apexDao.getArtifact(DummyEntity.class, new AxReferenceKey(owner0Key, "Entity0")));
 246         assertNull(apexDao.get(DummyEntity.class, new AxReferenceKey(owner0Key, "Entity1000")));
 247         assertNull(apexDao.getArtifact(DummyEntity.class, new AxReferenceKey(owner0Key, "Entity1000")));
 248         apexDao.delete(DummyEntity.class, new AxReferenceKey(owner0Key, "Entity0"));
 249
 250         final Set<AxReferenceKey> rKeySetIn = new TreeSet<AxReferenceKey>();
 251         rKeySetIn.add(new AxReferenceKey(owner4Key, "EntityB"));
 252         rKeySetIn.add(new AxReferenceKey(owner5Key, "EntityD"));
 253
 254         final int deletedRCount = apexDao.deleteByReferenceKey(DummyEntity.class, rKeySetIn);
 255         assertEquals(2, deletedRCount);
 256
 257         apexDao.update(new DummyEntity(new AxReferenceKey(owner5Key, "EntityF"), 120.0));
 258     }
 259 }