svn commit: r1521211 - in /pig/trunk/test/org/apache/pig/tools: ./ grunt/ grunt/TestGruntParser.java

[da...@apache.org]

Author: daijy
Date: Mon Sep  9 17:32:57 2013
New Revision: 1521211

URL: http://svn.apache.org/r1521211
Log:
PIG-3426: Add support for removing s3 files

Added:
    pig/trunk/test/org/apache/pig/tools/
    pig/trunk/test/org/apache/pig/tools/grunt/
    pig/trunk/test/org/apache/pig/tools/grunt/TestGruntParser.java

Added: pig/trunk/test/org/apache/pig/tools/grunt/TestGruntParser.java
URL: http://svn.apache.org/viewvc/pig/trunk/test/org/apache/pig/tools/grunt/TestGruntParser.java?rev=1521211&view=auto
==============================================================================
--- pig/trunk/test/org/apache/pig/tools/grunt/TestGruntParser.java (added)
+++ pig/trunk/test/org/apache/pig/tools/grunt/TestGruntParser.java Mon Sep  9 17:32:57 2013
@@ -0,0 +1,54 @@
+package org.apache.pig.tools.grunt;
+
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+import static org.junit.Assert.fail;
+
+import java.io.ByteArrayInputStream;
+import java.io.File;
+import java.io.IOException;
+
+import org.apache.pig.ExecType;
+import org.apache.pig.PigServer;
+import org.apache.pig.tools.grunt.GruntParser.ExplainState;
+import org.junit.Before;
+import org.junit.Test;
+
+public class TestGruntParser {
+    private GruntParser parser;
+    static PigServer pig;
+    
+    @Before
+    public void setup() throws IOException {
+        pig = new PigServer(ExecType.LOCAL);
+    }
+    
+    @Test
+    public void testProcessRemove() throws IOException {
+        File tmpFile = File.createTempFile("TestGruntParser", "testProcessRemove");
+        String dummyScript = "";
+        parser = new GruntParser(new ByteArrayInputStream(dummyScript.getBytes()));
+        parser.setParams(pig);
+        
+        //Delete existing file and check that it doesn't exist
+        parser.processRemove(tmpFile.getAbsolutePath(), "");
+        assertFalse(tmpFile.exists());
+        
+        //Delete non-existing file and check for exception
+        try {
+            parser.processRemove(tmpFile.getAbsolutePath(), "");
+            fail("processRemove should throw exception when deleting missing file.");
+        } catch (IOException e) {
+            //Yay
+        }
+        
+        //Delete non-existing file with force, no exception
+        parser.processRemove(tmpFile.getAbsolutePath(), "force");
+        
+        //Ensure that remove isn't processed when in explain mode
+        tmpFile = File.createTempFile("TestGruntParser", "testProcessRemove");
+        parser.setExplainState(new ExplainState("","","",false,""));
+        parser.processRemove(tmpFile.getAbsolutePath(), "force");
+        assertTrue(tmpFile.exists());
+    }
+}