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

@@ -32,3 +32,61 @@
 require 'annotate/helpers'
 require 'annotate/constants'
 require 'byebug'
+
+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