Refactoring: Allow re-usage of existing DB connection instead of creating/resetting existing one.

lib/sessions/event/base.rb

@@ -12,8 +12,13 @@ class Sessions::Event::Base
 
     return if !self.class.instance_variable_get(:@database_connection)
 
+    if ActiveRecord::Base.connected?
+      @reused_connection = true
+    else
+      @reused_connection = false
       ActiveRecord::Base.establish_connection
     end
+  end
 
   def self.inherited(subclass)
     subclass.instance_variable_set(:@database_connection, @database_connection)
@@ -138,6 +143,7 @@ class Sessions::Event::Base
   def destroy
     return if !@is_web_socket
     return if !self.class.instance_variable_get(:@database_connection)
+    return if @reused_connection
 
     ActiveRecord::Base.remove_connection
   end

test/unit/chat_test.rb

@@ -58,6 +58,9 @@ class ChatTest < ActiveSupport::TestCase
     # with websockets
     assert(User.first)
 
+    # make sure to emulate unconnected WS env
+    ActiveRecord::Base.remove_connection
+
     message = Sessions::Event.run(
       event: 'login',
       payload: {},