--- /dev/null
+/*-
+ * ============LICENSE_START=======================================================
+ * Copyright (C) 2019 AT&T Intellectual Property. 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.
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ * ============LICENSE_END=========================================================
+ */
+
+package org.onap.policy.pdp.xacml.application.common.std;
+
+import static org.assertj.core.api.Assertions.assertThatThrownBy;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertNull;
+import static org.junit.Assert.assertSame;
+import static org.junit.Assert.assertTrue;
+import static org.mockito.Matchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.times;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+import com.att.research.xacml.api.Request;
+import com.att.research.xacml.api.Response;
+import com.att.research.xacml.api.pdp.PDPEngine;
+import com.att.research.xacml.api.pdp.PDPEngineFactory;
+import com.att.research.xacml.api.pdp.PDPException;
+import com.att.research.xacml.util.FactoryException;
+import com.att.research.xacml.util.XACMLProperties;
+import com.google.common.io.Files;
+import java.io.File;
+import java.nio.file.Path;
+import java.util.HashSet;
+import java.util.Properties;
+import java.util.Set;
+import oasis.names.tc.xacml._3_0.core.schema.wd_17.PolicyType;
+import org.apache.commons.lang3.tuple.Pair;
+import org.junit.AfterClass;
+import org.junit.Before;
+import org.junit.BeforeClass;
+import org.junit.Test;
+import org.mockito.Mock;
+import org.mockito.MockitoAnnotations;
+import org.onap.policy.models.decisions.concepts.DecisionRequest;
+import org.onap.policy.models.decisions.concepts.DecisionResponse;
+import org.onap.policy.models.tosca.authorative.concepts.ToscaPolicy;
+import org.onap.policy.pdp.xacml.application.common.ToscaPolicyConversionException;
+import org.onap.policy.pdp.xacml.application.common.ToscaPolicyTranslator;
+import org.onap.policy.pdp.xacml.application.common.XacmlApplicationException;
+import org.onap.policy.pdp.xacml.application.common.XacmlPolicyUtils;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+public class StdXacmlApplicationServiceProviderTest {
+ private static final Logger logger = LoggerFactory.getLogger(StdXacmlApplicationServiceProviderTest.class);
+
+ private static final String TEMP_DIR_NAME = "src/test/resources/temp";
+ private static File TEMP_DIR = new File(TEMP_DIR_NAME);
+ private static Path TEMP_PATH = TEMP_DIR.toPath();
+ private static File SOURCE_PROP_FILE = new File("src/test/resources/test.properties");
+ private static File PROP_FILE = new File(TEMP_DIR, XacmlPolicyUtils.XACML_PROPERTY_FILE);
+ private static final String EXPECTED_EXCEPTION = "expected exception";
+ private static final String POLICY_NAME = "my-name";
+ private static final String POLICY_VERSION = "1.2.3";
+ private static final String POLICY_TYPE = "my-type";
+
+ @Mock
+ private ToscaPolicyTranslator trans;
+
+ @Mock
+ private PDPEngineFactory engineFactory;
+
+ @Mock
+ private PDPEngine engine;
+
+ @Mock
+ private Request req;
+
+ @Mock
+ private Response resp;
+
+ private ToscaPolicy policy;
+ private PolicyType internalPolicy;
+
+ private StdXacmlApplicationServiceProvider prov;
+
+ /**
+ * Creates the temp directory.
+ */
+ @BeforeClass
+ public static void setUpBeforeClass() {
+ assertTrue(TEMP_DIR.mkdir());
+ }
+
+ /**
+ * Deletes the temp directory and its contents.
+ */
+ @AfterClass
+ public static void tearDownAfterClass() {
+ for (File file : TEMP_DIR.listFiles()) {
+ if (!file.delete()) {
+ logger.warn("cannot delete: {}", file);
+ }
+ }
+
+ if (!TEMP_DIR.delete()) {
+ logger.warn("cannot delete: {}", TEMP_DIR);
+ }
+ }
+
+ /**
+ * Initializes objects, including the provider.
+ *
+ * @throws Exception if an error occurs
+ */
+ @Before
+ public void setUp() throws Exception {
+ MockitoAnnotations.initMocks(this);
+
+ policy = new ToscaPolicy();
+ policy.setType(POLICY_TYPE);
+ policy.setName(POLICY_NAME);
+ policy.setVersion(POLICY_VERSION);
+
+ internalPolicy = new PolicyType();
+ internalPolicy.setPolicyId(POLICY_NAME);
+ internalPolicy.setVersion(POLICY_VERSION);
+
+ when(engineFactory.newEngine(any())).thenReturn(engine);
+
+ when(engine.decide(req)).thenReturn(resp);
+
+ when(trans.convertPolicy(policy)).thenReturn(internalPolicy);
+
+ prov = new MyProv();
+
+ Files.copy(SOURCE_PROP_FILE, PROP_FILE);
+ }
+
+ @Test
+ public void testApplicationName() {
+ assertNotNull(prov.applicationName());
+ }
+
+ @Test
+ public void testActionDecisionsSupported() {
+ assertTrue(prov.actionDecisionsSupported().isEmpty());
+ }
+
+ @Test
+ public void testInitialize_testGetXxx() throws XacmlApplicationException {
+ prov.initialize(TEMP_PATH);
+
+ assertEquals(TEMP_PATH, prov.getDataPath());
+ assertNotNull(prov.getEngine());
+
+ Properties props = prov.getProperties();
+ assertEquals("rootstart", props.getProperty("xacml.rootPolicies"));
+ }
+
+ @Test
+ public void testInitialize_Ex() throws XacmlApplicationException {
+ assertThatThrownBy(() -> prov.initialize(new File(TEMP_DIR_NAME + "-nonExistent").toPath()))
+ .isInstanceOf(XacmlApplicationException.class).hasMessage("Failed to load xacml.properties");
+ }
+
+ @Test
+ public void testSupportedPolicyTypes() {
+ assertThatThrownBy(() -> prov.supportedPolicyTypes()).isInstanceOf(UnsupportedOperationException.class);
+ }
+
+ @Test
+ public void testCanSupportPolicyType() {
+ assertThatThrownBy(() -> prov.canSupportPolicyType(null)).isInstanceOf(UnsupportedOperationException.class);
+ }
+
+ @Test
+ public void testLoadPolicy_ConversionError() throws ToscaPolicyConversionException {
+ when(trans.convertPolicy(policy)).thenReturn(null);
+
+ assertFalse(prov.loadPolicy(policy));
+ }
+
+ @Test
+ public void testLoadPolicy_testUnloadPolicy() throws Exception {
+ prov.initialize(TEMP_PATH);
+ PROP_FILE.delete();
+
+ final Set<String> set = XACMLProperties.getRootPolicyIDs(prov.getProperties());
+
+ assertTrue(prov.loadPolicy(policy));
+
+ // policy file should have been created
+ File policyFile = new File(TEMP_DIR, "my-name_1.2.3.xml");
+ assertTrue(policyFile.exists());
+
+ // new property file should have been created
+ assertTrue(PROP_FILE.exists());
+
+ // should have re-created the engine
+ verify(engineFactory, times(2)).newEngine(any());
+
+ final Set<String> set2 = XACMLProperties.getRootPolicyIDs(prov.getProperties());
+ assertEquals(set.size() + 1, set2.size());
+
+ Set<String> set3 = new HashSet<>(set2);
+ set3.removeAll(set);
+ assertEquals("[root1]", set3.toString());
+
+
+ /*
+ * Prepare for unload.
+ */
+ PROP_FILE.delete();
+
+ assertTrue(prov.unloadPolicy(policy));
+
+ // policy file should have been removed
+ assertFalse(policyFile.exists());
+
+ // new property file should have been created
+ assertTrue(PROP_FILE.exists());
+
+ // should have re-created the engine
+ verify(engineFactory, times(3)).newEngine(any());
+
+ set3 = XACMLProperties.getRootPolicyIDs(prov.getProperties());
+ assertEquals(set.toString(), set3.toString());
+ }
+
+ @Test
+ public void testUnloadPolicy_NotDeployed() throws Exception {
+ prov.initialize(TEMP_PATH);
+
+ assertFalse(prov.unloadPolicy(policy));
+
+ // no additional calls
+ verify(engineFactory, times(1)).newEngine(any());
+ }
+
+ @Test
+ public void testMakeDecision() {
+ prov.createEngine(null);
+
+ DecisionRequest decreq = mock(DecisionRequest.class);
+ when(trans.convertRequest(decreq)).thenReturn(req);
+
+ DecisionResponse decresp = mock(DecisionResponse.class);
+ when(trans.convertResponse(resp)).thenReturn(decresp);
+
+ Pair<DecisionResponse, Response> result = prov.makeDecision(decreq);
+ assertSame(decresp, result.getKey());
+ assertSame(resp, result.getValue());
+
+ verify(trans).convertRequest(decreq);
+ verify(trans).convertResponse(resp);
+ }
+
+ @Test
+ public void testGetTranslator() {
+ assertSame(trans, prov.getTranslator());
+ }
+
+ @Test
+ public void testCreateEngine() throws FactoryException {
+ // success
+ prov.createEngine(null);
+ assertSame(engine, prov.getEngine());
+
+ // null - should be unchanged
+ when(engineFactory.newEngine(any())).thenReturn(null);
+ prov.createEngine(null);
+ assertSame(engine, prov.getEngine());
+
+ // exception - should be unchanged
+ when(engineFactory.newEngine(any())).thenThrow(new FactoryException(EXPECTED_EXCEPTION));
+ prov.createEngine(null);
+ assertSame(engine, prov.getEngine());
+ }
+
+ @Test
+ public void testXacmlDecision() throws PDPException {
+ prov.createEngine(null);
+
+ // success
+ assertSame(resp, prov.xacmlDecision(req));
+
+ // exception
+ when(engine.decide(req)).thenThrow(new PDPException(EXPECTED_EXCEPTION));
+ assertNull(prov.xacmlDecision(req));
+ }
+
+ @Test
+ public void testGetPdpEngineFactory() throws XacmlApplicationException {
+ // use the real engine factory
+ engineFactory = null;
+
+ prov = new MyProv();
+ prov.initialize(TEMP_PATH);
+
+ assertNotNull(prov.getEngine());
+ }
+
+ private class MyProv extends StdXacmlApplicationServiceProvider {
+
+ @Override
+ protected ToscaPolicyTranslator getTranslator(String type) {
+ return trans;
+ }
+
+ @Override
+ protected PDPEngineFactory getPdpEngineFactory() throws FactoryException {
+ return (engineFactory != null ? engineFactory : super.getPdpEngineFactory());
+ }
+ }
+}