spec/spec_helper.rb in mongo_rack-0.0.2 vs spec/spec_helper.rb in mongo_rack-0.0.3

@@ -1,19 +1,21 @@
 require 'rubygems'
 require 'rack'
 require 'rack/test'
 require 'rack/response'
+require 'yaml'
 
 require File.expand_path( File.join( File.dirname(__FILE__), %w[.. lib mongo_rack] ) )
 
 Spec::Runner.configure do |config|
 end
 
 def mongo_check( res, key, val )
   session_id = res['Set-Cookie'].match( /^#{@session_key}=(.*?);.*?/ )[1]
-  ses = @sessions.find_one( { :_id => session_id } )  
-  ses.should_not be_nil
-  ses['data'][key.to_s].should == val
+  result     = @sessions.find_one( { :_id => session_id } )  
+  result.should_not be_nil
+  ses       = YAML.load( result['data'] )
+  ses[key.to_s].should == val    
 end
 
 def clear_sessions
   @sessions.remove()
 end
\ No newline at end of file