action_spec.rb in synvert-0.0.4

- old
+ new
@@ -1,75 +1,175 @@
 require 'spec_helper'
 
 module Synvert
   describe Rewriter::ReplaceWithAction do
-    describe '#rewrite' do
-      it 'replaces code' do
-        action = Rewriter::ReplaceWithAction.new('create_list {{self.arguments}}')
-        source = "post = FactoryGirl.create_list :post, 2"
-        send_node = Parser::CurrentRuby.parse(source).children[1]
-        output = action.rewrite(source, send_node)
-        expect(output).to eq "post = create_list :post, 2"
+    subject {
+      source = "post = FactoryGirl.create_list :post, 2"
+      send_node = Parser::CurrentRuby.parse(source).children[1]
+      instance = double(:current_node => send_node)
+      Rewriter::ReplaceWithAction.new(instance, 'create_list {{arguments}}')
+    }
+
+    it 'gets begin_pos' do
+      expect(subject.begin_pos).to eq 7
+    end
+
+    it 'gets end_pos' do
+      expect(subject.end_pos).to eq 39
+    end
+
+    it 'gets rewritten_code' do
+      expect(subject.rewritten_code).to eq 'create_list :post, 2'
+    end
+  end
+
+  describe Rewriter::AppendAction < Rewriter::Action do
+    describe 'class node' do
+      subject {
+        source = "class User\n  has_many :posts\nend"
+        class_node = Parser::CurrentRuby.parse(source)
+        instance = double(:current_node => class_node)
+        Rewriter::AppendAction.new(instance, "def as_json\n  super\nend")
+      }
+
+      it 'gets begin_pos' do
+        expect(subject.begin_pos).to eq 28
       end
+
+      it 'gets end_pos' do
+        expect(subject.end_pos).to eq 28
+      end
+
+      it 'gets rewritten_code' do
+        expect(subject.rewritten_code).to eq "\n\n  def as_json\n    super\n  end"
+      end
     end
   end
 
   describe Rewriter::InsertAction do
-    describe '#rewrite' do
-      it 'insert code to block node' do
-        action = Rewriter::InsertAction.new('{{self.arguments.first}}.include FactoryGirl::Syntax::Methods')
-        source = "RSpec.configure do |config|\nend"
+    describe 'block node without args' do
+      subject {
+        source = "Synvert::Application.configure do\nend"
         block_node = Parser::CurrentRuby.parse(source)
-        output = action.rewrite(source, block_node)
-        expect(output).to eq "RSpec.configure do |config|\n  config.include FactoryGirl::Syntax::Methods\nend"
+        instance = double(:current_node => block_node)
+        Rewriter::InsertAction.new(instance, 'config.eager_load = true')
+      }
+
+      it 'gets begin_pos' do
+        expect(subject.begin_pos).to eq 33
       end
 
-      it 'insert code to class node' do
-        action = Rewriter::InsertAction.new('include FactoryGirl::Syntax::Methods')
-        source = "class Test::Unit::TestCase\nend"
-        block_node = Parser::CurrentRuby.parse(source)
-        output = action.rewrite(source, block_node)
-        expect(output).to eq "class Test::Unit::TestCase\n  include FactoryGirl::Syntax::Methods\nend"
+      it 'gets end_pos' do
+        expect(subject.end_pos).to eq 33
       end
 
-      it 'insert code to begin node' do
-        action = Rewriter::InsertAction.new('World(FactoryGirl::Syntax::Methods)')
-        source = "require 'cucumber/rails'\nActionController::Base.allow_rescue = false"
-        block_node = Parser::CurrentRuby.parse(source)
-        output = action.rewrite(source, block_node)
-        expect(output).to eq "require 'cucumber/rails'\nActionController::Base.allow_rescue = false\nWorld(FactoryGirl::Syntax::Methods)"
+      it 'gets rewritten_code' do
+        expect(subject.rewritten_code).to eq "\n  config.eager_load = true"
       end
+    end
 
-      it 'insert code to other node' do
-        action = Rewriter::InsertAction.new('World(FactoryGirl::Syntax::Methods)')
-        source = "require 'cucumber/rails'"
+    describe 'block node with args' do
+      subject {
+        source = "RSpec.configure do |config|\nend"
         block_node = Parser::CurrentRuby.parse(source)
-        output = action.rewrite(source, block_node)
-        expect(output).to eq "require 'cucumber/rails'\nWorld(FactoryGirl::Syntax::Methods)"
+        instance = double(:current_node => block_node)
+        Rewriter::InsertAction.new(instance, '{{arguments.first}}.include FactoryGirl::Syntax::Methods')
+      }
+
+      it 'gets begin_pos' do
+        expect(subject.begin_pos).to eq 27
       end
+
+      it 'gets end_pos' do
+        expect(subject.end_pos).to eq 27
+      end
+
+      it 'gets rewritten_code' do
+        expect(subject.rewritten_code).to eq "\n  config.include FactoryGirl::Syntax::Methods"
+      end
     end
+
+    describe 'class node without superclass' do
+      subject {
+        source = "class User\n  has_many :posts\nend"
+        class_node = Parser::CurrentRuby.parse(source)
+        instance = double(:current_node => class_node)
+        Rewriter::InsertAction.new(instance, 'include Deletable')
+      }
+
+      it 'gets begin_pos' do
+        expect(subject.begin_pos).to eq 10
+      end
+
+      it 'gets end_pos' do
+        expect(subject.end_pos).to eq 10
+      end
+
+      it 'gets rewritten_code' do
+        expect(subject.rewritten_code).to eq "\n  include Deletable"
+      end
+    end
+
+    describe 'class node with superclass' do
+      subject {
+        source = "class User < ActiveRecord::Base\n  has_many :posts\nend"
+        class_node = Parser::CurrentRuby.parse(source)
+        instance = double(:current_node => class_node)
+        Rewriter::InsertAction.new(instance, 'include Deletable')
+      }
+
+      it 'gets begin_pos' do
+        expect(subject.begin_pos).to eq 31
+      end
+
+      it 'gets end_pos' do
+        expect(subject.end_pos).to eq 31
+      end
+
+      it 'gets rewritten_code' do
+        expect(subject.rewritten_code).to eq "\n  include Deletable"
+      end
+    end
   end
 
   describe Rewriter::InsertAfterAction do
-    describe '#rewrite' do
-      it 'insert_after code' do
-        action = Rewriter::InsertAfterAction.new('include Bar')
-        source = "  include Foo"
-        node = Parser::CurrentRuby.parse(source)
-        output = action.rewrite(source, node)
-        expect(output).to eq "  include Foo\n  include Bar"
-      end
+    subject {
+      source = "  include Foo"
+      node = Parser::CurrentRuby.parse(source)
+      instance = double(:current_node => node)
+      Rewriter::InsertAfterAction.new(instance, 'include Bar')
+    }
+
+    it 'gets begin_pos' do
+      expect(subject.begin_pos).to eq 13
     end
+
+    it 'gets end_pos' do
+      expect(subject.end_pos).to eq 13
+    end
+
+    it 'gets rewritten_code' do
+      expect(subject.rewritten_code).to eq "\n  include Bar"
+    end
   end
 
   describe Rewriter::RemoveAction do
-    describe '#rewrite' do
-      it 'remove code' do
-        action = Rewriter::RemoveAction.new
-        source = "user = User.new params[:user]\nuser.save\nrender\n"
-        send_node = Parser::CurrentRuby.parse(source).children[1]
-        output = action.rewrite(source, send_node)
-        expect(output).to eq "user = User.new params[:user]\nrender\n"
-      end
+    subject {
+      source = "user = User.new params[:user]\nuser.save\nrender\n"
+      send_node = Parser::CurrentRuby.parse(source).children[1]
+      instance = double(:current_node => send_node)
+      Rewriter::RemoveAction.new(instance)
+    }
+
+    it 'gets begin_pos' do
+      expect(subject.begin_pos).to eq 30
+    end
+
+    it 'gets end_pos' do
+      expect(subject.end_pos).to eq 39
+    end
+
+    it 'gets rewritten_code' do
+      expect(subject.rewritten_code).to eq ""
     end
   end
 end