From 5852d10a44f99bd8a08b8c83f9f8de77c5b7648e Mon Sep 17 00:00:00 2001
From: Cody Cutrer <cody@instructure.com>
Date: Thu, 7 Sep 2023 15:05:35 -0600
Subject: [PATCH] fix pessimistic check constraints frd

---
 Gemfile.lock                                              | 2 +-
 lib/active_record/pg_extensions/pessimistic_migrations.rb | 6 ++++--
 lib/active_record/pg_extensions/version.rb                | 2 +-
 3 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/Gemfile.lock b/Gemfile.lock
index 293eaae..c419a75 100644
--- a/Gemfile.lock
+++ b/Gemfile.lock
@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    activerecord-pg-extensions (0.5.1)
+    activerecord-pg-extensions (0.5.2)
       activerecord (~> 7.0.0)
       railties (~> 7.0.0)
 
diff --git a/lib/active_record/pg_extensions/pessimistic_migrations.rb b/lib/active_record/pg_extensions/pessimistic_migrations.rb
index a0e6da9..fdb1e47 100644
--- a/lib/active_record/pg_extensions/pessimistic_migrations.rb
+++ b/lib/active_record/pg_extensions/pessimistic_migrations.rb
@@ -90,14 +90,16 @@ def add_index(table_name, column_name, **options)
       end
 
       def add_check_constraint(table_name, expression, if_not_exists: false, **options)
-        return if if_not_exists && check_constraint_for(table_name, expression, **options)
+        options = check_constraint_options(table_name, expression, options)
+        return if if_not_exists && check_constraint_for(table_name, **options)
 
         super
       end
 
       if ActiveRecord.version < Gem::Version.new("7.1")
         def remove_check_constraint(table_name, expression = nil, if_exists: false, **options)
-          return if if_exists && !check_constraint_for(table_name, expression, **options)
+          options = check_constraint_options(table_name, expression, options)
+          return if if_exists && !check_constraint_for(table_name, **options)
 
           super
         end
diff --git a/lib/active_record/pg_extensions/version.rb b/lib/active_record/pg_extensions/version.rb
index 9c3f642..62a34d8 100644
--- a/lib/active_record/pg_extensions/version.rb
+++ b/lib/active_record/pg_extensions/version.rb
@@ -2,6 +2,6 @@
 
 module ActiveRecord
   module PGExtensions
-    VERSION = "0.5.1"
+    VERSION = "0.5.2"
   end
 end