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

PNixx · Sep 6, 2024 · 2b4c2dc · 2b4c2dc
1 parent 02e46a0
commit 2b4c2dc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/spec/cluster/migration_spec.rb b/spec/cluster/migration_spec.rb
@@ -126,6 +126,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