Add tests for RenderPath

test/tests/render_path.rb

+class RenderPathTests < Test::Unit::TestCase
+  def setup
+    @r = Brakeman::RenderPath.new
+  end
+
+  def test_include_controller
+    @r.add_controller_render :TestController, :test
+
+    assert @r.include_controller? :TestController
+  end
+
+  def test_rendered_from_controller
+    @r.add_controller_render :TestController, :test
+
+    assert @r.rendered_from_controller?
+  end
+
+  def test_include_template
+    @r.add_template_render 'some/template'
+
+    assert @r.include_template? :'some/template'
+  end
+
+  def test_include_any_method
+    @r.add_controller_render :TestController, :test
+    @r.add_controller_render :TestController, :test2
+    @r.add_controller_render :TestController, :test3
+
+    assert @r.include_any_method? ['test']
+  end
+
+  def test_each
+    @r.add_controller_render :TestController, :test
+    @r.add_template_render 'some/template'
+
+    @r.each do |loc|
+      case loc[:type]
+      when :template
+        assert_equal :'some/template', loc[:name]
+      when :controller
+        assert_equal :TestController, loc[:class]
+        assert_equal :test, loc[:method]
+      end
+    end
+  end
+
+  def test_dup
+    @r.add_controller_render :TestController, :test
+
+    s = @r.dup
+    s.add_template_render 'some/template'
+
+    assert_equal 1, @r.length
+    assert_equal 2, s.length
+  end
+end