Refactor RSpec for AnnotateModels - with Globalize gem (ctran#749)

Refactor test cases for AnnotateModels.get_schema_info with Globalize gem

Author: nard-tech

spec/lib/annotate/annotate_models_spec.rb

@@ -761,6 +761,49 @@ def mock_column(name, type, options = {})
  end
  end
  end
+
+ context 'with Globalize gem' do
+ let :translation_klass do
+ double('Post::Translation',
+ to_s: 'Post::Translation',
+ columns: [
+ mock_column(:id, :integer, limit: 8),
+ mock_column(:post_id, :integer, limit: 8),
+ mock_column(:locale, :string, limit: 50),
+ mock_column(:title, :string, limit: 50),
+ ])
+ end
+
+ let :klass do
+ mock_class(:posts, primary_key, columns, indexes, foreign_keys).tap do |mock_klass|
+ allow(mock_klass).to receive(:translation_class).and_return(translation_klass)
+ end
+ end
+
+ let :columns do
+ [
+ mock_column(:id, :integer, limit: 8),
+ mock_column(:author_name, :string, limit: 50),
+ ]
+ end
+
+ let :expected_result do
+ <<~EOS
+ # Schema Info
+ #
+ # Table name: posts
+ #
+ # id :integer not null, primary key
+ # author_name :string(50) not null
+ # title :string(50) not null
+ #
+ EOS
+ end
+
+ it 'returns schema info' do
+ is_expected.to eq expected_result
+ end
+ end
  end
 
  context 'when the primary key is an array (using composite_primary_keys)' do
@@ -1140,39 +1183,6 @@ def mock_column(name, type, options = {})
  end
  end
 
- it 'should work with the Globalize gem' do
- klass = mock_class(:posts,
- :id,
- [
- mock_column(:id, :integer, limit: 8),
- mock_column(:author_name, :string, limit: 50),
- ])
-
- options = {
- to_s: 'Post::Translation',
- columns: [
- mock_column(:id, :integer, limit: 8),
- mock_column(:post_id, :integer, limit: 8),
- mock_column(:locale, :string, limit: 50),
- mock_column(:title, :string, limit: 50),
- ]
- }
- translation_klass = double('Post::Translation', options)
- allow(klass).to receive(:translation_class).and_return(translation_klass)
-
- expected_schema_info = <<~EOS
- # Schema Info
- #
- # Table name: posts
- #
- # id :integer not null, primary key
- # author_name :string(50) not null
- # title :string(50) not null
- #
- EOS
- expect(AnnotateModels.get_schema_info(klass, 'Schema Info')).to eql(expected_schema_info)
- end
-
  describe '.set_defaults' do
  subject do
  Annotate::Helpers.true?(ENV['show_complete_foreign_keys'])