Then /^I should see "(.*)"$/ do |text|
#  webrat_session.wait_for :timeout => 5 do
    webrat_session.response.body.should contain(/#{text}/m)
#  end
end

Then /^I should not see "(.*)"$/ do |text|
#  webrat_session.wait_for :timeout => 5 do
    webrat_session.response.body.should_not contain(/#{text}/m)
#  end
end

Then /^I should see (a|an)? "(.*)" message$/ do |_, message_type|
#  webrat_session.wait_for :timeout => 5 do
    webrat_session.response.body.should have_selector(".#{message_type.gsub(' ', '-')}")
#  end
end

Then /^I should not see (a|an)? "(.*)" message$/ do |_, message_type|
#  webrat_session.wait_for :timeout => 5 do
    webrat_session.response.body.should_not have_selector(".#{message_type.gsub(' ', '-')}")
#  end
end