Handle BEGIN and END blocks properly

lib/ripper_ruby_parser/sexp_processor.rb

@@ -129,6 +129,16 @@ def process_comment exp
       sexp
     end
 
+    def process_BEGIN exp
+      _, body = exp.shift 2
+      s(:iter, s(:preexe), s(:args), *map_body(body))
+    end
+
+    def process_END exp
+      _, body = exp.shift 2
+      s(:iter, s(:postexe), s(:args), *map_body(body))
+    end
+
     # number literals
     def process_at_int exp
       make_literal(exp) {|val| Integer(val) }

test/unit/parser_test.rb

@@ -685,6 +685,20 @@
       end
     end
 
+    describe "for the END keyword" do
+      it "converts to a :postexe iterator" do
+        "END { foo }".
+          must_be_parsed_as s(:iter, s(:postexe), s(:args), s(:call, nil, :foo))
+      end
+    end
+
+    describe "for the BEGIN keyword" do
+      it "converts to a :preexe iterator" do
+        "BEGIN { foo }".
+          must_be_parsed_as s(:iter, s(:preexe), s(:args), s(:call, nil, :foo))
+      end
+    end
+
     describe "for constant lookups" do
       it "works when explicitely starting from the root namespace" do
         "::Foo".