Add unit test for response templating

knuton · Jan 8, 2012 · 78a1310 · 78a1310
1 parent 0f73fa3
commit 78a1310
Showing 1 changed file with 13 additions and 7 deletions.
diff --git a/test/test_response.rb b/test/test_response.rb
@@ -4,20 +4,26 @@
 
 class TestResponse < Test::Unit::TestCase
 
-  def setup
+  def test_yaml_frontmatter
-    @response = SecondMate::Response.new(
+    response = SecondMate::Response.new(
       "---\nstatus: 201\nheader:\n  Foo: Baz\n---\nBody",
       {},
       200,
       {'Foo' => 'Bar'}
-    )
+    ).finish
-  end
-
-  def test_yaml_frontmatter
-    response = @response.finish
     assert_equal ['Body'], response.last.body
     assert_equal 'Baz', response[1]['Foo']
   end
 
+  def test_templating
+    response = SecondMate::Response.new(
+      "<%= params['foo'] %>",
+      {'foo' => 'Bar'},
+      200,
+      {}
+    ).finish
+    assert_equal ['Bar'], response.last.body
+  end
+
 end