2 * ============LICENSE_START=======================================================
3 * Copyright (C) 2020-2021 Nordix Foundation.
4 * Modifications Copyright (C) 2019 AT&T Intellectual Property. All rights reserved.
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
10 * http://www.apache.org/licenses/LICENSE-2.0
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.
18 * SPDX-License-Identifier: Apache-2.0
19 * ============LICENSE_END=========================================================
22 package org.onap.policy.models.tosca.simple.provider;
24 import static org.assertj.core.api.Assertions.assertThatThrownBy;
25 import static org.junit.Assert.assertEquals;
26 import static org.junit.Assert.assertNull;
28 import java.util.Properties;
29 import org.eclipse.persistence.config.PersistenceUnitProperties;
30 import org.junit.After;
31 import org.junit.Before;
32 import org.junit.Test;
33 import org.onap.policy.models.base.PfConceptKey;
34 import org.onap.policy.models.base.PfModelException;
35 import org.onap.policy.models.dao.DaoParameters;
36 import org.onap.policy.models.dao.PfDao;
37 import org.onap.policy.models.dao.PfDaoFactory;
38 import org.onap.policy.models.dao.impl.DefaultPfDao;
39 import org.onap.policy.models.tosca.simple.concepts.JpaToscaDataType;
40 import org.onap.policy.models.tosca.simple.concepts.JpaToscaDataTypes;
41 import org.onap.policy.models.tosca.simple.concepts.JpaToscaPolicyType;
42 import org.onap.policy.models.tosca.simple.concepts.JpaToscaPolicyTypes;
43 import org.onap.policy.models.tosca.simple.concepts.JpaToscaServiceTemplate;
46 * Test the {@link SimpleToscaProvider} class.
48 * @author Liam Fallon (liam.fallon@est.tech)
50 public class SimpleToscaServiceTemplateProviderTest {
51 private static final String TEMPLATE_IS_NULL = "^serviceTemplate is marked .*on.*ull but is null$";
52 private static final String DAO_IS_NULL = "^dao is marked .*on.*ull but is null$";
57 * Set up the DAO towards the database.
59 * @throws Exception on database errors
62 public void setupDao() throws Exception {
63 final DaoParameters daoParameters = new DaoParameters();
64 daoParameters.setPluginClass(DefaultPfDao.class.getName());
66 daoParameters.setPersistenceUnit("ToscaConceptTest");
68 Properties jdbcProperties = new Properties();
69 jdbcProperties.setProperty(PersistenceUnitProperties.JDBC_USER, "policy");
70 jdbcProperties.setProperty(PersistenceUnitProperties.JDBC_PASSWORD, "P01icY");
72 if (System.getProperty("USE-MARIADB") != null) {
73 jdbcProperties.setProperty(PersistenceUnitProperties.JDBC_DRIVER, "org.mariadb.jdbc.Driver");
74 jdbcProperties.setProperty(PersistenceUnitProperties.JDBC_URL, "jdbc:mariadb://localhost:3306/policy");
76 jdbcProperties.setProperty(PersistenceUnitProperties.JDBC_DRIVER, "org.h2.Driver");
77 jdbcProperties.setProperty(PersistenceUnitProperties.JDBC_URL, "jdbc:h2:mem:testdb");
80 daoParameters.setJdbcProperties(jdbcProperties);
82 pfDao = new PfDaoFactory().createPfDao(daoParameters);
83 pfDao.init(daoParameters);
87 public void teardown() {
92 public void testCreateUpdateGetDeleteDataType() throws PfModelException {
93 JpaToscaServiceTemplate serviceTemplate = new JpaToscaServiceTemplate();
95 JpaToscaServiceTemplate dbServiceTemplate =
96 new SimpleToscaServiceTemplateProvider().write(pfDao, serviceTemplate);
98 assertEquals(serviceTemplate, dbServiceTemplate);
100 JpaToscaServiceTemplate readServiceTemplate = new SimpleToscaServiceTemplateProvider().read(pfDao);
101 assertEquals(serviceTemplate, readServiceTemplate);
103 assertNull(readServiceTemplate.getDataTypes());
105 PfConceptKey dataType0Key = new PfConceptKey("DataType0", "0.0.1");
106 JpaToscaDataType dataType0 = new JpaToscaDataType();
107 dataType0.setKey(dataType0Key);
108 serviceTemplate.setDataTypes(new JpaToscaDataTypes());
109 serviceTemplate.getDataTypes().getConceptMap().put(dataType0Key, dataType0);
111 dbServiceTemplate = new SimpleToscaServiceTemplateProvider().write(pfDao, serviceTemplate);
112 assertEquals(serviceTemplate, dbServiceTemplate);
114 readServiceTemplate = new SimpleToscaServiceTemplateProvider().read(pfDao);
115 assertEquals(serviceTemplate, readServiceTemplate);
117 assertEquals(1, readServiceTemplate.getDataTypes().getConceptMap().size());
118 assertEquals(dataType0, readServiceTemplate.getDataTypes().get(dataType0Key));
119 assertNull(readServiceTemplate.getDataTypes().get(dataType0Key).getDescription());
121 dataType0.setDescription("Updated Description");
123 dbServiceTemplate = new SimpleToscaServiceTemplateProvider().write(pfDao, serviceTemplate);
124 assertEquals(serviceTemplate, dbServiceTemplate);
126 readServiceTemplate = new SimpleToscaServiceTemplateProvider().read(pfDao);
127 assertEquals(serviceTemplate, readServiceTemplate);
129 assertEquals(dataType0, readServiceTemplate.getDataTypes().get(dataType0Key));
130 assertEquals("Updated Description", readServiceTemplate.getDataTypes().get(dataType0Key).getDescription());
132 PfConceptKey policyType0Key = new PfConceptKey("PolicyType0", "0.0.1");
134 JpaToscaPolicyType policyType0 = new JpaToscaPolicyType();
136 policyType0.setKey(policyType0Key);
137 serviceTemplate.setPolicyTypes(new JpaToscaPolicyTypes());
139 serviceTemplate.getPolicyTypes().getConceptMap().put(policyType0Key, policyType0);
141 dbServiceTemplate = new SimpleToscaServiceTemplateProvider().write(pfDao, serviceTemplate);
142 assertEquals(serviceTemplate, dbServiceTemplate);
144 readServiceTemplate = new SimpleToscaServiceTemplateProvider().read(pfDao);
145 assertEquals(serviceTemplate, readServiceTemplate);
147 JpaToscaServiceTemplate deletedServiceTemplate = new SimpleToscaServiceTemplateProvider().delete(pfDao);
148 assertEquals(serviceTemplate, deletedServiceTemplate);
150 readServiceTemplate = new SimpleToscaServiceTemplateProvider().read(pfDao);
151 assertNull(readServiceTemplate);
155 public void testNonNulls() {
156 assertThatThrownBy(() -> {
157 new SimpleToscaServiceTemplateProvider().write(null, null);
158 }).hasMessageMatching(DAO_IS_NULL);
160 assertThatThrownBy(() -> {
161 new SimpleToscaServiceTemplateProvider().write(pfDao, null);
162 }).hasMessageMatching(TEMPLATE_IS_NULL);
164 assertThatThrownBy(() -> {
165 new SimpleToscaServiceTemplateProvider().write(null, new JpaToscaServiceTemplate());
166 }).hasMessageMatching(DAO_IS_NULL);
168 assertThatThrownBy(() -> {
169 new SimpleToscaServiceTemplateProvider().read(null);
170 }).hasMessageMatching(DAO_IS_NULL);
172 assertThatThrownBy(() -> {
173 new SimpleToscaServiceTemplateProvider().delete(null);
174 }).hasMessageMatching(DAO_IS_NULL);