Handle custom session_class in trim task

Better approach to #178

Author: ghiculescu

lib/tasks/database.rake

@@ -8,13 +8,13 @@ namespace 'db:sessions' do
 
   desc "Clear the sessions table"
   task :clear => [:environment, 'db:load_config'] do
-    ActiveRecord::Base.connection.execute "TRUNCATE TABLE #{ActiveRecord::SessionStore::Session.table_name}"
+    ActiveRecord::Base.connection.truncate(ActionDispatch::Session::ActiveRecordStore.session_class.table_name)
   end
 
   desc "Trim old sessions from the table (default: > 30 days)"
   task :trim => [:environment, 'db:load_config'] do
     cutoff_period = (ENV['SESSION_DAYS_TRIM_THRESHOLD'] || 30).to_i.days.ago
-    ActiveRecord::SessionStore::Session.
+    ActionDispatch::Session::ActiveRecordStore.session_class.
       where("updated_at < ?", cutoff_period).
       delete_all
   end

test/tasks/database_rake_test.rb

@@ -63,6 +63,14 @@ def test_upgrade_task
         assert Session.find_by_session_id(Rack::Session::SessionId.new("original_session_id").private_id)
         assert Session.find_by_session_id("2::secure_session_id")
       end
+
+      def test_clear_task
+        Session.create!(data: "obsolete")
+
+        Rake.application.invoke_task 'db:sessions:clear'
+
+        assert_equal 0, Session.count
+      end
     end
   end
 end