localtower 0.4.2 → 1.0.0

data/spec/lib/localtower/generators/migration_spec.rb

@@ -3,111 +3,84 @@ require 'spec_helper'
 module Localtower
   module Generators
     describe Migration do
-      before(:all) do
-        clean_files
-      end
-
-      after(:all) do
-      end
-
-      it 'create_table' do
-        data = attributes_for(:create_table)
-        data["run_migrate"] = true
-
-        generator = ::Localtower::Generators::Migration.new(data).run
-
-        expect(::Localtower::Tools.word_in_file?(last_migration, /create_table :posts/)).to eq(true)
-      end
-
-      it 'create_table_two' do
-        data = attributes_for(:create_table_two)
-        data["run_migrate"] = true
-
-        generator = ::Localtower::Generators::Migration.new(data).run
-
-        expect(::Localtower::Tools.word_in_file?(last_migration, /create_table :users/)).to eq(true)
-      end
-
       it 'add_column' do
-        data = attributes_for(:add_column)
-        data["run_migrate"] = true
+        data = attributes_for(:add_column)[:migrations]
 
         generator = ::Localtower::Generators::Migration.new(data).run
 
-        expect(::Localtower::Tools.word_in_file?(last_migration, /add_column :posts, :tags, :text, default: \[\], array: true/)).to eq(true)
-        expect(::Localtower::Tools.word_in_file?(last_migration, /add_index :posts, :tags, using: :gin/)).to eq(true)
-        expect(::Localtower::Tools.word_in_file?(last_migration, /add_column :posts, :views, :integer, default: 0, null: false, index: true/)).to eq(true)
-        expect(::Localtower::Tools.word_in_file?(last_migration, /add_column :posts, :content, :text/)).to eq(true)
-        expect(::Localtower::Tools.word_in_file?(last_migration, /add_column :posts, :title, :string, index: true/)).to eq(true)
+        expect(word_in_file?(last_migration, /add_column :posts, :tags, :string, default: \[\], array: true/)).to eq(true)
+        expect(word_in_file?(last_migration, /add_column :posts, :views, :integer, default: 0, null: false/)).to eq(true)
+        expect(word_in_file?(last_migration, /add_column :posts, :content, :text/)).to eq(true)
+        expect(word_in_file?(last_migration, /add_column :posts, :title, :string/)).to eq(true)
       end
 
-      it 'add_column fail' do
-        data = attributes_for(:add_column_fail)
-        data["run_migrate"] = true
+      it 'remove_column' do
+        data = attributes_for(:remove_column)[:migrations]
 
         generator = ::Localtower::Generators::Migration.new(data).run
 
-        expect(::Localtower::Tools.word_in_file?(last_migration, /add_column :posts, :tags, :text, default: \[\], array: true/)).to eq(false)
-        expect(::Localtower::Tools.word_in_file?(last_migration, /add_index :posts, :tags, using: :gin/)).to eq(false)
-        expect(::Localtower::Tools.word_in_file?(last_migration, /add_column :posts, :views, :integer, default: 0, null: false, index: true/)).to eq(false)
-        expect(::Localtower::Tools.word_in_file?(last_migration, /add_column :posts, :content, :text/)).to eq(false)
-        expect(::Localtower::Tools.word_in_file?(last_migration, /add_column :posts, :title, :string, index: true/)).to eq(false)
+        expect(word_in_file?(last_migration, /remove_column :posts, :title/)).to eq(true)
+        expect(word_in_file?(last_migration, /rename_column :posts, :views, :views_by_users/)).to eq(true)
       end
 
       it 'rename_column' do
-        data = attributes_for(:rename_column)
-        data["run_migrate"] = true
+        data = attributes_for(:rename_column)[:migrations]
 
         generator = ::Localtower::Generators::Migration.new(data).run
 
-        expect(::Localtower::Tools.word_in_file?(last_migration, /rename_column :posts, :content, :content_new/)).to eq(true)
+        expect(word_in_file?(last_migration, /rename_column :posts, :content, :content_new/)).to eq(true)
       end
 
-      it 'remove_column' do
-        data = attributes_for(:remove_column)
-        data["run_migrate"] = true
+      it 'change_column_type' do
+        data = attributes_for(:change_column_type)[:migrations]
 
         generator = ::Localtower::Generators::Migration.new(data).run
 
-        expect(::Localtower::Tools.word_in_file?(last_migration, /remove_column :posts, :title/)).to eq(true)
-        expect(::Localtower::Tools.word_in_file?(last_migration, /rename_column :posts, :views, :views_by_users/)).to eq(true)
+        expect(word_in_file?(last_migration, /add_index :posts, :views_by_users/)).to eq(true)
+        expect(word_in_file?(last_migration, /change_column :posts, :content_new, :string/)).to eq(true)
       end
 
-      it 'change_column_type' do
-        data = attributes_for(:change_column_type)
-        data["run_migrate"] = true
+      it 'add_index_to_column' do
+        data = attributes_for(:add_index_to_column)[:migrations]
 
         generator = ::Localtower::Generators::Migration.new(data).run
 
-        expect(::Localtower::Tools.word_in_file?(last_migration, /add_index :posts, :views_by_users/)).to eq(true)
-        expect(::Localtower::Tools.word_in_file?(last_migration, /change_column :posts, :content_new, :string/)).to eq(true)
+        expect(word_in_file?(last_migration, /disable_ddl_transaction!/)).to eq(true)
+        expect(word_in_file?(last_migration, /add_index :posts, :content/)).to eq(true)
+        expect(word_in_file?(last_migration, /add_index :posts, :content_second, unique: true/)).to eq(true)
+        expect(word_in_file?(last_migration, /add_index :posts, :content_third, using: :gin, algorithm: :concurrently/)).to eq(true)
       end
 
-      it 'add_index_to_column' do
-        data = attributes_for(:add_index_to_column)
-        data["run_migrate"] = true
+      it 'belongs_to' do
+        data = attributes_for(:belongs_to)[:migrations]
+
+        model_post = "#{Rails.root}/app/models/post.rb"
+        model_user = "#{Rails.root}/app/models/user.rb"
+
+        File.open(model_post, 'w') { |f| f.write("class Post < ApplicationRecord\nend")}
+        File.open(model_user, 'w') { |f| f.write("class User < ApplicationRecord\nend")}
 
         generator = ::Localtower::Generators::Migration.new(data).run
 
-        expect(::Localtower::Tools.word_in_file?(last_migration, /add_index :posts, :content_new/)).to eq(true)
+        expect(word_in_file?(model_post, /# belongs_to :user/)).to eq(true)
+        expect(word_in_file?(model_user, /# has_many :posts/)).to eq(true)
+        expect(word_in_file?(last_migration, /add_reference :posts, :user, foreign_key: true, index: true/)).to eq(true)
       end
 
       it 'remove_index_to_column' do
-        data = attributes_for(:remove_index_to_column)
-        data["run_migrate"] = true
+        data = attributes_for(:remove_index_to_column)[:migrations]
 
         generator = ::Localtower::Generators::Migration.new(data).run
 
-        expect(::Localtower::Tools.word_in_file?(last_migration, /remove_index :posts, :views_by_users/)).to eq(true)
+        expect(word_in_file?(last_migration, /remove_index :posts, :views_by_users/)).to eq(true)
       end
 
       it 'drop_table' do
-        data = attributes_for(:drop_table)
-        data["run_migrate"] = true
+        data = attributes_for(:drop_table)[:migrations]
 
         generator = ::Localtower::Generators::Migration.new(data).run
 
-        expect(::Localtower::Tools.word_in_file?(last_migration, /drop_table :posts, force: :cascade/)).to eq(true)
+        expect(word_in_file?(last_migration, /drop_table :posts/)).to eq(true)
       end
     end
   end

data/spec/lib/localtower/generators/model_spec.rb

@@ -3,40 +3,49 @@ require 'spec_helper'
 module Localtower
   module Generators
     describe Model do
-      before(:all) do
-        clean_files
-      end
-
-      after(:all) do
-      end
-
-      it 'create a post' do
-        data = attributes_for(:post_one)
-        data["run_migrate"] = true
-
-        ::Localtower::Generators::Model.new(data).run
-
-        expect(File.exist?("#{Rails.root}/app/models/post.rb")).to eq(true)
-        expect(::Localtower::Tools.word_in_file?("#{Rails.root}/app/models/post.rb", /class Post/)).to eq(true)
-
-        expect(::Localtower::Tools.word_in_file?("#{Rails.root}/db/schema.rb", /create_table "posts"/)).to eq(true)
-        expect(::Localtower::Tools.word_in_file?("#{Rails.root}/db/schema.rb", /t.string(.*)"title"/)).to eq(true)
-        expect(::Localtower::Tools.word_in_file?("#{Rails.root}/db/schema.rb", /t.text(.*)"content"/)).to eq(true)
-        expect(::Localtower::Tools.word_in_file?("#{Rails.root}/db/schema.rb", /t.index(.*)\["title"\], name: "index_posts_on_title"/)).to eq(true)
-      end
-
-      it 'create a user' do
-        data = attributes_for(:user_one)
-        data["run_migrate"] = true
-
-        ::Localtower::Generators::Model.new(data).run
-
-        expect(File.exist?("#{Rails.root}/app/models/user.rb")).to eq(true)
-        expect(::Localtower::Tools.word_in_file?("#{Rails.root}/app/models/user.rb", /class User/)).to eq(true)
-        expect(::Localtower::Tools.word_in_file?("#{Rails.root}/db/schema.rb", /create_table "users"/)).to eq(true)
-        expect(::Localtower::Tools.word_in_file?("#{Rails.root}/db/schema.rb", /t.string(.*)"name"/)).to eq(true)
-        expect(::Localtower::Tools.word_in_file?("#{Rails.root}/db/schema.rb", /t.jsonb(.*)"metadata"/)).to eq(true)
-        expect(::Localtower::Tools.word_in_file?("#{Rails.root}/db/schema.rb", /t.index(.*)\["name"\], name: "index_users_on_name"/)).to eq(true)
+      context 'model 1' do
+        let(:expected_migration) do
+          <<-TEXT_MULTILINE.strip_heredoc
+          class CreatePosts < ActiveRecord::Migration[5.2]
+            disable_ddl_transaction!
+
+            def change
+              create_table :posts do |t|
+                t.string :title
+                t.string :tags, array: true, default: [], null: false
+                t.text :content, null: false
+                t.integer :likes_count, default: 0
+                t.float :score
+                t.jsonb :metadata, default: {}, null: false
+
+                t.timestamps
+              end
+              add_index :posts, :title, unique: true
+              add_index :posts, :tags, using: :gin, unique: true, algorithm: :concurrently
+              add_index :posts, :content, using: :gist
+              add_index :posts, :likes_count
+              add_index :posts, :score
+              add_index :posts, :metadata
+            end
+          end
+          TEXT_MULTILINE
+        end
+
+        let(:expected_model) do
+          <<-TEXT_MULTILINE.strip_heredoc
+          class Post < ApplicationRecord
+          end
+          TEXT_MULTILINE
+        end
+
+        let(:attributes) { attributes_for(:post_one).deep_stringify_keys }
+
+        it 'create a post' do
+          ::Localtower::Generators::Model.new(attributes).run
+
+          expect(File.read("#{Rails.root}/app/models/post.rb")).to eq(expected_model)
+          expect(File.read(last_migration)).to eq(expected_migration)
+        end
       end
     end
   end

data/spec/lib/localtower/generators/service_objects/insert_array_spec.rb

@@ -0,0 +1,47 @@
+require 'spec_helper'
+
+describe ::Localtower::Generators::ServiceObjects::InsertArray do
+  let(:service) { described_class.new(attributes) }
+
+  let(:attributes) {
+    ['tags']
+  }
+
+  let(:base_file_content) {
+    <<-MIGRATION.strip_heredoc
+      class CreateTests < ActiveRecord::Migration[7.0]
+        def change
+          create_table :tests do |t|
+            t.string :tags
+
+            t.timestamps
+          end
+        end
+      end
+    MIGRATION
+  }
+
+  let(:expected_file_content) {
+    <<-MIGRATION.strip_heredoc
+      class CreateTests < ActiveRecord::Migration[7.0]
+        def change
+          create_table :tests do |t|
+            t.string :tags, array: true
+
+            t.timestamps
+          end
+        end
+      end
+    MIGRATION
+  }
+
+  before do
+    File.open("#{Rails.root}/db/migrate/0_migration_name_.rb", 'w') { |f| f.write(base_file_content) }
+  end
+
+  it 'works' do
+    service.call
+
+    expect(File.read(last_migration)).to eq(expected_file_content)
+  end
+end

data/spec/lib/localtower/generators/service_objects/insert_defaults_spec.rb

@@ -1,32 +1,28 @@
 require 'spec_helper'
 
 describe ::Localtower::Generators::ServiceObjects::InsertDefaults do
-
-  let(:latest_migration) do
-    Tempfile.new('foo').tap do |file|
-      file.puts base_file_content
-      file.rewind
-    end
-  end
+  let(:service) { described_class.new(attributes) }
 
   let(:attributes) {
     [
-      { 'first'  => "true"  },
-      { 'third'  => "nil"   },
+      { 'first'  => "true" },
+      { 'third'  => "nil" },
       { 'fourth' => "false" },
-      { 'fifth'  => "0"     }
+      { 'foo' => "{}" },
+      { 'fifth'  => "0" }
     ]
   }
 
   let(:base_file_content) {
     <<-MIGRATION.strip_heredoc
-      class CreateTests < ActiveRecord::Migration[5.0]
+      class CreateTests < ActiveRecord::Migration[7.0]
         def change
           create_table :tests do |t|
             t.string  :first
             t.integer :second
             t.string  :third
             t.string  :fourth
+            t.jsonb  :foo
             t.integer :fifth
 
             t.timestamps
@@ -37,34 +33,33 @@ describe ::Localtower::Generators::ServiceObjects::InsertDefaults do
     MIGRATION
   }
 
-  let(:expected_file_structure_array) {
-    [
-      "class CreateTests < ActiveRecord::Migration[5.0]\n",
-      "  def change\n",
-      "    create_table :tests do |t|\n",
-      "      t.string  :first, default: true\n",
-      "      t.integer :second\n",
-      "      t.string  :third, default: nil\n",
-      "      t.string  :fourth, default: false\n",
-      "      t.integer :fifth, default: 0\n",
-      "\n",
-      "      t.timestamps\n",
-      "    end\n",
-      "    add_index :tests, :first\n",
-      "  end\n",
-      "end\n"
-    ]
-  }
+  let(:expected_file_content) {
+    <<-MIGRATION.strip_heredoc
+      class CreateTests < ActiveRecord::Migration[7.0]
+        def change
+          create_table :tests do |t|
+            t.string  :first, default: true
+            t.integer :second
+            t.string  :third, default: nil
+            t.string  :fourth, default: false
+            t.jsonb  :foo, default: {}
+            t.integer :fifth, default: 0
 
-  let(:service) { described_class.new(attributes) }
+            t.timestamps
+          end
+          add_index :tests, :first
+        end
+      end
+    MIGRATION
+  }
 
   before do
-    allow(service).to receive(:latest_migration).and_return(latest_migration)
-    service.call
+    File.open("#{Rails.root}/db/migrate/0_migration_name_.rb", 'w') { |f| f.write(base_file_content) }
   end
 
-  it 'inserts default values for attributes' do
-    lines_array = File.readlines(latest_migration)
-    expect(lines_array).to match_array expected_file_structure_array
+  it 'works' do
+    service.call
+
+    expect(File.read(last_migration)).to eq(expected_file_content)
   end
 end

data/spec/lib/localtower/generators/service_objects/insert_indexes_spec.rb

@@ -0,0 +1,90 @@
+require 'spec_helper'
+
+describe ::Localtower::Generators::ServiceObjects::InsertIndexes do
+  let(:service) { described_class.new(attributes) }
+
+  let(:attributes) {
+    [
+      { 'first'  =>
+        {
+          'using' => 'gin',
+          'algorithm' => 'concurrently',
+        }
+      },
+      { 'third'  =>
+        {
+          'using' => 'gist',
+          'unique' => true,
+        }
+      },
+      { 'fourth' =>
+        {
+          'using' => 'default',
+          'algorithm' => 'concurrently',
+        }
+      },
+      { 'fifth'  =>
+        {
+          'using' => 'default',
+          'unique' => true,
+        }
+      }
+    ]
+  }
+
+  let(:base_file_content) {
+    <<-MIGRATION.strip_heredoc
+      class CreateTests < ActiveRecord::Migration[7.0]
+        def change
+          create_table :tests do |t|
+            t.string  :zero
+            t.string  :first
+            t.string  :third
+            t.string  :fourth
+            t.integer :fifth
+
+            t.timestamps
+          end
+          add_index :tests, :first
+          add_index :tests, :third
+          add_index :tests, :fourth
+          add_index :tests, :fifth
+        end
+      end
+    MIGRATION
+  }
+
+  let(:expected_file_content) {
+    <<-MIGRATION.strip_heredoc
+      class CreateTests < ActiveRecord::Migration[7.0]
+        disable_ddl_transaction!
+
+        def change
+          create_table :tests do |t|
+            t.string  :zero
+            t.string  :first
+            t.string  :third
+            t.string  :fourth
+            t.integer :fifth
+
+            t.timestamps
+          end
+          add_index :tests, :first, using: :gin, algorithm: :concurrently
+          add_index :tests, :third, using: :gist, unique: true
+          add_index :tests, :fourth, algorithm: :concurrently
+          add_index :tests, :fifth, unique: true
+        end
+      end
+    MIGRATION
+  }
+
+  before do
+    File.open("#{Rails.root}/db/migrate/0_migration_name_.rb", 'w') { |f| f.write(base_file_content) }
+  end
+
+  it 'works' do
+    service.call
+
+    expect(File.read(last_migration)).to eq(expected_file_content)
+  end
+end

data/spec/lib/localtower/generators/service_objects/insert_nullable_spec.rb

@@ -0,0 +1,61 @@
+require 'spec_helper'
+
+describe ::Localtower::Generators::ServiceObjects::InsertNullable do
+  let(:service) { described_class.new(attributes) }
+
+  let(:attributes) {
+    [
+      'first',
+      'third',
+      'fourth'
+    ]
+  }
+
+  let(:base_file_content) {
+    <<-MIGRATION.strip_heredoc
+      class CreateTests < ActiveRecord::Migration[7.0]
+        def change
+          create_table :tests do |t|
+            t.string  :first
+            t.integer :second
+            t.string  :third
+            t.string  :fourth, default: 'foo'
+            t.integer :fifth
+
+            t.timestamps
+          end
+          add_index :tests, :first
+        end
+      end
+    MIGRATION
+  }
+
+  let(:expected_file_content) {
+    <<-MIGRATION.strip_heredoc
+      class CreateTests < ActiveRecord::Migration[7.0]
+        def change
+          create_table :tests do |t|
+            t.string  :first, null: false
+            t.integer :second
+            t.string  :third, null: false
+            t.string  :fourth, null: false, default: 'foo'
+            t.integer :fifth
+
+            t.timestamps
+          end
+          add_index :tests, :first
+        end
+      end
+    MIGRATION
+  }
+
+  before do
+    File.open("#{Rails.root}/db/migrate/0_migration_name_.rb", 'w') { |f| f.write(base_file_content) }
+  end
+
+  it 'works' do
+    service.call
+
+    expect(File.read(last_migration)).to eq(expected_file_content)
+  end
+end

data/spec/lib/localtower/tools_spec.rb

@@ -1,14 +1,4 @@
 require 'spec_helper'
 
-describe Localtower::Tools do
-  before(:each) do
-  end
-
-  after(:all) do
-  end
-
-  # TODO:
-  it 'list models' do
-    models = ::Localtower::Tools.models_presented
-  end
+RSpec.describe Localtower::Tools do
 end

data/spec/spec_helper.rb

@@ -1,7 +1,24 @@
 def clean_files
-  ::Localtower::Tools.sql_drop_all_tables
-  ::Localtower::Generators::Migration.new({}).remove_all_migrations
-  Dir["#{Rails.root}/app/models/**/*.*"].each { |model_file| File.delete(model_file) }
+  ::ActiveRecord::Base.connection.tables.each do |table|
+    cmd = "DROP TABLE if exists #{table.upcase} cascade;"
+    ::ActiveRecord::Base.connection.execute(cmd)
+  end
+
+  # remove Schema:
+  Dir["#{Rails.root}/db/migrate/*"].each { |migration_file| File.delete(migration_file) }
+
+  content_schema = """
+ActiveRecord::Schema.define(version: 0) do
+end
+  """
+
+  File.open("#{Rails.root}/db/schema.rb", "w") do |f|
+    f.write(content_schema)
+  end
+
+  Dir["#{Rails.root}/app/models/**/*.*"]
+  .reject { |file_name| file_name['application_record.rb']}
+  .each { |model_file| File.delete(model_file) }
 end
 
 def migration_files
@@ -12,6 +29,10 @@ def last_migration
   migration_files.sort.last
 end
 
+def word_in_file?(file, word_or_exp)
+  File.readlines(file).grep(word_or_exp).size > 0
+end
+
 #============================
 ENV['RAILS_ENV'] = 'test'
 
@@ -30,13 +51,12 @@ RSpec.configure do |config|
   config.infer_base_class_for_anonymous_controllers = true
 
   config.order = 123
-  # config.order = 'random'
 
   config.before(:suite) do
     FactoryBot.find_definitions
   end
 
-  config.before(:all) do
+  config.before(:each) do
     clean_files
   end