Fix method visibility issues in transaction_connection_validator specs

jeremyevans · Jan 30, 2024 · 35a2922 · 35a2922
1 parent 15cdea0
commit 35a2922
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/spec/extensions/transaction_connection_validator_spec.rb b/spec/extensions/transaction_connection_validator_spec.rb
@@ -10,16 +10,17 @@ def post_execute(conn, sql); end
       def disconnect_connection(conn)
         @sqls << 'disconnect'
       end
+      def connect(server)
+        @sqls << 'connect'
+        super
+      end
+      private
       define_method(:database_error_classes) do
         [database_error]
       end
       def disconnect_error?(e, opts)
         e.message.include? 'disconnect'
       end
-      def connect(server)
-        @sqls << 'connect'
-        super
-      end
       def log_connection_execute(conn, sql)
         res = super
         post_execute(conn, sql)