Move helpers in annotate_models_spec.rb

Author: nard-tech

spec/lib/annotate/annotate_models_spec.rb

@@ -20,64 +20,6 @@
     '# -*- frozen_string_literal : true -*-'
   ].freeze
 
-  def mock_index(name, params = {})
-    double('IndexKeyDefinition',
-           name:          name,
-           columns:       params[:columns] || [],
-           unique:        params[:unique] || false,
-           orders:        params[:orders] || {},
-           where:         params[:where],
-           using:         params[:using])
-  end
-
-  def mock_foreign_key(name, from_column, to_table, to_column = 'id', constraints = {})
-    double('ForeignKeyDefinition',
-           name:         name,
-           column:       from_column,
-           to_table:     to_table,
-           primary_key:  to_column,
-           on_delete:    constraints[:on_delete],
-           on_update:    constraints[:on_update])
-  end
-
-  def mock_connection(indexes = [], foreign_keys = [])
-    double('Conn',
-           indexes:      indexes,
-           foreign_keys: foreign_keys,
-           supports_foreign_keys?: true)
-  end
-
-  def mock_class(table_name, primary_key, columns, indexes = [], foreign_keys = [])
-    options = {
-      connection:       mock_connection(indexes, foreign_keys),
-      table_exists?:    true,
-      table_name:       table_name,
-      primary_key:      primary_key,
-      column_names:     columns.map { |col| col.name.to_s },
-      columns:          columns,
-      column_defaults:  Hash[columns.map { |col| [col.name, col.default] }],
-      table_name_prefix: ''
-    }
-
-    double('An ActiveRecord class', options)
-  end
-
-  def mock_column(name, type, options = {})
-    default_options = {
-      limit: nil,
-      null: false,
-      default: nil,
-      sql_type: type
-    }
-
-    stubs = default_options.dup
-    stubs.merge!(options)
-    stubs[:name] = name
-    stubs[:type] = type
-
-    double('Column', stubs)
-  end
-
   describe '.quote' do
     subject do
       AnnotateModels.quote(value)

spec/spec_helper.rb

@@ -93,3 +93,61 @@ def self.clean?(test_rig)
     end
   end
 end
+
+def mock_index(name, params = {})
+  double('IndexKeyDefinition',
+         name:          name,
+         columns:       params[:columns] || [],
+         unique:        params[:unique] || false,
+         orders:        params[:orders] || {},
+         where:         params[:where],
+         using:         params[:using])
+end
+
+def mock_foreign_key(name, from_column, to_table, to_column = 'id', constraints = {})
+  double('ForeignKeyDefinition',
+         name:         name,
+         column:       from_column,
+         to_table:     to_table,
+         primary_key:  to_column,
+         on_delete:    constraints[:on_delete],
+         on_update:    constraints[:on_update])
+end
+
+def mock_connection(indexes = [], foreign_keys = [])
+  double('Conn',
+         indexes:      indexes,
+         foreign_keys: foreign_keys,
+         supports_foreign_keys?: true)
+end
+
+def mock_class(table_name, primary_key, columns, indexes = [], foreign_keys = [])
+  options = {
+    connection:       mock_connection(indexes, foreign_keys),
+    table_exists?:    true,
+    table_name:       table_name,
+    primary_key:      primary_key,
+    column_names:     columns.map { |col| col.name.to_s },
+    columns:          columns,
+    column_defaults:  Hash[columns.map { |col| [col.name, col.default] }],
+    table_name_prefix: ''
+  }
+
+  double('An ActiveRecord class', options)
+end
+
+def mock_column(name, type, options = {})
+  default_options = {
+    limit: nil,
+    null: false,
+    default: nil,
+    sql_type: type
+  }
+
+  stubs = default_options.dup
+  stubs.merge!(options)
+  stubs[:name] = name
+  stubs[:type] = type
+
+  double('Column', stubs)
+end