Fix SqlBypass.data_column= problem. SqlBypass.find_by_session_id method didn't use this assignment.

1 files changed, 14 insertions, 0 deletions

diff --git a/activerecord/test/cases/session_store/sql_bypass_test.rb b/activerecord/test/cases/session_store/sql_bypass_test.rb
index 6749d4ce98..b8cf4cf2cc 100644
--- a/activerecord/test/cases/session_store/sql_bypass_test.rb
+++ b/activerecord/test/cases/session_store/sql_bypass_test.rb
@@ -56,6 +56,20 @@ module ActiveRecord
         s.destroy
         assert_nil SqlBypass.find_by_session_id session_id
       end
+
+      def test_data_column
+        SqlBypass.drop_table! if exists = Session.table_exists?
+        old, SqlBypass.data_column = SqlBypass.data_column, 'foo'
+        SqlBypass.create_table!
+
+        session_id = 20
+        SqlBypass.new(:data => 'hello', :session_id => session_id).save
+        assert_equal 'hello', SqlBypass.find_by_session_id(session_id).data
+      ensure
+        SqlBypass.drop_table!
+        SqlBypass.data_column = old
+        SqlBypass.create_table! if exists
+      end
     end
   end
 end