X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=ONAP-XACML%2Fsrc%2Ftest%2Fjava%2Forg%2Fonap%2Fpolicy%2Fxacml%2Ftest%2Futil%2FXACMLPolicyWriterTest.java;h=446a87d1aa85d8846b37fc205512d199eca73601;hb=9a1113e4128f1585dbf85aabc429e9da385a76a8;hp=c948716dcfb49a1eb9ea26d635b37ccdf96ecf29;hpb=073cc188efe9abb4c010cf674e34e2cf46ef1c52;p=policy%2Fengine.git

diff --git a/ONAP-XACML/src/test/java/org/onap/policy/xacml/test/util/XACMLPolicyWriterTest.java b/ONAP-XACML/src/test/java/org/onap/policy/xacml/test/util/XACMLPolicyWriterTest.java
index c948716dc..446a87d1a 100644
--- a/ONAP-XACML/src/test/java/org/onap/policy/xacml/test/util/XACMLPolicyWriterTest.java
+++ b/ONAP-XACML/src/test/java/org/onap/policy/xacml/test/util/XACMLPolicyWriterTest.java
@@ -1,8 +1,8 @@
-/*-
+/*
  * ============LICENSE_START=======================================================
  * ONAP-XACML
  * ================================================================================
- * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
+ * Copyright (C) 2017-2018 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.
@@ -19,10 +19,13 @@
  */
 package org.onap.policy.xacml.test.util;
 
+import static org.junit.Assert.assertNotNull;
 import static org.junit.Assert.assertTrue;
 
+import java.io.ByteArrayOutputStream;
 import java.io.File;
 import java.io.IOException;
+import java.io.OutputStream;
 import java.nio.file.Path;
 
 import org.apache.commons.logging.Log;
@@ -31,32 +34,60 @@ import org.junit.Before;
 import org.junit.Test;
 import org.onap.policy.xacml.util.XACMLPolicyWriter;
 
+import oasis.names.tc.xacml._3_0.core.schema.wd_17.PolicySetType;
+import oasis.names.tc.xacml._3_0.core.schema.wd_17.PolicyType;
+
 public class XACMLPolicyWriterTest {
-	private static final Log logger				= LogFactory.getLog(XACMLPolicyWriterTest.class);
-	private static Path configPolicyPathValue;
-	private static Path actionPolicyPathValue;
-	
-	@Before
-	public void setUp() {
-		File templateFile;
-		ClassLoader classLoader = getClass().getClassLoader();
-        try {
-        	templateFile = new File(classLoader.getResource("Config_SampleTest1206.1.xml").getFile());
-        	configPolicyPathValue = templateFile.toPath();
-        	templateFile = new File(classLoader.getResource("Action_TestActionPolicy.1.xml").getFile());
-        	actionPolicyPathValue = templateFile.toPath();
-		} catch (Exception e1) {
-			logger.error("Exception Occured"+e1);
-		}
-	}
-	
-	@SuppressWarnings("static-access")
-	@Test
-	public void xacmlPolicyWriterTest() throws IOException{
-		XACMLPolicyWriter writer = new XACMLPolicyWriter();
-		String configResponseValue = writer.changeFileNameInXmlWhenRenamePolicy(configPolicyPathValue);
-		assertTrue(configResponseValue.equals("txt"));
-		String actionResponseValue = writer.changeFileNameInXmlWhenRenamePolicy(actionPolicyPathValue);
-		assertTrue(actionResponseValue.equals("json"));
-	}
+  private static final Log logger = LogFactory.getLog(XACMLPolicyWriterTest.class);
+  private static Path configPolicyPathValue;
+  private static Path actionPolicyPathValue;
+
+  @Before
+  public void setUp() {
+    File templateFile;
+    ClassLoader classLoader = getClass().getClassLoader();
+    try {
+      templateFile = new File(classLoader.getResource("Config_SampleTest1206.1.xml").getFile());
+      configPolicyPathValue = templateFile.toPath();
+      templateFile = new File(classLoader.getResource("Action_TestActionPolicy.1.xml").getFile());
+      actionPolicyPathValue = templateFile.toPath();
+    } catch (Exception e1) {
+      logger.error("Exception Occured" + e1);
+    }
+  }
+
+  @SuppressWarnings("static-access")
+  @Test
+  public void xacmlPolicyWriterTest() throws IOException {
+    XACMLPolicyWriter writer = new XACMLPolicyWriter();
+    String configResponseValue = writer.changeFileNameInXmlWhenRenamePolicy(configPolicyPathValue);
+    assertTrue(configResponseValue.equals("txt"));
+    String actionResponseValue = writer.changeFileNameInXmlWhenRenamePolicy(actionPolicyPathValue);
+    assertTrue(actionResponseValue.equals("json"));
+  }
+
+  @Test
+  public void testWrites() throws Exception {
+    // Set up test data
+    PolicyType policyType = new PolicyType();
+    OutputStream os = new ByteArrayOutputStream();
+    File tmpfile = File.createTempFile("foo", null);
+    Path filename = tmpfile.toPath();
+    PolicySetType policySet = new PolicySetType();
+    
+    // delete tmp file before running the test
+    tmpfile.delete();
+
+    // ensure its deleted after writePolicyFile() re-creates it
+    tmpfile.deleteOnExit();
+
+    // Test write combinations
+    assertNotNull(XACMLPolicyWriter.writePolicyFile(filename, policySet));
+    assertNotNull(XACMLPolicyWriter.writePolicyFile(filename, policyType));
+    assertNotNull(XACMLPolicyWriter.getXmlAsInputStream(policyType));
+    XACMLPolicyWriter.writePolicyFile(os, policySet);
+    assertNotNull(os.toString());
+    XACMLPolicyWriter.writePolicyFile(os, policyType);
+    assertNotNull(os.toString());
+  }
 }