Rename cop. Add default settings. Add documentation.

Author: numbata

.rubocop.yml

@@ -16,3 +16,6 @@ Metrics/BlockLength:
 Naming/FileName:
   Exclude:
     - 'lib/rubocop-sequel.rb'
+
+RSpec/ExampleLength:
+  Enabled: false

config/default.yml

@@ -11,6 +11,13 @@ Sequel/ConcurrentIndex:
   VersionAdded: 0.0.1
   VersionChanged: 0.3.3
 
+Sequel/EnforceNotValidConstraint:
+  Description: 'Ensures that new constraints are created with the `not_valid: true` option.'
+  Reference: https://www.rubydoc.info/gems/rubocop-sequel/RuboCop/Cop/Sequel/EnforceNotValidConstraint
+  Enabled: false
+  VersionAdded: '0.3.4'
+  VersionChanged: '0.3.4'
+
 Sequel/JSONColumn:
   Description: >-
     Advocates the use of the `jsonb` column type over `json` or `hstore` for performance

lib/rubocop-sequel.rb

@@ -12,4 +12,4 @@
 require 'rubocop/cop/sequel/migration_name'
 require 'rubocop/cop/sequel/save_changes'
 require 'rubocop/cop/sequel/partial_constraint'
-require 'rubocop/cop/sequel/not_valid_constraint'
+require 'rubocop/cop/sequel/enforce_not_valid_constraint'

lib/rubocop/cop/sequel/enforce_not_valid_constraint.rb

@@ -0,0 +1,123 @@
+# frozen_string_literal: true
+
+module RuboCop
+  module Cop
+    module Sequel
+      # Ensures that new constraints are created with the `not_valid: true` option.
+      #
+      # This cop checks database migration files for `add_foreign_key` and
+      # `add_constraint` operations to ensure that new constraints are created
+      # with the `not_valid: true` option, which is necessary for deferred validation.
+      #
+      # This cop can be configured to exclude or include specific tables through
+      # the `ExcludeTables` and `IncludeTables` options.
+      #
+      # @example
+      #   # bad
+      #   add_foreign_key :users, :orders, column: :user_id
+      #
+      #   # good
+      #   add_foreign_key :users, :orders, column: :user_id, not_valid: true
+      #
+      #   # bad
+      #   alter_table :users do
+      #     add_constraint(:user_email_check) { "email IS NOT NULL" }
+      #   end
+      #
+      #   # good
+      #   alter_table :users do
+      #     add_constraint(:user_email_check, not_valid: true) { "email IS NOT NULL" }
+      #   end
+      #
+      # @example Configuration
+      #   Sequel/EnforceNotValidConstraint:
+      #     Enabled: true
+      #     Include:
+      #       - "db/migrations"
+      #     ExcludeTables:
+      #       - users
+      #     IncludeTables:
+      #       - comments
+      class EnforceNotValidConstraint < Base
+        MSG = "Add the 'not_valid: true' option when creating new constraints to ensure " \
+              'they are not validated immediately.'
+
+        # Matcher for alter_table blocks
+        def_node_matcher :alter_table_block?, <<-MATCHER
+          (block
+            (send _ :alter_table ({str sym} $_))
+            _
+            (begin
+              $...
+            )
+          )
+        MATCHER
+
+        # Matcher for constraint operations like add_foreign_key and add_constraint
+        def_node_matcher :constraint_operation?, <<-MATCHER
+          (send _ {:add_foreign_key :add_constraint} $...)
+        MATCHER
+
+        # Check constraint operations inside the alter_table block
+        #
+        # @param node [RuboCop::AST::Node] the AST node being checked
+        def on_block(node)
+          alter_table_block?(node) do |table_name, block_nodes|
+            next if skip_table?(table_name)
+
+            block_nodes.each { |block_node| validate_node(block_node) }
+          end
+        end
+
+        private
+
+        # Validate constraint operations
+        #
+        # @param node [RuboCop::AST::Node] the AST node being checked
+        def validate_node(node)
+          constraint_operation?(node) do |args|
+            next if not_valid_option_present?(args)
+
+            add_offense(node, message: MSG)
+          end
+        end
+
+        # Check if the not_valid: true option is present
+        #
+        # @param args [Array<RuboCop::AST::Node>] the operation arguments
+        # @return [Boolean] whether the not_valid: true option is present
+        def not_valid_option_present?(args)
+          args.any? do |arg|
+            arg.hash_type? && arg.pairs.any? do |pair|
+              key = pair.key
+              value = pair.value
+              key.sym_type? && key.children.first == :not_valid && value.true_type?
+            end
+          end
+        end
+
+        # Skip the table if it is excluded or not included
+        #
+        # @param table_name [String] the name of the table
+        # @return [Boolean] whether the table should be skipped
+        def skip_table?(table_name)
+          return true if excluded_table?(table_name.to_s)
+
+          !included_table?(table_name.to_s)
+        end
+
+        def excluded_table?(table_name)
+          return false unless cop_config.key?('ExcludeTables')
+
+          cop_config['ExcludeTables'].include?(table_name)
+        end
+
+        def included_table?(table_name)
+          return true unless cop_config.key?('IncludeTables')
+
+          cop_config['IncludeTables'].include?(table_name)
+        end
+      end
+    end
+  end
+end

lib/rubocop/cop/sequel/not_valid_constraint.rb

@@ -15,7 +15,7 @@
       expect(description.include?("\n")).to be(false)
     end
 
-    it 'stars from a verb' do # rubocop:disable RSpec/ExampleLength
+    it 'stars from a verb' do
       description = config.dig(cop_name, 'Description')
       start_with_subject = description.match(/\AThis cop (?<verb>.+?) .*/)
       suggestion = start_with_subject[:verb]&.capitalize if start_with_subject
@@ -97,7 +97,7 @@
     end
   end
 
-  it 'sorts cop names alphabetically' do # rubocop:disable RSpec/ExampleLength
+  it 'sorts cop names alphabetically' do
     previous_key = ''
     config_default = YAML.load_file('config/default.yml')
 

spec/project_spec.rb

@@ -0,0 +1,116 @@
+# frozen_string_literal: true
+
+RSpec.describe RuboCop::Cop::Sequel::EnforceNotValidConstraint do
+  subject(:cop) { described_class.new(config) }
+
+  let(:cop_config) { { 'Enabled' => true } }
+  let(:config) { RuboCop::Config.new('Sequel/EnforceNotValidConstraint' => cop_config) }
+
+  context 'when add_constraint used' do
+    it 'registers an offense when used without `not_valid: true` option' do
+      offenses = inspect_source(<<~RUBY)
+        alter_table :foo do
+          add_constraint :not_valid, Sequel.lit("jsonb_typeof(column) = 'object'")
+          add_constraint({name: "not_valid"}, Sequel.lit("jsonb_typeof(column) = 'object'"))
+        end
+      RUBY
+
+      expect(offenses.size).to eq(2)
+    end
+
+    it 'does not register an offense when using `not_valid: true` option' do
+      offenses = inspect_source(<<~RUBY)
+        alter_table :foo do
+          add_constraint({name: "name", not_valid: true }, Sequel.lit("jsonb_typeof(column) = 'object'"))
+        end
+      RUBY
+
+      expect(offenses.size).to eq(0)
+    end
+  end
+
+  context 'when add_foreign_key used' do
+    it 'registers an offense when used without `not_valid: true` option' do
+      offenses = inspect_source(<<~RUBY)
+        alter_table :foo do
+          add_foreign_key(:not_valid, :table)
+          add_foreign_key(:not_valid, :table, name: "not_valid")
+        end
+      RUBY
+
+      expect(offenses.size).to eq(2)
+    end
+
+    it 'does not register an offense when using `not_valid: true` option' do
+      offenses = inspect_source(<<~RUBY)
+        alter_table :foo do
+          add_foreign_key(:not_valid, :table, name: "not_valid", not_valid: true)
+        end
+      RUBY
+
+      expect(offenses.size).to eq(0)
+    end
+  end
+
+  context 'with ExcludeTables option' do
+    let(:cop_config) do
+      {
+        'Enabled' => true,
+        'ExcludeTables' => ['excluded_table']
+      }
+    end
+
+    it 'does not register an offense when using `not_valid: true` option inside excluded table migration' do
+      offenses = inspect_source(<<~RUBY)
+        alter_table :excluded_table do
+          add_foreign_key(:not_valid, :table)
+          add_foreign_key(:not_valid, :table, name: "not_valid")
+        end
+      RUBY
+
+      expect(offenses.size).to eq(0)
+    end
+
+    it 'register an offense when using `not_valid: true` option for not excluded table' do
+      offenses = inspect_source(<<~RUBY)
+        alter_table :not_excluded_table do
+          add_foreign_key(:not_valid, :table)
+          add_foreign_key(:not_valid, :table, name: "not_valid")
+        end
+      RUBY
+
+      expect(offenses.size).to eq(2)
+    end
+  end
+
+  context 'with IncludeTables option' do
+    let(:cop_config) do
+      {
+        'Enabled' => true,
+        'IncludeTables' => ['included_table']
+      }
+    end
+
+    it 'does not register an offense when using `not_valid: true` option inside excluded table migration' do
+      offenses = inspect_source(<<~RUBY)
+        alter_table :included_table do
+          add_foreign_key(:not_valid, :table)
+          add_foreign_key(:not_valid, :table, name: "not_valid")
+        end
+      RUBY
+
+      expect(offenses.size).to eq(2)
+    end
+
+    it 'register an offense when using `not_valid: true` option for not excluded table' do
+      offenses = inspect_source(<<~RUBY)
+        alter_table :not_included_table do
+          add_foreign_key(:not_valid, :table)
+          add_foreign_key(:not_valid, :table, name: "not_valid")
+        end
+      RUBY
+
+      expect(offenses.size).to eq(0)
+    end
+  end
+end