Squish check constraint expressions

Signed-off-by: Lovro Bikic <[email protected]>

Author: lovro-bikic

lib/annotate/annotate_models.rb

@@ -372,9 +372,9 @@ def get_check_constraint_info(klass, options = {})
       max_size = check_constraints.map { |check_constraint| check_constraint.name.size }.max + 1
       check_constraints.sort_by(&:name).each do |check_constraint|
         cc_info << if options[:format_markdown]
-                     sprintf("# * `%s`: `(%s)`\n", check_constraint.name, check_constraint.expression)
+                     sprintf("# * `%s`: `(%s)`\n", check_constraint.name, check_constraint.expression.squish)
                    else
-                     sprintf("#  %-#{max_size}.#{max_size}s (%s)\n", check_constraint.name, check_constraint.expression)
+                     sprintf("#  %-#{max_size}.#{max_size}s (%s)\n", check_constraint.name, check_constraint.expression.squish)
                    end
       end
 

spec/lib/annotate/annotate_models_spec.rb

@@ -787,7 +787,13 @@ def mock_column(name, type, options = {})
                   let :check_constraints do
                     [
                       mock_check_constraint('alive', 'age < 150'),
-                      mock_check_constraint('must_be_adult', 'age >= 18')
+                      mock_check_constraint('must_be_adult', 'age >= 18'),
+                      mock_check_constraint('multiline_test', <<~SQL)
+                        CASE
+                          WHEN (age >= 18) THEN (age <= 21)
+                          ELSE true
+                        END
+                      SQL
                     ]
                   end
 
@@ -802,8 +808,9 @@ def mock_column(name, type, options = {})
                       #
                       # Check Constraints
                       #
-                      #  alive          (age < 150)
-                      #  must_be_adult  (age >= 18)
+                      #  alive           (age < 150)
+                      #  multiline_test  (CASE WHEN (age >= 18) THEN (age <= 21) ELSE true END)
+                      #  must_be_adult   (age >= 18)
                       #
                     EOS
                   end
@@ -1581,7 +1588,13 @@ def mock_column(name, type, options = {})
                 context 'when check constraints are defined' do
                   let :check_constraints do
                     [
-                      mock_check_constraint('min_name_length', 'LENGTH(name) > 2')
+                      mock_check_constraint('min_name_length', 'LENGTH(name) > 2'),
+                      mock_check_constraint('multiline_test', <<~SQL)
+                        CASE
+                          WHEN (age >= 18) THEN (age <= 21)
+                          ELSE true
+                        END
+                      SQL
                     ]
                   end
 
@@ -1601,6 +1614,7 @@ def mock_column(name, type, options = {})
                       # ### Check Constraints
                       #
                       # * `min_name_length`: `(LENGTH(name) > 2)`
+                      # * `multiline_test`: `(CASE WHEN (age >= 18) THEN (age <= 21) ELSE true END)`
                       #
                     EOS
                   end