Support building with Gradle in POIFileScanner tests

git-svn-id: https://svn.apache.org/repos/asf/poi/trunk@1875849 13f79535-47bb-0310-9956-ffa450edef68

1 files changed, 12 insertions, 2 deletions

diff --git a/src/integrationtest/org/apache/poi/POIFileScanner.java b/src/integrationtest/org/apache/poi/POIFileScanner.java
index db6030f738..a8220ec8e7 100644
--- a/src/integrationtest/org/apache/poi/POIFileScanner.java
+++ b/src/integrationtest/org/apache/poi/POIFileScanner.java
@@ -51,6 +51,16 @@ import static org.junit.Assert.assertEquals;
  * Can also be used to get the appropriate FileHandler for a single file.
  */
 public class POIFileScanner {
+    private final static File ROOT_DIR;
+    static {
+        // when running in Gradle, current directory might be "build/integrationtest"
+        if(new File("../../test-data").exists()) {
+            ROOT_DIR = new File("../../test-data");
+        } else {
+            ROOT_DIR = new File("test-data");
+        }
+    }
+
     /**
      * Scan a folder for files and return a collection of
      * found files together with the matching {@link FileHandler}.
@@ -195,11 +205,11 @@ public class POIFileScanner {
 
     @Test
     public void testDetectUnnamedFile() throws IOException {
-        POIFileScanner.detectUnnamedFile(new File("test-data/spreadsheet"), "49156.xlsx");
+        POIFileScanner.detectUnnamedFile(new File(ROOT_DIR, "spreadsheet"), "49156.xlsx");
     }
 
     @Test
     public void test() throws IOException {
-        POIFileScanner.scan(new File("test-data"));
+        POIFileScanner.scan(ROOT_DIR);
     }
 }