Test with some simple html for #body

lib/webcat/culerity_driver.rb

@@ -28,7 +28,7 @@ class Webcat::Driver::Culerity
   end
   
   def body
-    browser.text
+    browser.html
   end
 
 private

spec/drivers_spec.rb

@@ -2,8 +2,8 @@ require File.expand_path('spec_helper', File.dirname(__FILE__))
 
 shared_examples_for 'driver' do
   
-  describe '#get' do
-    it "should fetch a response" do
+  describe '#visit' do
+    it "should move to another page" do
       @driver.visit('/')
       @driver.body.should == 'Hello world!'
       @driver.visit('/foo')
@@ -11,4 +11,16 @@ shared_examples_for 'driver' do
     end
   end
   
+  describe '#body' do
+    it "should return text reponses" do
+      @driver.visit('/')
+      @driver.body.should == 'Hello world!'
+    end
+    
+    it "should return the full response html" do
+      @driver.visit('/with_simple_html')
+      @driver.body.should == '<h1>Bar</h1>'
+    end
+  end
+  
 end

spec/spec_helper.rb

@@ -1,18 +1,10 @@
 $:.unshift(File.expand_path('../lib', File.dirname(__FILE__)))
+$:.unshift(File.dirname(__FILE__))
 
 require 'rubygems'
 require 'webcat'
 require 'sinatra/base'
 require 'rack'
+require 'test_app'
 
 alias :running :lambda
-
-class TestApp < Sinatra::Base
-  get '/' do
-    'Hello world!'
-  end
-
-  get '/foo' do
-    'Another World'
-  end
-end

spec/test_app.rb

@@ -0,0 +1,15 @@
+class TestApp < Sinatra::Base
+  set :views, File.dirname(__FILE__) + '/views'
+
+  get '/' do
+    'Hello world!'
+  end
+
+  get '/foo' do
+    'Another World'
+  end
+  
+  get '/with_simple_html' do
+    erb :with_simple_html
+  end
+end

spec/views/with_simple_html.erb

@@ -0,0 +1 @@
+<h1>Bar</h1>