Add tests for OutputProcessor

test/tests/test_output_processor.rb

+class OutputProcessorTests < Test::Unit::TestCase
+  def assert_output expected, original
+    output = Brakeman::OutputProcessor.new.format original
+
+    assert_equal expected, output
+  end
+
+  def test_output_nil
+    assert_output "", nil
+  end
+
+  def test_output_empty_sexp
+    assert_output "", Sexp.new
+  end
+
+  def test_output_missing_node_type
+    assert_output "", Sexp.new(Sexp.new(:str, 'x'))
+  end
+
+  def test_output_bad_input
+    assert_output "", Sexp.new(:call, "what is this doing here", Sexp.new(:huh))
+  end
+
+  def test_output_local_variable
+    assert_output "(local x)", Sexp.new(:lvar, :x)
+  end
+
+  def test_output_ignore
+    assert_output "[ignored]", Sexp.new(:ignore, :whatever)
+  end
+
+  def test_output_params
+    assert_output "params", Sexp.new(:params, :anything)
+  end
+
+  def test_output_session
+    assert_output "session", Sexp.new(:session)
+  end
+
+  def test_output_cookies
+    assert_output "cookies[:yum]", Sexp.new(:call,
+                                            Sexp.new(:cookies),
+                                            :[],
+                                            Sexp.new(:arglist,
+                                                     Sexp.new(:lit, :yum)))
+  end
+
+  def test_output_output
+    assert_output "[Output] (local x)", Sexp.new(:output,
+                                                 Sexp.new(:lvar, :x))
+  end
+
+  def test_output_output_format
+    assert_output "", Sexp.new(:output,
+                               Sexp.new(:format, Sexp.new(:str, 'bye')))
+  end
+
+  def test_output_escaped_output
+    assert_output '[Escaped Output] @x', Sexp.new(:escaped_output,
+                                                   Sexp.new(:ivar, :@x))
+  end
+
+  def test_output_string_output
+    assert_output '', Sexp.new(:output, Sexp.new(:str, 'x'))
+    assert_output '', Sexp.new(:escaped_output, Sexp.new(:str, 'x'))
+  end
+
+  def test_output_format_string_literal
+    assert_output "", Sexp.new(:output,
+                               Sexp.new(:format, Sexp.new(:str, 'hi')))
+
+  end
+
+  def test_output_escaped_format_string_literal
+    assert_output "", Sexp.new(:escaped_output,
+                               Sexp.new(:format, Sexp.new(:str, 'hi')))
+
+  end
+
+
+  def test_output_string_interp
+    assert_output '"#{@x}"', Sexp.new(:string_interp,
+                                      "",
+                                      Sexp.new(:string_eval,
+                                               Sexp.new(:ivar, :@x)))
+  end
+
+  def test_output_format
+    assert_output "[Format] @x", Sexp.new(:format, Sexp.new(:ivar, :@x))
+  end
+
+  def test_output_format_escaped
+    assert_output "[Escaped] @x", Sexp.new(:format_escaped,
+                                            Sexp.new(:ivar, :@x))
+  end
+
+  def test_output_format_escaped_string_literal
+    assert_output "", Sexp.new(:format_escaped, Sexp.new(:str, "hi"))
+  end
+
+  def test_output_format_escaped_with_escaped_literal
+    assert_output "", Sexp.new(:format_escaped,
+                               Sexp.new(:escaped_output, Sexp.new(:str, 'hi')))
+  end
+
+
+  def test_output_format_string_literal
+    assert_output "", Sexp.new(:format, Sexp.new(:str, "hi"))
+  end
+
+  def test_output_format_escaped_literal
+    assert_output "", Sexp.new(:format,
+                               Sexp.new(:escaped_output, Sexp.new(:str, 'hi')))
+  end
+
+  def test_output_unknown_model
+    assert_output "(Unresolved Model)", Sexp.new(:const,
+                                                 Brakeman::Tracker::UNKNOWN_MODEL)
+  end
+
+  def test_output_render
+    assert_output 'render(partial => "x/y", { :locals => ({ :user => (@user) }) })',
+      Sexp.new(:render,
+               :partial,
+               Sexp.new(:str, "x/y"),
+               Sexp.new(:hash, 
+                        Sexp.new(:lit, :locals),
+                        Sexp.new(:hash,
+                                 Sexp.new(:lit, :user),
+                                 Sexp.new(:ivar, :@user))))
+  end
+
+  def test_output_rlist
+    assert_output "a\nb",
+      Sexp.new(:rlist,
+               Sexp.new(:call, nil, :a, Sexp.new(:arglist)),
+               Sexp.new(:call, nil, :b, Sexp.new(:arglist)))
+  end
+
+  def test_output_call_with_block
+    assert_output "x do\n y\n end",
+      Sexp.new(:call_with_block,
+               Sexp.new(:call, nil, :x, Sexp.new(:arglist)),
+               Sexp.new(:call, nil, :y, Sexp.new(:arglist)))
+  end
+end