1 files changed, 96 insertions, 0 deletions
diff --git a/src/test/java/net/sf/antcontrib/cpptasks/TestXMLConsumer.java b/src/test/java/net/sf/antcontrib/cpptasks/TestXMLConsumer.java
new file mode 100644
index 0000000..5e7b007
--- /dev/null
+++ b/src/test/java/net/sf/antcontrib/cpptasks/TestXMLConsumer.java
@@ -0,0 +1,96 @@
+/*
+ * 
+ * Copyright 2003-2004 The Ant-Contrib project
+ *
+ *  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.
+ */
+package net.sf.antcontrib.cpptasks;
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+
+import junit.framework.TestCase;
+/**
+ * Base class for tests on classes that consume or public XML documents.
+ * 
+ * @author Curt Arnold
+ *  
+ */
+public abstract class TestXMLConsumer extends TestCase {
+    /**
+     * copies a resource to a temporary directory.
+     * 
+     * @param resourceName
+     *            resouce name, such as "files/openshore/history.xml".
+     * @param tmpFile name for temporary file created in /tmp or similar.
+     */
+    public static final void copyResourceToTmpDir(String resourceName,
+            String tmpFile) throws IOException {
+        String tmpDir = System.getProperty("java.io.tmpdir");
+        //
+        //  attempt to get resource from jar
+        //      (should succeed unless testing in IDE)
+        InputStream src = null;
+        if (TestTargetHistoryTable.class.getClassLoader().getResource(
+                resourceName) != null) {
+            src = TestTargetHistoryTable.class.getClassLoader()
+                    .getResourceAsStream(resourceName);
+        }
+        //
+        //  if not found, try to find it relative to the current directory
+        //
+        if (src == null) {
+            src = new FileInputStream(resourceName);
+        }
+        assertNotNull("Could not locate resource " + resourceName, src);
+        try {
+            File destFile = new File(tmpDir, tmpFile);
+            FileOutputStream dest = new FileOutputStream(destFile);
+            try {
+                int bytesRead = 0;
+                byte[] buffer = new byte[4096];
+                do {
+                    bytesRead = src.read(buffer);
+                    if (bytesRead > 0) {
+                        dest.write(buffer, 0, bytesRead);
+                    }
+                } while (bytesRead == buffer.length);
+            } finally {
+                dest.close();
+            }
+        } finally {
+            src.close();
+        }
+    }
+    /**
+     * Deletes a file, if it exists, from the user's temporary directory.
+     * 
+     * @param tmpName
+     *            file name, may not be null
+     */
+    public static void deleteTmpFile(String tmpName) throws IOException {
+        String tmpDir = System.getProperty("java.io.tmpdir");
+        File tmpFile = new File(tmpDir, tmpName);
+        if (tmpFile.exists()) {
+            tmpFile.delete();
+        }
+    }
+    /**
+     * @param testName
+     */
+    protected TestXMLConsumer(final String testName) {
+        super(testName);
+    }
+}