line_graph_tag and test

rossmeissl · Apr 27, 2010 · 1914ec3 · 1914ec3
1 parent b892f20
commit 1914ec3
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 1 deletion.
diff --git a/lib/grafico/helpers.rb b/lib/grafico/helpers.rb
@@ -6,5 +6,9 @@ def graph_tag(graph_type, element, data, options = {})
       args << options.to_json unless options.empty?
       javascript_tag "var #{options[:variable_name] || (element + graph_type)} = new Grafico.#{graph_type}($('#{element}'), #{args.join});"
     end
+
+    def line_graph_tag(element, data, options = {})
+      graph_tag 'LineGraph', element, data, options
+    end
   end
 end
diff --git a/test/helper.rb b/test/helper.rb
@@ -10,6 +10,9 @@
 require 'action_controller'
 
 class Test::Unit::TestCase
+  def wrap_as_javascript(javascript)
+    "<script type=\"text/javascript\">\n//<![CDATA[\n#{javascript}\n//]]>\n</script>"
+  end
 end
 
 class TestHelper

diff --git a/test/test_grafico.rb b/test/test_grafico.rb
@@ -10,6 +10,10 @@ def test_setup
   end
 
   def test_graph_tag
-    assert_equal @helper.graph_tag('FooGraph', 'my_element', [1, 2, 3]), "<script type=\"text/javascript\">\n//<![CDATA[\nvar my_elementFooGraph = new Grafico.FooGraph($('my_element'), [1,2,3]);\n//]]>\n</script>"
+    assert_equal @helper.graph_tag('FooGraph', 'my_element', [1, 2, 3]), wrap_as_javascript("var my_elementFooGraph = new Grafico.FooGraph($('my_element'), [1,2,3]);")
+  end
+
+  def test_line_graph_tag
+    assert_equal @helper.line_graph_tag('my_element', [1, 2, 3]), wrap_as_javascript("var my_elementLineGraph = new Grafico.LineGraph($('my_element'), [1,2,3]);")
   end
 end