Allow adapter to receive :execute with other statements, too.

PNixx · Nov 22, 2024 · 90082f3 · 90082f3
1 parent 4ccddc0
commit 90082f3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/spec/cluster/migration_spec.rb b/spec/cluster/migration_spec.rb
@@ -129,6 +129,8 @@
 
       it 'creates a table' do
 
+        allow_any_instance_of(ActiveRecord::ConnectionAdapters::ClickhouseAdapter).to receive(:execute).and_call_original
+
         expect_any_instance_of(ActiveRecord::ConnectionAdapters::ClickhouseAdapter).to receive(:execute)
            .with('ALTER TABLE some ON CLUSTER ' + connection_config[:cluster_name] + ' DROP INDEX idx')
            .and_call_original