From 841f915f874dbeceb1b0ab60d1227c28ca62b98f Mon Sep 17 00:00:00 2001
From: Justin Collins <justin@presidentbeef.com>
Date: Tue, 13 Nov 2012 10:56:46 -0800
Subject: [PATCH] Fix remove_method in RescanTestHelper

---
 test/test.rb | 16 ++++++----------
 1 file changed, 6 insertions(+), 10 deletions(-)

diff --git a/test/test.rb b/test/test.rb
index 872820c6..6cda7a39 100644
--- a/test/test.rb
+++ b/test/test.rb
@@ -219,18 +219,14 @@ module BrakemanTester::RescanTestHelper
     replace_with_sexp file do |parsed|
       class_body = parsed.body
 
-      if class_body[1].node_type == :block
-        class_body[1].reject! do |node|
-          node.is_a? Sexp and
-          node.node_type == :defn and
-          node.method_name == method_name
-        end
-      elsif class_body[1].node_type == :defn and
-        class_body[1].method_name == method_name
-
-        class_body.delete_at 1
+      class_body.reject! do |node|
+        node.is_a? Sexp and
+        node.node_type == :defn and
+        node.method_name == method_name
       end
 
+      parsed.body = class_body
+
       parsed
     end
   end
-- 
GitLab