Forward discard! to the replica connection

Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    janus-ar (8.0.0)
+    janus-ar (8.0.1)
       activerecord (>= 8.0, < 9.0)
 
 GEM

lib/janus-ar/adapter_extensions.rb

@@ -74,6 +74,11 @@ def reconnect!(...)
       super
     end
 
+    def discard!(...)
+      replica_connection.discard!(...)
+      super
+    end
+
     def disconnect!(...)
       replica_connection.disconnect!(...)
       super

spec/lib/janus-ar/active_record/connection_adapters/janus_mysql_adapter_spec.rb

@@ -70,6 +70,12 @@
     end
   end
 
+  describe 'Connection lifecycle' do
+    let(:replica_adapter_class) { ActiveRecord::ConnectionAdapters::Mysql2Adapter }
+
+    it_behaves_like 'an adapter forwarding lifecycle calls'
+  end
+
   describe 'Integration tests' do
     let(:table_name) { 'table_name_mysql2' }
 

spec/lib/janus-ar/active_record/connection_adapters/janus_trilogy_adapter_spec.rb

@@ -69,6 +69,12 @@
     end
   end
 
+  describe 'Connection lifecycle' do
+    let(:replica_adapter_class) { ActiveRecord::ConnectionAdapters::TrilogyAdapter }
+
+    it_behaves_like 'an adapter forwarding lifecycle calls'
+  end
+
   describe 'Integration tests' do
     let(:table_name) { 'table_name_trilogy' }
 

spec/shared_examples/an_adapter_forwarding_lifecycle_calls.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+RSpec.shared_examples 'an adapter forwarding lifecycle calls' do
+  let(:replica_connection) { instance_double(replica_adapter_class) }
+
+  before do
+    allow(subject).to receive(:replica_connection).and_return(replica_connection)
+  end
+
+  it 'forwards discard! to the replica connection' do
+    allow(replica_connection).to receive(:discard!)
+
+    subject.discard!
+
+    expect(replica_connection).to have_received(:discard!)
+  end
+end

spec/spec_helper.rb

@@ -12,6 +12,7 @@
 ActiveRecord::ConnectionAdapters.register("janus_mysql2", "ActiveRecord::ConnectionAdapters::JanusMysql2Adapter")
 
 require './spec/shared_examples/a_mysql_like_server.rb'
+require './spec/shared_examples/an_adapter_forwarding_lifecycle_calls.rb'
 
 class QueryLogger
   def initialize