add expectation for lock

Co-authored-by: Ryan Krage <[email protected]>
braintree · Nov 21, 2023 · e202818 · e202818
1 parent e55f9f1
commit e202818
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/spec/safe_statements_spec.rb b/spec/safe_statements_spec.rb
@@ -1275,7 +1275,8 @@ def up
 
             expect do
               test_migration.suppress_messages { test_migration.migrate(:up) }
-            end.to make_database_queries(matching: /CREATE INDEX "index_foos_on_bar"/, count: 1)
+            end.to make_database_queries(matching: /LOCK "public"\."foos" IN SHARE MODE/, count: 1)
+              .and(make_database_queries(matching: /CREATE INDEX "index_foos_on_bar"/, count: 1))
 
             indexes = ActiveRecord::Base.connection.indexes("foos")
             expect(indexes.size).to eq(1)