diff options
Diffstat (limited to 'spec/support/database/prevent_cross_database_modification.rb')
| -rw-r--r-- | spec/support/database/prevent_cross_database_modification.rb | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/spec/support/database/prevent_cross_database_modification.rb b/spec/support/database/prevent_cross_database_modification.rb index 587d1fec586..b4c968e3c41 100644 --- a/spec/support/database/prevent_cross_database_modification.rb +++ b/spec/support/database/prevent_cross_database_modification.rb @@ -74,7 +74,9 @@ module Database return if cross_database_context[:transaction_depth_by_db].values.all?(&:zero?) - tables = PgQuery.parse(sql).dml_tables + parsed_query = PgQuery.parse(sql) + tables = sql.downcase.include?(' for update') ? parsed_query.tables : parsed_query.dml_tables + return if tables.empty? cross_database_context[:modified_tables_by_db][database].merge(tables) |