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
9 * http://www.apache.org/licenses/LICENSE-2.0
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.
17 * SPDX-License-Identifier: Apache-2.0
18 * ============LICENSE_END=========================================================
21 package org.onap.policy.apex.model.basicmodel.dao;
23 import static org.junit.Assert.assertEquals;
24 import static org.junit.Assert.assertNotNull;
25 import static org.junit.Assert.assertNull;
26 import static org.junit.Assert.assertTrue;
27 import static org.junit.Assert.fail;
29 import java.util.ArrayList;
30 import java.util.List;
32 import java.util.TreeSet;
33 import java.util.UUID;
35 import org.junit.Test;
36 import org.onap.policy.apex.model.basicmodel.concepts.ApexException;
37 import org.onap.policy.apex.model.basicmodel.concepts.AxArtifactKey;
38 import org.onap.policy.apex.model.basicmodel.concepts.AxKeyInfo;
39 import org.onap.policy.apex.model.basicmodel.concepts.AxReferenceKey;
40 import org.onap.policy.apex.model.basicmodel.concepts.DummyEntity;
45 public class EntityTest {
46 private ApexDao apexDao;
49 public void testEntityTestSanity() throws ApexException {
50 final DaoParameters daoParameters = new DaoParameters();
52 apexDao = new ApexDaoFactory().createApexDao(daoParameters);
56 fail("Test should throw an exception here");
57 } catch (final Exception e) {
58 assertEquals("Apex persistence unit parameter not set", e.getMessage());
62 apexDao.init(daoParameters);
63 fail("Test should throw an exception here");
64 } catch (final Exception e) {
65 assertEquals("Apex persistence unit parameter not set", e.getMessage());
68 daoParameters.setPluginClass("somewhere.over.the.rainbow");
69 daoParameters.setPersistenceUnit("Dorothy");
71 apexDao.init(daoParameters);
72 fail("Test should throw an exception here");
73 } catch (final Exception e) {
74 assertEquals("Creation of Apex persistence unit \"Dorothy\" failed", e.getMessage());
77 apexDao.create(new AxArtifactKey());
78 fail("Test should throw an exception here");
79 } catch (final Exception e) {
80 assertEquals("Apex DAO has not been initialized", e.getMessage());
86 public void testEntityTestAllOpsJpa() throws ApexException {
87 final DaoParameters daoParameters = new DaoParameters();
88 daoParameters.setPluginClass("org.onap.policy.apex.model.basicmodel.dao.impl.DefaultApexDao");
89 daoParameters.setPersistenceUnit("DaoTest");
91 apexDao = new ApexDaoFactory().createApexDao(daoParameters);
92 apexDao.init(daoParameters);
99 public void testEntityTestBadVals() throws ApexException {
100 final DaoParameters daoParameters = new DaoParameters();
101 daoParameters.setPluginClass("org.onap.policy.apex.model.basicmodel.dao.impl.DefaultApexDao");
102 daoParameters.setPersistenceUnit("DaoTest");
104 apexDao = new ApexDaoFactory().createApexDao(daoParameters);
105 apexDao.init(daoParameters);
107 final AxArtifactKey nullKey = null;
108 final AxReferenceKey nullRefKey = null;
109 final List<AxArtifactKey> nullKeyList = null;
110 final List<AxArtifactKey> emptyKeyList = new ArrayList<>();
111 final List<AxReferenceKey> nullRKeyList = null;
112 final List<AxReferenceKey> emptyRKeyList = new ArrayList<>();
114 apexDao.create(nullKey);
115 apexDao.createCollection(nullKeyList);
116 apexDao.createCollection(emptyKeyList);
118 apexDao.delete(nullKey);
119 apexDao.deleteCollection(nullKeyList);
120 apexDao.deleteCollection(emptyKeyList);
121 apexDao.delete(AxArtifactKey.class, nullKey);
122 apexDao.delete(AxReferenceKey.class, nullRefKey);
123 apexDao.deleteByArtifactKey(AxArtifactKey.class, nullKeyList);
124 apexDao.deleteByArtifactKey(AxArtifactKey.class, emptyKeyList);
125 apexDao.deleteByReferenceKey(AxReferenceKey.class, nullRKeyList);
126 apexDao.deleteByReferenceKey(AxReferenceKey.class, emptyRKeyList);
128 apexDao.get(null, nullKey);
129 apexDao.get(null, nullRefKey);
130 apexDao.getAll(null);
131 apexDao.getAll(null, nullKey);
132 apexDao.getArtifact(null, nullKey);
133 apexDao.getArtifact(AxArtifactKey.class, nullKey);
134 apexDao.getArtifact(null, nullRefKey);
135 apexDao.getArtifact(AxReferenceKey.class, nullRefKey);
141 private void testAllOps() {
142 final AxArtifactKey aKey0 = new AxArtifactKey("A-KEY0", "0.0.1");
143 final AxArtifactKey aKey1 = new AxArtifactKey("A-KEY1", "0.0.1");
144 final AxArtifactKey aKey2 = new AxArtifactKey("A-KEY2", "0.0.1");
145 final AxKeyInfo keyInfo0 = new AxKeyInfo(aKey0, UUID.fromString("00000000-0000-0000-0000-000000000000"),
146 "key description 0");
147 final AxKeyInfo keyInfo1 = new AxKeyInfo(aKey1, UUID.fromString("00000000-0000-0000-0000-000000000001"),
148 "key description 1");
149 final AxKeyInfo keyInfo2 = new AxKeyInfo(aKey2, UUID.fromString("00000000-0000-0000-0000-000000000002"),
150 "key description 2");
152 apexDao.create(keyInfo0);
154 final AxKeyInfo keyInfoBack0 = apexDao.get(AxKeyInfo.class, aKey0);
155 assertTrue(keyInfo0.equals(keyInfoBack0));
157 final AxKeyInfo keyInfoBackNull = apexDao.get(AxKeyInfo.class, AxArtifactKey.getNullKey());
158 assertNull(keyInfoBackNull);
160 final AxKeyInfo keyInfoBack1 = apexDao.getArtifact(AxKeyInfo.class, aKey0);
161 assertTrue(keyInfoBack0.equals(keyInfoBack1));
163 final AxKeyInfo keyInfoBack2 = apexDao.getArtifact(AxKeyInfo.class, new AxArtifactKey("A-KEY3", "0.0.1"));
164 assertNull(keyInfoBack2);
166 final Set<AxKeyInfo> keyInfoSetIn = new TreeSet<AxKeyInfo>();
167 keyInfoSetIn.add(keyInfo1);
168 keyInfoSetIn.add(keyInfo2);
170 apexDao.createCollection(keyInfoSetIn);
172 Set<AxKeyInfo> keyInfoSetOut = new TreeSet<AxKeyInfo>(apexDao.getAll(AxKeyInfo.class));
174 keyInfoSetIn.add(keyInfo0);
175 assertTrue(keyInfoSetIn.equals(keyInfoSetOut));
177 apexDao.delete(keyInfo1);
178 keyInfoSetIn.remove(keyInfo1);
179 keyInfoSetOut = new TreeSet<AxKeyInfo>(apexDao.getAll(AxKeyInfo.class));
180 assertTrue(keyInfoSetIn.equals(keyInfoSetOut));
182 apexDao.deleteCollection(keyInfoSetIn);
183 keyInfoSetOut = new TreeSet<AxKeyInfo>(apexDao.getAll(AxKeyInfo.class));
184 assertEquals(0, keyInfoSetOut.size());
186 keyInfoSetIn.add(keyInfo0);
187 keyInfoSetIn.add(keyInfo1);
188 keyInfoSetIn.add(keyInfo0);
189 apexDao.createCollection(keyInfoSetIn);
190 keyInfoSetOut = new TreeSet<AxKeyInfo>(apexDao.getAll(AxKeyInfo.class));
191 assertTrue(keyInfoSetIn.equals(keyInfoSetOut));
193 apexDao.delete(AxKeyInfo.class, aKey0);
194 keyInfoSetOut = new TreeSet<AxKeyInfo>(apexDao.getAll(AxKeyInfo.class));
195 assertEquals(2, keyInfoSetOut.size());
196 assertEquals(2, apexDao.size(AxKeyInfo.class));
198 final Set<AxArtifactKey> keySetIn = new TreeSet<AxArtifactKey>();
202 final int deletedCount = apexDao.deleteByArtifactKey(AxKeyInfo.class, keySetIn);
203 assertEquals(2, deletedCount);
205 keyInfoSetOut = new TreeSet<AxKeyInfo>(apexDao.getAll(AxKeyInfo.class));
206 assertEquals(0, keyInfoSetOut.size());
208 keyInfoSetIn.add(keyInfo0);
209 keyInfoSetIn.add(keyInfo1);
210 keyInfoSetIn.add(keyInfo0);
211 apexDao.createCollection(keyInfoSetIn);
212 keyInfoSetOut = new TreeSet<AxKeyInfo>(apexDao.getAll(AxKeyInfo.class));
213 assertTrue(keyInfoSetIn.equals(keyInfoSetOut));
215 apexDao.deleteAll(AxKeyInfo.class);
216 assertEquals(0, apexDao.size(AxKeyInfo.class));
218 final AxArtifactKey owner0Key = new AxArtifactKey("Owner0", "0.0.1");
219 final AxArtifactKey owner1Key = new AxArtifactKey("Owner1", "0.0.1");
220 final AxArtifactKey owner2Key = new AxArtifactKey("Owner2", "0.0.1");
221 final AxArtifactKey owner3Key = new AxArtifactKey("Owner3", "0.0.1");
222 final AxArtifactKey owner4Key = new AxArtifactKey("Owner4", "0.0.1");
223 final AxArtifactKey owner5Key = new AxArtifactKey("Owner5", "0.0.1");
225 apexDao.create(new DummyEntity(new AxReferenceKey(owner0Key, "Entity0"), 100.0));
226 apexDao.create(new DummyEntity(new AxReferenceKey(owner0Key, "Entity1"), 101.0));
227 apexDao.create(new DummyEntity(new AxReferenceKey(owner0Key, "Entity2"), 102.0));
228 apexDao.create(new DummyEntity(new AxReferenceKey(owner0Key, "Entity3"), 103.0));
229 apexDao.create(new DummyEntity(new AxReferenceKey(owner0Key, "Entity4"), 104.0));
230 apexDao.create(new DummyEntity(new AxReferenceKey(owner1Key, "Entity5"), 105.0));
231 apexDao.create(new DummyEntity(new AxReferenceKey(owner1Key, "Entity6"), 106.0));
232 apexDao.create(new DummyEntity(new AxReferenceKey(owner1Key, "Entity7"), 107.0));
233 apexDao.create(new DummyEntity(new AxReferenceKey(owner2Key, "Entity8"), 108.0));
234 apexDao.create(new DummyEntity(new AxReferenceKey(owner2Key, "Entity9"), 109.0));
235 apexDao.create(new DummyEntity(new AxReferenceKey(owner3Key, "EntityA"), 110.0));
236 apexDao.create(new DummyEntity(new AxReferenceKey(owner4Key, "EntityB"), 111.0));
237 apexDao.create(new DummyEntity(new AxReferenceKey(owner5Key, "EntityC"), 112.0));
238 apexDao.create(new DummyEntity(new AxReferenceKey(owner5Key, "EntityD"), 113.0));
239 apexDao.create(new DummyEntity(new AxReferenceKey(owner5Key, "EntityE"), 114.0));
240 apexDao.create(new DummyEntity(new AxReferenceKey(owner5Key, "EntityF"), 115.0));
242 TreeSet<DummyEntity> testEntitySetOut = new TreeSet<DummyEntity>(apexDao.getAll(DummyEntity.class));
243 assertEquals(16, testEntitySetOut.size());
245 testEntitySetOut = new TreeSet<DummyEntity>(apexDao.getAll(DummyEntity.class, owner0Key));
246 assertEquals(5, testEntitySetOut.size());
248 testEntitySetOut = new TreeSet<DummyEntity>(apexDao.getAll(DummyEntity.class, owner1Key));
249 assertEquals(3, testEntitySetOut.size());
251 testEntitySetOut = new TreeSet<DummyEntity>(apexDao.getAll(DummyEntity.class, owner2Key));
252 assertEquals(2, testEntitySetOut.size());
254 testEntitySetOut = new TreeSet<DummyEntity>(apexDao.getAll(DummyEntity.class, owner3Key));
255 assertEquals(1, testEntitySetOut.size());
257 testEntitySetOut = new TreeSet<DummyEntity>(apexDao.getAll(DummyEntity.class, owner4Key));
258 assertEquals(1, testEntitySetOut.size());
260 testEntitySetOut = new TreeSet<DummyEntity>(apexDao.getAll(DummyEntity.class, owner5Key));
261 assertEquals(4, testEntitySetOut.size());
263 assertNotNull(apexDao.get(DummyEntity.class, new AxReferenceKey(owner0Key, "Entity0")));
264 assertNotNull(apexDao.getArtifact(DummyEntity.class, new AxReferenceKey(owner0Key, "Entity0")));
265 assertNull(apexDao.get(DummyEntity.class, new AxReferenceKey(owner0Key, "Entity1000")));
266 assertNull(apexDao.getArtifact(DummyEntity.class, new AxReferenceKey(owner0Key, "Entity1000")));
267 apexDao.delete(DummyEntity.class, new AxReferenceKey(owner0Key, "Entity0"));
269 final Set<AxReferenceKey> rKeySetIn = new TreeSet<AxReferenceKey>();
270 rKeySetIn.add(new AxReferenceKey(owner4Key, "EntityB"));
271 rKeySetIn.add(new AxReferenceKey(owner5Key, "EntityD"));
273 final int deletedRCount = apexDao.deleteByReferenceKey(DummyEntity.class, rKeySetIn);
274 assertEquals(2, deletedRCount);
276 apexDao.update(new DummyEntity(new AxReferenceKey(owner5Key, "EntityF"), 120.0));