aboutsummaryrefslogtreecommitdiffstats
path: root/site-manager/src/test/java/org
diff options
context:
space:
mode:
authoreikrwaq <waqas.ikram@ericsson.com>2018-03-21 15:31:05 +0000
committereikrwaq <waqas.ikram@ericsson.com>2018-03-22 15:00:51 +0000
commit586e553d141e087566a6b064121910089cf52cff (patch)
tree73cac4eb73e083e567346488e1ac9b4c895b90df /site-manager/src/test/java/org
parent64627f345bfe5e5bd62524783594655dd8456ace (diff)
Add junit and refactor code
Adding Junit tests in modules and refactoring code Change-Id: I887f1b2371d9e3d4a4e2ce6e7e1f3c97a361f283 Issue-ID: POLICY-704 Signed-off-by: eikrwaq <waqas.ikram@ericsson.com>
Diffstat (limited to 'site-manager/src/test/java/org')
-rw-r--r--site-manager/src/test/java/org/onap/policy/common/sitemanager/MainTest.java495
-rw-r--r--site-manager/src/test/java/org/onap/policy/common/sitemanager/MainTestRunner.java73
-rw-r--r--site-manager/src/test/java/org/onap/policy/common/sitemanager/exception/IllegalCommandLineArgumentExceptionTest.java35
-rw-r--r--site-manager/src/test/java/org/onap/policy/common/sitemanager/exception/MissingPropertyExceptionTest.java35
-rw-r--r--site-manager/src/test/java/org/onap/policy/common/sitemanager/exception/NoMatchingEntryFoundExceptionTest.java35
-rw-r--r--site-manager/src/test/java/org/onap/policy/common/sitemanager/exception/PropertyFileProcessingExceptionTest.java35
-rw-r--r--site-manager/src/test/java/org/onap/policy/common/sitemanager/utils/CommandLineHelperTest.java194
-rw-r--r--site-manager/src/test/java/org/onap/policy/common/sitemanager/utils/ExtraCommandLineArgumentTest.java84
-rw-r--r--site-manager/src/test/java/org/onap/policy/common/sitemanager/utils/PersistenceUnitPropertiesProviderTest.java97
-rw-r--r--site-manager/src/test/java/org/onap/policy/common/sitemanager/utils/PrintableImpl.java39
10 files changed, 1122 insertions, 0 deletions
diff --git a/site-manager/src/test/java/org/onap/policy/common/sitemanager/MainTest.java b/site-manager/src/test/java/org/onap/policy/common/sitemanager/MainTest.java
new file mode 100644
index 00000000..d8cb5f5f
--- /dev/null
+++ b/site-manager/src/test/java/org/onap/policy/common/sitemanager/MainTest.java
@@ -0,0 +1,495 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * site-manager
+ * ================================================================================
+ * Copyright (C) 2018 Ericsson. All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+package org.onap.policy.common.sitemanager;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+import static org.junit.Assert.fail;
+import java.io.BufferedWriter;
+import java.io.File;
+import java.io.IOException;
+import java.nio.file.Files;
+import java.util.Arrays;
+import java.util.Date;
+import java.util.List;
+import java.util.Properties;
+import javax.persistence.EntityManager;
+import javax.persistence.EntityManagerFactory;
+import javax.persistence.EntityTransaction;
+import javax.persistence.Persistence;
+import javax.persistence.TypedQuery;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.rules.TemporaryFolder;
+import org.onap.policy.common.im.jpa.ResourceRegistrationEntity;
+import org.onap.policy.common.im.jpa.StateManagementEntity;
+import org.onap.policy.common.sitemanager.MainTestRunner.ExitCodeTestException;
+import org.onap.policy.common.sitemanager.data.service.DatabaseAccessService;
+import org.onap.policy.common.sitemanager.data.service.DatabaseAccessServiceImpl;
+import org.onap.policy.common.sitemanager.utils.Constants;
+import org.onap.policy.common.sitemanager.utils.ErrorMessages;
+import org.onap.policy.common.sitemanager.utils.PrintableImpl;
+import org.onap.policy.common.utils.jpa.EntityMgrCloser;
+import org.onap.policy.common.utils.jpa.EntityTransCloser;
+
+/**
+ * This class contains the main entry point for Site Manager.
+ */
+public class MainTest {
+
+ private static final String SITE_NAME = "SITE";
+
+ private static final String PROPERTY_FILE_NAME = "siteManagerPropertyFile.properties";
+
+ private static final String RESOURCE_LOCAL = "RESOURCE_LOCAL";
+
+ private static final String DROP_AND_CREATE_TABLES = "drop-and-create-tables";
+
+ private static final String NULL_STRING = "NULL";
+
+ private static final String DATABASE_H2_PLATFORM = "org.eclipse.persistence.platform.database.H2Platform";
+
+ private static final String ECLIPSELINK_ID_VALIDATION = "eclipselink.id-validation";
+
+ private static final String ECLIPSELINK_TARGET_DATABASE = "eclipselink.target-database";
+
+ private static final String DDL_GENERATION = "eclipselink.ddl-generation";
+
+ private static final String TRANSACTION_TYPE = "javax.persistence.transactionType";
+
+ private static final String RESOURCE_NAME = "RESOURCE_NAME";
+
+ private static final String ADMIN_STATE_NEW_VALUE = "NEW_VALUE";
+
+ private static final Date CURRENT_DATE = new Date();
+
+ private static final String COMMENTS = "";
+
+ private static final String DEFAULT_DB_URL_PREFIX = "jdbc:h2:mem:";
+
+ private static final String IN_MEMORY_DB_URL = DEFAULT_DB_URL_PREFIX + "myDb";
+
+ private static final String H2_DB_DRIVER = "org.h2.Driver";
+
+ private final Properties properties = new Properties();
+
+ @Rule
+ public TemporaryFolder temporaryFolder = new TemporaryFolder();
+
+ private EntityManagerFactory managerFactory;
+
+ private final MainTestRunner testRunner = new MainTestRunner();
+
+ @Before
+ public void setUp() throws Exception {
+
+ properties.put(Constants.JDBC_DRIVER_PROPERTY_NAME, H2_DB_DRIVER);
+ properties.put(Constants.JDBC_URL_PROPERTY_NAME, IN_MEMORY_DB_URL);
+ properties.put(Constants.JDBC_USER_PROPERTY_NAME, "test");
+ properties.put(Constants.JDBC_PASSWORD_PROPERTY_NAME, "test");
+ properties.put(TRANSACTION_TYPE, RESOURCE_LOCAL);
+ properties.put(DDL_GENERATION, DROP_AND_CREATE_TABLES);
+ properties.put(ECLIPSELINK_TARGET_DATABASE, DATABASE_H2_PLATFORM);
+ properties.put(ECLIPSELINK_ID_VALIDATION, NULL_STRING);
+
+ final File file = temporaryFolder.newFile(PROPERTY_FILE_NAME);
+ creatPropertyFile(file, properties);
+ System.setProperty(Constants.SITE_MANAGER_PROPERTIES_PROPERTY_NAME, file.toString());
+ managerFactory = Persistence.createEntityManagerFactory(Constants.OPERATIONAL_PERSISTENCE_UNIT, properties);
+
+ testRunner.setUp();
+ }
+
+ @After
+ public void destroy() {
+ Main.stateManagementTable.clear();
+ Main.resourceRegistrationTable.clear();
+ testRunner.destroy();
+ properties.clear();;
+ if (managerFactory.isOpen()) {
+ managerFactory.close();
+ }
+ }
+
+ @Test
+ public void test_process_initializitonWithOutException() throws IOException {
+
+ persist(managerFactory.createEntityManager(), getStateManagementEntity());
+
+ persist(managerFactory.createEntityManager(), getResourceRegistrationEntity());
+
+ final String[] args = new String[] {"show"};
+ final PrintableImpl printable = new PrintableImpl();
+ final Main objUnderTest = getMain();
+ objUnderTest.process(args, printable);
+
+ assertEquals(1, Main.resourceRegistrationTable.size());
+ assertEquals(1, Main.stateManagementTable.size());
+ assertTrue(printable.getResult().isEmpty());
+
+ }
+
+ @Test
+ public void test_process_setAdminStateWithResourceNameArgument() throws IOException {
+ persist(managerFactory.createEntityManager(), getStateManagementEntity());
+ persist(managerFactory.createEntityManager(), getResourceRegistrationEntity());
+
+ final String[] args = new String[] {"setAdminState", "-r", RESOURCE_NAME, ADMIN_STATE_NEW_VALUE};
+ final PrintableImpl printable = new PrintableImpl();
+ final Main objUnderTest = getMain();
+ objUnderTest.process(args, printable);
+
+ assertEquals(1, Main.resourceRegistrationTable.size());
+ assertEquals(1, Main.stateManagementTable.size());
+ assertTrue(printable.getResult().isEmpty());
+
+ final List<StateManagementEntity> execute = execute(managerFactory.createEntityManager(),
+ StateManagementEntity.class, Constants.STATE_MANAGEMENT_QUERY);
+
+ assertEquals(ADMIN_STATE_NEW_VALUE, execute.get(0).getAdminState());
+
+ }
+
+ @Test
+ public void test_process_setAdminStateWithStateArgument() throws IOException {
+ persist(managerFactory.createEntityManager(), getStateManagementEntity());
+ persist(managerFactory.createEntityManager(), getResourceRegistrationEntity());
+
+ final String[] args = new String[] {"setAdminState", "-s", SITE_NAME, RESOURCE_NAME};
+ final PrintableImpl printable = new PrintableImpl();
+ final Main objUnderTest = getMain();
+ objUnderTest.process(args, printable);
+
+ assertEquals(1, Main.resourceRegistrationTable.size());
+ assertEquals(1, Main.stateManagementTable.size());
+ final List<String> result = printable.getResult();
+ assertTrue(result.isEmpty());
+
+ }
+
+ @Test
+ public void test_process_propertyFileNotAvailable() throws IOException {
+
+ final File file = temporaryFolder.newFile("New" + PROPERTY_FILE_NAME);
+ System.setProperty(Constants.SITE_MANAGER_PROPERTIES_PROPERTY_NAME, file.toString());
+
+ final String[] args = new String[] {"setAdminState", "-s", SITE_NAME, RESOURCE_NAME};
+ final PrintableImpl printable = new PrintableImpl();
+ try {
+ final Main objUnderTest = getMain();
+ objUnderTest.process(args, printable);
+ fail("ExitCodeTestException must be thrown in MainTestRunner class when System.exit() is called in Main class");
+ } catch (final ExitCodeTestException exitCodeTestException) {
+ assertEquals(3, exitCodeTestException.exitCode);
+
+ }
+ assertTrue(Main.stateManagementTable.isEmpty());
+ assertTrue(Main.resourceRegistrationTable.isEmpty());
+ assertFalse(printable.getResult().isEmpty());
+
+ }
+
+ @Test
+ public void test_process_emptyPropertyFile() throws IOException {
+
+ final File file = temporaryFolder.newFile("New" + PROPERTY_FILE_NAME);
+ System.setProperty(Constants.SITE_MANAGER_PROPERTIES_PROPERTY_NAME, file.toString());
+ creatPropertyFile(file, new Properties());
+ final String[] args = new String[] {"setAdminState", "-s", SITE_NAME, RESOURCE_NAME};
+ final PrintableImpl printable = new PrintableImpl();
+ try {
+ final Main objUnderTest = getMain();
+ objUnderTest.process(args, printable);
+ fail("ExitCodeTestException must be thrown in MainTestRunner class when System.exit() is called in Main class");
+ } catch (final ExitCodeTestException exitCodeTestException) {
+ assertEquals(3, exitCodeTestException.exitCode);
+
+ }
+ assertTrue(Main.stateManagementTable.isEmpty());
+ assertTrue(Main.resourceRegistrationTable.isEmpty());
+ assertFalse(printable.getResult().isEmpty());
+
+ }
+
+ @Test
+ public void test_process_MissingAttributesInPropertyFile() throws IOException {
+
+ final File file = temporaryFolder.newFile("New" + PROPERTY_FILE_NAME);
+ System.setProperty(Constants.SITE_MANAGER_PROPERTIES_PROPERTY_NAME, file.toString());
+ final Properties properties = new Properties();
+ properties.put(Constants.JDBC_DRIVER_PROPERTY_NAME, H2_DB_DRIVER);
+ creatPropertyFile(file, properties);
+ final String[] args = new String[] {"setAdminState", "-s", SITE_NAME, RESOURCE_NAME};
+ final PrintableImpl printable = new PrintableImpl();
+ try {
+ final Main objUnderTest = getMain();
+ objUnderTest.process(args, printable);
+ fail("ExitCodeTestException must be thrown in MainTestRunner class when System.exit() is called in Main class");
+ } catch (final ExitCodeTestException exitCodeTestException) {
+ assertEquals(3, exitCodeTestException.exitCode);
+
+ }
+ assertTrue(Main.stateManagementTable.isEmpty());
+ assertTrue(Main.resourceRegistrationTable.isEmpty());
+ assertFalse(printable.getResult().isEmpty());
+
+ }
+
+ @Test
+ public void test_process_MissingOneAttributesInPropertyFile() throws IOException {
+
+ final File file = temporaryFolder.newFile("New" + PROPERTY_FILE_NAME);
+ System.setProperty(Constants.SITE_MANAGER_PROPERTIES_PROPERTY_NAME, file.toString());
+ final Properties properties = new Properties();
+ properties.put(Constants.JDBC_DRIVER_PROPERTY_NAME, H2_DB_DRIVER);
+ properties.put(Constants.JDBC_URL_PROPERTY_NAME, IN_MEMORY_DB_URL);
+ properties.put(Constants.JDBC_USER_PROPERTY_NAME, "test");
+ creatPropertyFile(file, properties);
+ final String[] args = new String[] {"setAdminState", "-s", SITE_NAME, RESOURCE_NAME};
+ final PrintableImpl printable = new PrintableImpl();
+ try {
+ final Main objUnderTest = getMain();
+ objUnderTest.process(args, printable);
+ fail("ExitCodeTestException must be thrown in MainTestRunner class when System.exit() is called in Main class");
+ } catch (final ExitCodeTestException exitCodeTestException) {
+ assertEquals(3, exitCodeTestException.exitCode);
+
+ }
+ assertTrue(Main.stateManagementTable.isEmpty());
+ assertTrue(Main.resourceRegistrationTable.isEmpty());
+ assertFalse(printable.getResult().isEmpty());
+
+ }
+
+ @Test
+ public void test_process_emptyArguments_printError() throws IOException {
+
+ final PrintableImpl printable = new PrintableImpl();
+ try {
+ final String[] args = new String[] {};
+ final Main objUnderTest = getMain();
+ objUnderTest.process(args, printable);
+ fail("ExitCodeTestException must be thrown in MainTestRunner class when System.exit() is called in Main class");
+ } catch (final ExitCodeTestException exitCodeTestException) {
+ assertEquals(2, exitCodeTestException.exitCode);
+
+ }
+
+ assertTrue(Main.stateManagementTable.isEmpty());
+ assertTrue(Main.resourceRegistrationTable.isEmpty());
+ final List<String> actualMessages = printable.getResult();
+ assertFalse(actualMessages.isEmpty());
+
+ assertEquals(Arrays.asList(ErrorMessages.NO_COMMAND_SPECIFIED, ErrorMessages.HELP_STRING), actualMessages);
+
+ }
+
+ @Test
+ public void test_process_nonEmptyWithHelpArguments_printHelp() throws IOException {
+
+ final PrintableImpl printable = new PrintableImpl();
+ try {
+ final String[] args = new String[] {"---", ""};
+ final Main objUnderTest = getMain();
+ objUnderTest.process(args, printable);
+ fail("ExitCodeTestException must be thrown in MainTestRunner class when System.exit() is called in Main class");
+ } catch (final ExitCodeTestException exitCodeTestException) {
+ assertEquals(1, exitCodeTestException.exitCode);
+
+ }
+ assertTrue(Main.stateManagementTable.isEmpty());
+ assertTrue(Main.resourceRegistrationTable.isEmpty());
+ final List<String> actualMessages = printable.getResult();
+ assertFalse(actualMessages.isEmpty());
+ assertEquals(Arrays.asList("Unrecognized option: ---", ErrorMessages.HELP_STRING), actualMessages);
+
+ }
+
+ @Test
+ public void test_process_HelpArguments_printHelp() {
+
+ final PrintableImpl printable = new PrintableImpl();
+ try {
+ final String[] args = new String[] {"-h"};
+ final Main objUnderTest = getMain();
+ objUnderTest.process(args, printable);
+ fail("ExitCodeTestException must be thrown in MainTestRunner class when System.exit() is called in Main class");
+ } catch (final ExitCodeTestException exitCodeTestException) {
+ assertEquals(0, exitCodeTestException.exitCode);
+
+ }
+ assertTrue(Main.stateManagementTable.isEmpty());
+ assertTrue(Main.resourceRegistrationTable.isEmpty());
+ final List<String> actualMessages = printable.getResult();
+ assertFalse(actualMessages.isEmpty());
+ assertEquals(Arrays.asList(ErrorMessages.HELP_STRING), actualMessages);
+
+ }
+
+ @Test
+ public void test_process_missingArguments_printHelpAndErrorMessage() {
+
+ final PrintableImpl printable = new PrintableImpl();
+ try {
+ final String[] args = new String[] {"setAdminState", "-s", RESOURCE_NAME};
+ final Main objUnderTest = getMain();
+ objUnderTest.process(args, printable);
+ fail("ExitCodeTestException must be thrown in MainTestRunner class when System.exit() is called in Main class");
+ } catch (final ExitCodeTestException exitCodeTestException) {
+ assertEquals(2, exitCodeTestException.exitCode);
+
+ }
+ assertTrue(Main.stateManagementTable.isEmpty());
+ assertTrue(Main.resourceRegistrationTable.isEmpty());
+ final List<String> actualMessages = printable.getResult();
+ assertFalse(actualMessages.isEmpty());
+ assertEquals(Arrays.asList(ErrorMessages.SET_ADMIN_STATE_MISSING_NEW_STATE_VALUE, ErrorMessages.HELP_STRING),
+ actualMessages);
+
+ }
+
+ @Test
+ public void test_process_extraShowArguments_printHelpAndErrorMessage() {
+ final String[] args = new String[] {"show", RESOURCE_NAME};
+
+ runAndAssertErrorMessage(args, ErrorMessages.SHOW_EXTRA_ARGUMENTS, ErrorMessages.HELP_STRING);
+ }
+
+ @Test
+ public void test_process_lockShowArguments_printHelpAndErrorMessage() {
+ final String[] args = new String[] {"lock", RESOURCE_NAME};
+
+ runAndAssertErrorMessage(args, ErrorMessages.LOCK_EXTRA_ARGUMENTS,
+ ErrorMessages.LOCK_EITHER_S_OR_R_OPTION_IS_NEEDED, ErrorMessages.HELP_STRING);
+ }
+
+ @Test
+ public void test_process_unknowArguments_printHelpAndErrorMessage() {
+ final String[] args = new String[] {"x", RESOURCE_NAME};
+
+ runAndAssertErrorMessage(args, "x" + ErrorMessages.UNKNOWN_COMMAND, ErrorMessages.HELP_STRING);
+ }
+
+ @Test
+ public void test_process_SetAdminStateArguments_printHelpAndErrorMessage() {
+ final String[] args = new String[] {"setAdminState", RESOURCE_NAME, RESOURCE_NAME};
+
+ runAndAssertErrorMessage(args, ErrorMessages.SET_ADMIN_STATE_EXTRA_ARGUMENTS,
+ ErrorMessages.SET_ADMIN_STATE_EITHER_S_OR_R_OPTION_IS_NEEDED, ErrorMessages.HELP_STRING);
+ }
+
+ @Test
+ public void test_process_SetAdminStateArguments_printHelpAndErrorMessage2() {
+ final String[] args = new String[] {"lock", "-s", SITE_NAME, "-r", RESOURCE_NAME};
+
+ runAndAssertErrorMessage(args, "lock" + ErrorMessages.R_AND_S_OPTIONS_ARE_MUTUALLY_EXCLUSIVE,
+ ErrorMessages.HELP_STRING);
+ }
+
+ @Test
+ public void test_process_unlockShowArguments_printHelpAndErrorMessage() {
+ final String[] args = new String[] {"unlock", RESOURCE_NAME};
+
+ runAndAssertErrorMessage(args, ErrorMessages.UNLOCK_EXTRA_ARGUMENTS,
+ ErrorMessages.UNLOCK_EITHER_S_OR_R_OPTION_IS_NEEDED, ErrorMessages.HELP_STRING);
+ }
+
+ @Test
+ public void test_process_StateAndResourceArguments_printHelpAndErrorMessage() {
+ final String[] args = new String[] {"-s", SITE_NAME, "-r", RESOURCE_NAME};
+
+ runAndAssertErrorMessage(args, ErrorMessages.NO_COMMAND_SPECIFIED, ErrorMessages.HELP_STRING);
+
+ }
+
+ private void runAndAssertErrorMessage(final String[] args, final String... expectedErrorMessages) {
+ final PrintableImpl printable = new PrintableImpl();
+ try {
+ final Main objUnderTest = getMain();
+ objUnderTest.process(args, printable);
+ fail("ExitCodeTestException must be thrown in MainTestRunner class when System.exit() is called in Main class");
+ } catch (final ExitCodeTestException exitCodeTestException) {
+ assertEquals(2, exitCodeTestException.exitCode);
+
+ }
+ assertTrue(Main.stateManagementTable.isEmpty());
+ assertTrue(Main.resourceRegistrationTable.isEmpty());
+ final List<String> actualMessages = printable.getResult();
+ assertFalse(actualMessages.isEmpty());
+ assertEquals(Arrays.asList(expectedErrorMessages), actualMessages);
+ }
+
+ private <T> void persist(final EntityManager entityManager, final T entity) {
+ final EntityTransaction entityTransaction = entityManager.getTransaction();
+ try (final EntityMgrCloser emc = new EntityMgrCloser(entityManager);
+ final EntityTransCloser transaction = new EntityTransCloser(entityTransaction)) {
+ entityManager.persist(entity);
+ transaction.commit();
+ }
+ }
+
+ public <T> List<T> execute(final EntityManager entityManager, final Class<T> clazz, final String query) {
+ try (final EntityMgrCloser entityMgrCloser = new EntityMgrCloser(entityManager);) {
+ final TypedQuery<T> typedQuery = entityManager.createQuery(query, clazz);
+ return typedQuery.getResultList();
+ }
+ }
+
+ private Main getMain() {
+ final Main objUnderTest = new Main() {
+ @Override
+ DatabaseAccessService getDatabaseAccessService(final String persistenceUnitName,
+ final Properties properties) {
+ return new DatabaseAccessServiceImpl(managerFactory);
+ };
+ };
+ return objUnderTest;
+ }
+
+ private StateManagementEntity getStateManagementEntity() {
+ final StateManagementEntity entity = new StateManagementEntity();
+ entity.setModifiedDate(CURRENT_DATE);
+ entity.setAdminState("AdminState");
+ entity.setResourceName(RESOURCE_NAME);
+ entity.setAvailStatus("AvailStatus");
+ entity.setOpState("OpState");
+ entity.setStandbyStatus("StandbyStatus");
+ return entity;
+ }
+
+ private ResourceRegistrationEntity getResourceRegistrationEntity() {
+ final ResourceRegistrationEntity entity = new ResourceRegistrationEntity();
+ entity.setLastUpdated(CURRENT_DATE);
+ entity.setNodeType("NODE_NAME");
+ entity.setResourceName(RESOURCE_NAME);
+ entity.setResourceUrl("/path/to/something");
+ entity.setSite(SITE_NAME);
+ return entity;
+ }
+
+ private void creatPropertyFile(final File file, final Properties properties) throws IOException {
+ try (final BufferedWriter bufferedWriter = Files.newBufferedWriter(file.toPath());) {
+ properties.store(bufferedWriter, COMMENTS);
+ }
+ }
+}
diff --git a/site-manager/src/test/java/org/onap/policy/common/sitemanager/MainTestRunner.java b/site-manager/src/test/java/org/onap/policy/common/sitemanager/MainTestRunner.java
new file mode 100644
index 00000000..f763d4a4
--- /dev/null
+++ b/site-manager/src/test/java/org/onap/policy/common/sitemanager/MainTestRunner.java
@@ -0,0 +1,73 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * site-manager
+ * ================================================================================
+ * Copyright (C) 2018 Ericsson. All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+
+package org.onap.policy.common.sitemanager;
+
+import java.security.Permission;
+
+public class MainTestRunner {
+
+ private SecurityManager savedSecurityManager;
+ private final ExitCodeSecurityManager exitCodeSecurityManager = new ExitCodeSecurityManager();;
+ private boolean isSetUp = false;
+
+ public void setUp() {
+ if (!isSetUp) {
+ savedSecurityManager = System.getSecurityManager();
+ System.setSecurityManager(exitCodeSecurityManager);
+ isSetUp = true;
+ }
+ }
+
+ public void destroy() {
+ if (isSetUp) {
+ System.setSecurityManager(savedSecurityManager);
+ isSetUp = false;
+ }
+ }
+
+ class ExitCodeTestException extends SecurityException {
+ private static final long serialVersionUID = 2690072276259821984L;
+ public final int exitCode;
+
+ public ExitCodeTestException(final int exitCode) {
+ super("Test specific exit code exception to handle System.exit, value: " + exitCode);
+ this.exitCode = exitCode;
+ }
+ }
+
+ private class ExitCodeSecurityManager extends SecurityManager {
+
+ @Override
+ public void checkPermission(final Permission perm) {
+ }
+
+ @Override
+ public void checkPermission(final Permission perm, final Object context) {
+ }
+
+ @Override
+ public void checkExit(final int status) {
+ super.checkExit(status);
+ throw new ExitCodeTestException(status);
+ }
+
+ }
+}
diff --git a/site-manager/src/test/java/org/onap/policy/common/sitemanager/exception/IllegalCommandLineArgumentExceptionTest.java b/site-manager/src/test/java/org/onap/policy/common/sitemanager/exception/IllegalCommandLineArgumentExceptionTest.java
new file mode 100644
index 00000000..73a56919
--- /dev/null
+++ b/site-manager/src/test/java/org/onap/policy/common/sitemanager/exception/IllegalCommandLineArgumentExceptionTest.java
@@ -0,0 +1,35 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * site-manager
+ * ================================================================================
+ * Copyright (C) 2018 Ericsson. All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+
+package org.onap.policy.common.sitemanager.exception;
+
+import static org.junit.Assert.assertEquals;
+import org.junit.Test;
+import org.onap.policy.common.sitemanager.exception.IllegalCommandLineArgumentException;
+import org.onap.policy.common.utils.test.ExceptionsTester;
+
+public class IllegalCommandLineArgumentExceptionTest extends ExceptionsTester {
+
+ @Test
+ public void testIllegalCommandLineArgumentException() throws Exception {
+ assertEquals(1, test(IllegalCommandLineArgumentException.class));
+ }
+
+}
diff --git a/site-manager/src/test/java/org/onap/policy/common/sitemanager/exception/MissingPropertyExceptionTest.java b/site-manager/src/test/java/org/onap/policy/common/sitemanager/exception/MissingPropertyExceptionTest.java
new file mode 100644
index 00000000..7a9963f2
--- /dev/null
+++ b/site-manager/src/test/java/org/onap/policy/common/sitemanager/exception/MissingPropertyExceptionTest.java
@@ -0,0 +1,35 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * site-manager
+ * ================================================================================
+ * Copyright (C) 2018 Ericsson. All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+
+package org.onap.policy.common.sitemanager.exception;
+
+import static org.junit.Assert.assertEquals;
+import org.junit.Test;
+import org.onap.policy.common.sitemanager.exception.MissingPropertyException;
+import org.onap.policy.common.utils.test.ExceptionsTester;
+
+public class MissingPropertyExceptionTest extends ExceptionsTester {
+
+ @Test
+ public void testMissingPropertyException() throws Exception {
+ assertEquals(1, test(MissingPropertyException.class));
+ }
+
+}
diff --git a/site-manager/src/test/java/org/onap/policy/common/sitemanager/exception/NoMatchingEntryFoundExceptionTest.java b/site-manager/src/test/java/org/onap/policy/common/sitemanager/exception/NoMatchingEntryFoundExceptionTest.java
new file mode 100644
index 00000000..1b7bfa32
--- /dev/null
+++ b/site-manager/src/test/java/org/onap/policy/common/sitemanager/exception/NoMatchingEntryFoundExceptionTest.java
@@ -0,0 +1,35 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * site-manager
+ * ================================================================================
+ * Copyright (C) 2018 Ericsson. All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+
+package org.onap.policy.common.sitemanager.exception;
+
+import static org.junit.Assert.assertEquals;
+import org.junit.Test;
+import org.onap.policy.common.sitemanager.exception.NoMatchingEntryFoundException;
+import org.onap.policy.common.utils.test.ExceptionsTester;
+
+public class NoMatchingEntryFoundExceptionTest extends ExceptionsTester {
+
+ @Test
+ public void testNoMatchingEntryFoundException() throws Exception {
+ assertEquals(1, test(NoMatchingEntryFoundException.class));
+ }
+
+}
diff --git a/site-manager/src/test/java/org/onap/policy/common/sitemanager/exception/PropertyFileProcessingExceptionTest.java b/site-manager/src/test/java/org/onap/policy/common/sitemanager/exception/PropertyFileProcessingExceptionTest.java
new file mode 100644
index 00000000..5289a816
--- /dev/null
+++ b/site-manager/src/test/java/org/onap/policy/common/sitemanager/exception/PropertyFileProcessingExceptionTest.java
@@ -0,0 +1,35 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * site-manager
+ * ================================================================================
+ * Copyright (C) 2018 Ericsson. All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+
+package org.onap.policy.common.sitemanager.exception;
+
+import static org.junit.Assert.assertEquals;
+import org.junit.Test;
+import org.onap.policy.common.sitemanager.exception.PropertyFileProcessingException;
+import org.onap.policy.common.utils.test.ExceptionsTester;
+
+public class PropertyFileProcessingExceptionTest extends ExceptionsTester {
+
+ @Test
+ public void testPropertyFileProcessingException() throws Exception {
+ assertEquals(2, test(PropertyFileProcessingException.class));
+ }
+
+}
diff --git a/site-manager/src/test/java/org/onap/policy/common/sitemanager/utils/CommandLineHelperTest.java b/site-manager/src/test/java/org/onap/policy/common/sitemanager/utils/CommandLineHelperTest.java
new file mode 100644
index 00000000..d15c1dc9
--- /dev/null
+++ b/site-manager/src/test/java/org/onap/policy/common/sitemanager/utils/CommandLineHelperTest.java
@@ -0,0 +1,194 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * site-manager
+ * ================================================================================
+ * Copyright (C) 2018 Ericsson. All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+
+package org.onap.policy.common.sitemanager.utils;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+import static org.onap.policy.common.sitemanager.utils.ExtraCommandLineArgument.LOCK;
+import static org.onap.policy.common.sitemanager.utils.ExtraCommandLineArgument.SET_ADMIN_STATE;
+import static org.onap.policy.common.sitemanager.utils.ExtraCommandLineArgument.SHOW;
+import static org.onap.policy.common.sitemanager.utils.ExtraCommandLineArgument.UNLOCK;
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.List;
+import org.junit.Test;
+import org.onap.policy.common.sitemanager.exception.IllegalCommandLineArgumentException;
+
+public class CommandLineHelperTest {
+
+ private static final String STATE_NAME = "STATE_NAME";
+ private static final String RESOURCE_NAME = "RESOURCE_NAME";
+
+ @Test
+ public void test_CommandLineHelper_emptyArgs() throws IOException {
+ final String[] args = new String[] {};
+ final PrintableImpl printable = new PrintableImpl();
+ final CommandLineHelper objUnderTest = new CommandLineHelper(args, printable);
+
+ assertFalse(objUnderTest.isValid());
+ assertEquals(Arrays.asList(ErrorMessages.NO_COMMAND_SPECIFIED), printable.getResult());
+ }
+
+ @Test
+ public void test_CommandLineHelper_NullArgs() throws IOException {
+ final String[] args = null;
+ final PrintableImpl printable = new PrintableImpl();
+ final CommandLineHelper objUnderTest = new CommandLineHelper(args, printable);
+
+ assertFalse(objUnderTest.isValid());
+ assertEquals(Arrays.asList(ErrorMessages.NO_COMMAND_SPECIFIED), printable.getResult());
+ }
+
+ @Test
+ public void test_CommandLineHelper_ShowValidArgs() throws IOException {
+ final String[] args = new String[] {SHOW.getValue(), "-s", RESOURCE_NAME};
+ final PrintableImpl printable = new PrintableImpl();
+ final CommandLineHelper objUnderTest = new CommandLineHelper(args, printable);
+
+ assertTrue(objUnderTest.isValid());
+ assertTrue(printable.getResult().isEmpty());
+ }
+
+ @Test
+ public void test_CommandLineHelper_LockValidArgs() throws IOException {
+ final String[] args = new String[] {LOCK.getValue(), "-s", RESOURCE_NAME};
+ final PrintableImpl printable = new PrintableImpl();
+ final CommandLineHelper objUnderTest = new CommandLineHelper(args, printable);
+
+ assertTrue(objUnderTest.isValid());
+ assertTrue(printable.getResult().isEmpty());
+ }
+
+ @Test
+ public void test_CommandLineHelper_UnLockValidArgs() throws IOException {
+ final String[] args = new String[] {UNLOCK.getValue(), "-s", RESOURCE_NAME};
+ final PrintableImpl printable = new PrintableImpl();
+ final CommandLineHelper objUnderTest = new CommandLineHelper(args, printable);
+
+ assertTrue(objUnderTest.isValid());
+ assertTrue(printable.getResult().isEmpty());
+ }
+
+ @Test
+ public void test_CommandLineHelper_SetAdminStateValidArgs() throws IOException {
+ final String[] args = new String[] {SET_ADMIN_STATE.getValue(), RESOURCE_NAME, "-r", RESOURCE_NAME};
+ final PrintableImpl printable = new PrintableImpl();
+ final CommandLineHelper objUnderTest = new CommandLineHelper(args, printable);
+
+ assertTrue(objUnderTest.isValid());
+ assertTrue(printable.getResult().isEmpty());
+ assertEquals(RESOURCE_NAME, objUnderTest.getResourceName());
+ }
+
+ @Test
+ public void test_CommandLineHelper_SetAdminStateWithStateValidArgs() throws IOException {
+ final String[] args = new String[] {SET_ADMIN_STATE.getValue(), RESOURCE_NAME, "-s", STATE_NAME};
+ final PrintableImpl printable = new PrintableImpl();
+ final CommandLineHelper objUnderTest = new CommandLineHelper(args, printable);
+
+ assertTrue(objUnderTest.isValid());
+ assertTrue(printable.getResult().isEmpty());
+ assertEquals(STATE_NAME, objUnderTest.getSite());
+
+ assertEquals(Arrays.asList(SET_ADMIN_STATE.getValue(), RESOURCE_NAME), objUnderTest.getArgList());
+ }
+
+ @Test
+ public void test_CommandLineHelper_HelpValidArgs() throws IOException {
+ final String[] args = new String[] {"-h"};
+ final PrintableImpl printable = new PrintableImpl();
+ final CommandLineHelper objUnderTest = new CommandLineHelper(args, printable);
+ assertTrue(objUnderTest.isHelpArgumentSet());
+ }
+
+ @Test(expected = IllegalCommandLineArgumentException.class)
+ public void test_CommandLineHelper_invaidArgs_printHelp() throws IOException {
+
+ final PrintableImpl printable = new PrintableImpl();
+
+ final String[] args = new String[] {"---", ""};
+ final CommandLineHelper objUnderTest = new CommandLineHelper(args, printable);
+
+ assertFalse(objUnderTest.isValid());
+
+ }
+
+ @Test
+ public void test_CommandLineHelper_invaidArgsMissingAttributes_printHelp() throws IOException {
+
+ final PrintableImpl printable = new PrintableImpl();
+ final String[] args = new String[] {SET_ADMIN_STATE.getValue(), "-s", RESOURCE_NAME};
+ final CommandLineHelper objUnderTest = new CommandLineHelper(args, printable);
+
+ assertFalse(objUnderTest.isValid());
+ final List<String> actualMessages = printable.getResult();
+
+ assertEquals(Arrays.asList(ErrorMessages.SET_ADMIN_STATE_MISSING_NEW_STATE_VALUE), actualMessages);
+
+ }
+
+ @Test
+ public void test_CommandLineHelper_invaidShowArgsExtraAttributes_printHelp() throws IOException {
+
+ final PrintableImpl printable = new PrintableImpl();
+ final String[] args = new String[] {SHOW.getValue(), RESOURCE_NAME};
+ final CommandLineHelper objUnderTest = new CommandLineHelper(args, printable);
+
+ assertFalse(objUnderTest.isValid());
+ final List<String> actualMessages = printable.getResult();
+
+ assertEquals(Arrays.asList(ErrorMessages.SHOW_EXTRA_ARGUMENTS), actualMessages);
+
+ }
+
+ @Test
+ public void test_CommandLineHelper_invaidLockArgsExtraAttributes_printHelp() throws IOException {
+
+ final PrintableImpl printable = new PrintableImpl();
+ final String[] args = new String[] {LOCK.getValue(), RESOURCE_NAME};
+ final CommandLineHelper objUnderTest = new CommandLineHelper(args, printable);
+
+ assertFalse(objUnderTest.isValid());
+ final List<String> actualMessages = printable.getResult();
+
+ assertEquals(
+ Arrays.asList(ErrorMessages.LOCK_EXTRA_ARGUMENTS, ErrorMessages.LOCK_EITHER_S_OR_R_OPTION_IS_NEEDED),
+ actualMessages);
+
+ }
+
+ @Test
+ public void test_CommandLineHelper_invaidUnLockArgsExtraAttributes_printHelp() throws IOException {
+
+ final PrintableImpl printable = new PrintableImpl();
+ final String[] args = new String[] {UNLOCK.getValue(), RESOURCE_NAME};
+ final CommandLineHelper objUnderTest = new CommandLineHelper(args, printable);
+
+ assertFalse(objUnderTest.isValid());
+ final List<String> actualMessages = printable.getResult();
+
+ assertEquals(Arrays.asList(ErrorMessages.UNLOCK_EXTRA_ARGUMENTS,
+ ErrorMessages.UNLOCK_EITHER_S_OR_R_OPTION_IS_NEEDED), actualMessages);
+
+ }
+
+}
diff --git a/site-manager/src/test/java/org/onap/policy/common/sitemanager/utils/ExtraCommandLineArgumentTest.java b/site-manager/src/test/java/org/onap/policy/common/sitemanager/utils/ExtraCommandLineArgumentTest.java
new file mode 100644
index 00000000..b1b465b7
--- /dev/null
+++ b/site-manager/src/test/java/org/onap/policy/common/sitemanager/utils/ExtraCommandLineArgumentTest.java
@@ -0,0 +1,84 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * site-manager
+ * ================================================================================
+ * Copyright (C) 2018 Ericsson. All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+
+package org.onap.policy.common.sitemanager.utils;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+import static org.onap.policy.common.sitemanager.utils.ExtraCommandLineArgument.LOCK;
+import static org.onap.policy.common.sitemanager.utils.ExtraCommandLineArgument.SET_ADMIN_STATE;
+import static org.onap.policy.common.sitemanager.utils.ExtraCommandLineArgument.SHOW;
+import java.io.IOException;
+import java.util.Arrays;
+import org.junit.Test;
+
+public class ExtraCommandLineArgumentTest {
+
+ private static final String EMPTY_STRING = "";
+ private static final String RESOURCE_NAME = "RESOURCE_NAME";
+
+ @Test
+ public void test_ExtraCommandLineArgument_ShowValidArgs() throws IOException {
+ final PrintableImpl printable = new PrintableImpl();
+
+ final ExtraCommandLineArgument objUnderTest =
+ ExtraCommandLineArgument.getExtraCommandLineArgument(SHOW.getValue());
+
+ assertTrue(objUnderTest.isValid(Arrays.asList(SHOW.getValue()), printable));
+ assertTrue(printable.getResult().isEmpty());
+ }
+
+ @Test
+ public void test_ExtraCommandLineArgument_SetAdminStateValidArgs() throws IOException {
+ final PrintableImpl printable = new PrintableImpl();
+
+ final ExtraCommandLineArgument objUnderTest =
+ ExtraCommandLineArgument.getExtraCommandLineArgument(SET_ADMIN_STATE.getValue());
+
+ assertTrue(objUnderTest.isValid(Arrays.asList(SET_ADMIN_STATE.getValue(), RESOURCE_NAME), printable, true));
+ assertTrue(printable.getResult().isEmpty());
+ }
+
+ @Test
+ public void test_ExtraCommandLineArgument_LockStateValidArgs() throws IOException {
+ final PrintableImpl printable = new PrintableImpl();
+
+ final ExtraCommandLineArgument objUnderTest =
+ ExtraCommandLineArgument.getExtraCommandLineArgument(LOCK.getValue());
+
+ assertTrue(objUnderTest.isValid(Arrays.asList(LOCK.getValue()), printable, true));
+ assertTrue(printable.getResult().isEmpty());
+ }
+
+ @Test
+ public void test_ExtraCommandLineArgument_InValidArgs() throws IOException {
+ final PrintableImpl printable = new PrintableImpl();
+
+ final ExtraCommandLineArgument objUnderTest =
+ ExtraCommandLineArgument.getExtraCommandLineArgument(EMPTY_STRING);
+
+ assertTrue(objUnderTest.equals(ExtraCommandLineArgument.INVALID));
+ assertFalse(objUnderTest.isValid(Arrays.asList(EMPTY_STRING), printable, false));
+ assertFalse(printable.getResult().isEmpty());
+ assertEquals(Arrays.asList(ErrorMessages.UNKNOWN_COMMAND), printable.getResult());
+ }
+
+}
diff --git a/site-manager/src/test/java/org/onap/policy/common/sitemanager/utils/PersistenceUnitPropertiesProviderTest.java b/site-manager/src/test/java/org/onap/policy/common/sitemanager/utils/PersistenceUnitPropertiesProviderTest.java
new file mode 100644
index 00000000..5f4e4eb1
--- /dev/null
+++ b/site-manager/src/test/java/org/onap/policy/common/sitemanager/utils/PersistenceUnitPropertiesProviderTest.java
@@ -0,0 +1,97 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * site-manager
+ * ================================================================================
+ * Copyright (C) 2018 Ericsson. All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+
+package org.onap.policy.common.sitemanager.utils;
+
+import static org.junit.Assert.assertEquals;
+import java.io.BufferedWriter;
+import java.io.File;
+import java.io.IOException;
+import java.nio.file.Files;
+import java.util.Properties;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.rules.TemporaryFolder;
+import org.onap.policy.common.sitemanager.exception.MissingPropertyException;
+import org.onap.policy.common.sitemanager.exception.PropertyFileProcessingException;
+
+public class PersistenceUnitPropertiesProviderTest {
+ private static final String PROPERTIES_FILE_NAME = "file.properties";
+
+ private static final String COMMENTS = "";
+
+ @Rule
+ public TemporaryFolder temporaryFolder = new TemporaryFolder();
+
+ @Test(expected = PropertyFileProcessingException.class)
+ public void test_getProperties_null_throwException() {
+ final Printable printable = new PrintableImpl();
+
+ PersistenceUnitPropertiesProvider.getProperties(null, printable);
+ }
+
+ @Test(expected = PropertyFileProcessingException.class)
+ public void test_getProperties_emptyString_throwException() {
+ final Printable printable = new PrintableImpl();
+
+ PersistenceUnitPropertiesProvider.getProperties("", printable);
+ }
+
+ @Test(expected = MissingPropertyException.class)
+ public void test_getProperties_emptyPropertyFile_throwException() throws IOException {
+ final Printable printable = new PrintableImpl();
+ final File file = temporaryFolder.newFile(PROPERTIES_FILE_NAME);
+ creatPropertyFile(file, new Properties());
+ PersistenceUnitPropertiesProvider.getProperties(file.toString(), printable);
+ }
+
+ @Test(expected = MissingPropertyException.class)
+ public void test_getProperties_PropertyFileWithMissingProperties_throwException() throws IOException {
+ final Printable printable = new PrintableImpl();
+ final File file = temporaryFolder.newFile(PROPERTIES_FILE_NAME);
+ final Properties properties = new Properties();
+ properties.put(Constants.JDBC_DRIVER_PROPERTY_NAME, "org.h2");
+ creatPropertyFile(file, properties);
+ PersistenceUnitPropertiesProvider.getProperties(file.toString(), printable);
+ }
+
+ @Test
+ public void test_getProperties_PropertyFileValidProperties_throwException() throws IOException {
+ final Printable printable = new PrintableImpl();
+ final File file = temporaryFolder.newFile(PROPERTIES_FILE_NAME);
+
+ final Properties properties = new Properties();
+ properties.put(Constants.JDBC_DRIVER_PROPERTY_NAME, "Driver");
+ properties.put(Constants.JDBC_URL_PROPERTY_NAME, "inMem:Database");
+ properties.put(Constants.JDBC_USER_PROPERTY_NAME, "test");
+ properties.put(Constants.JDBC_PASSWORD_PROPERTY_NAME, "test");
+ creatPropertyFile(file, properties);
+ final Properties actualProperties = PersistenceUnitPropertiesProvider.getProperties(file.toString(), printable);
+
+ assertEquals(properties, actualProperties);
+ }
+
+ private void creatPropertyFile(final File file, final Properties properties) throws IOException {
+ try (final BufferedWriter bufferedWriter = Files.newBufferedWriter(file.toPath());) {
+ properties.store(bufferedWriter, COMMENTS);
+ }
+ }
+
+}
diff --git a/site-manager/src/test/java/org/onap/policy/common/sitemanager/utils/PrintableImpl.java b/site-manager/src/test/java/org/onap/policy/common/sitemanager/utils/PrintableImpl.java
new file mode 100644
index 00000000..dd0512fa
--- /dev/null
+++ b/site-manager/src/test/java/org/onap/policy/common/sitemanager/utils/PrintableImpl.java
@@ -0,0 +1,39 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * site-manager
+ * ================================================================================
+ * Copyright (C) 2018 Ericsson. All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+
+package org.onap.policy.common.sitemanager.utils;
+
+import java.util.ArrayList;
+import java.util.List;
+
+public class PrintableImpl implements Printable {
+
+ private final List<String> result = new ArrayList<>();
+
+ @Override
+ public void println(final String value) {
+ result.add(value);
+ }
+
+ public List<String> getResult() {
+ return result;
+ }
+
+}