test/javax/xml/jaxp/unittest/stream/XMLEventReaderTest/Bug6555001.java

@@ -20,30 +20,37 @@
  * or visit www.oracle.com if you need additional information or have any
  * questions.
  */
 package stream.XMLEventReaderTest;
 
+import java.io.FilePermission;
 import java.io.StringReader;
 
 import javax.xml.stream.XMLEventReader;
 import javax.xml.stream.XMLInputFactory;
 import javax.xml.stream.events.EntityReference;
 import javax.xml.stream.events.XMLEvent;
 
+import jaxp.library.JAXPTestUtilities;
+
 import org.testng.Assert;
+import org.testng.annotations.Listeners;
 import org.testng.annotations.Test;
 
 /*
  * @bug 6555001
  * @summary Test StAX parser replaces the entity reference as setting.
  */
+@Listeners({ jaxp.library.BasePolicy.class })
 public class Bug6555001 {
-    private static final String XML = "" + "<!DOCTYPE doc SYSTEM 'file:///tmp/this/does/not/exist/but/that/is/ok' [" + "<!ENTITY def '<para/>'>" + "]>"
-            + "<doc>&def;&undef;</doc>";
+    private static final String XML = ""
+            + "<!DOCTYPE doc SYSTEM 'file:///tmp/this/does/not/exist/but/that/is/ok' ["
+            + "<!ENTITY def '<para/>'>" + "]>" + "<doc>&def;&undef;</doc>";
 
     @Test
     public void testReplacing() throws Exception {
+        JAXPTestUtilities.tryRunWithTmpPermission(() -> {
         XMLInputFactory factory = XMLInputFactory.newInstance();
         factory.setProperty("javax.xml.stream.isReplacingEntityReferences", true);
 
         StringReader sr = new StringReader(XML);
         XMLEventReader reader = factory.createXMLEventReader(sr);

@@ -67,14 +74,16 @@
         }
 
         Assert.assertEquals(false, sawDef);
         Assert.assertEquals(true, sawUndef);
         reader.close();
+        }, new FilePermission("/tmp/this/does/not/exist/but/that/is/ok", "read"));
     }
 
     @Test
     public void testNotReplacing() throws Exception {
+        JAXPTestUtilities.tryRunWithTmpPermission(() -> {
         XMLInputFactory factory = XMLInputFactory.newInstance();
         factory.setProperty("javax.xml.stream.isReplacingEntityReferences", false);
 
         StringReader sr = new StringReader(XML);
         XMLEventReader reader = factory.createXMLEventReader(sr);

@@ -98,7 +107,8 @@
         }
 
         Assert.assertEquals(true, sawDef);
         Assert.assertEquals(true, sawUndef);
         reader.close();
+        }, new FilePermission("/tmp/this/does/not/exist/but/that/is/ok", "read"));
     }
 }