mvz-live_ast 1.1.0

data/lib/live_ast/to_ruby.rb

@@ -0,0 +1,24 @@
+require 'live_ast/base'
+
+[Method, UnboundMethod, Proc].each do |klass|
+  klass.class_eval do
+    def to_ruby  #:nodoc:
+      LiveAST.parser::Unparser.unparse(LiveAST.ast(self))
+    end
+  end
+end
+
+class Method
+  # :method: to_ruby
+  # Generate ruby code which reflects the AST of this object.
+end
+
+class UnboundMethod
+  # :method: to_ruby
+  # Generate ruby code which reflects the AST of this object.
+end
+
+class Proc
+  # :method: to_ruby
+  # Generate ruby code which reflects the AST of this object.
+end

data/lib/live_ast/version.rb

@@ -0,0 +1,3 @@
+module LiveAST
+  VERSION = "1.1.0"
+end

data/test/ast_eval_feature_test.rb

@@ -0,0 +1,11 @@
+require_relative 'main'
+
+class AAB_ASTEvalFeatureTest < BaseTest
+  def test_require
+    assert !private_methods.include?(:ast_eval)
+
+    require 'live_ast/ast_eval'
+
+    assert private_methods.include?(:ast_eval)
+  end
+end

data/test/ast_load_feature_test.rb

@@ -0,0 +1,11 @@
+require_relative 'main'
+
+class AAB_ASTLoadFeatureTest < BaseTest
+  def test_require
+    assert !private_methods.include?(:ast_load)
+
+    require 'live_ast/ast_load'
+
+    assert private_methods.include?(:ast_load)
+  end
+end

data/test/attr_test.rb

@@ -0,0 +1,24 @@
+require_relative 'main'
+
+class AttrTest < RegularTest
+  class A
+    attr_accessor :f
+    attr_reader :g
+    attr_writer :h
+  end
+
+  def test_attr
+    assert_raises LiveAST::ASTNotFoundError do
+      A.instance_method(:f).to_ast
+    end
+    assert_raises LiveAST::ASTNotFoundError do
+      A.instance_method(:f=).to_ast
+    end
+    assert_raises LiveAST::ASTNotFoundError do
+      A.instance_method(:g).to_ast
+    end
+    assert_raises LiveAST::ASTNotFoundError do
+      A.instance_method(:h=).to_ast
+    end
+  end
+end

data/test/backtrace_test.rb

@@ -0,0 +1,158 @@
+require_relative 'main'
+
+# test for raise redefinition side-effects: unsort this TestCase from
+# other TestCases.
+
+define_unsorted_test_case "BacktraceTest", RegularTest do
+  def test_raise_in_eval
+    3.times do
+      orig = exception_backtrace do
+        eval %{
+  
+          raise
+  
+  
+        }, binding, "somewhere", 1000
+      end
+  
+      live = exception_backtrace do
+        ast_eval %{
+  
+          raise
+  
+  
+        }, binding, "somewhere", 1000
+      end
+      
+      assert_equal orig.first, live.first
+      assert_match(/somewhere:1002/, live.first)
+    end
+  end
+
+  def test_raise_no_overrides
+    3.times do
+      orig = exception_backtrace do
+        eval %{
+  
+
+          raise
+  
+        }, binding, __FILE__, (__LINE__ + 9)
+      end
+  
+      live = exception_backtrace do
+        ast_eval %{
+  
+
+          raise
+  
+        }, binding
+      end
+
+      assert_equal orig.first, live.first
+      here = Regexp.quote __FILE__
+      assert_match(/#{here}/, live.first)
+    end
+  end
+
+  def test_raise_using_overrides
+    3.times do
+      orig = exception_backtrace do
+        eval %{
+  
+
+          raise
+  
+        }, binding, __FILE__, (__LINE__ + 9)
+      end
+  
+      live = exception_backtrace do
+        ast_eval %{
+  
+
+          raise
+  
+        }, binding, __FILE__, __LINE__
+      end
+
+      assert_equal orig.first, live.first
+      here = Regexp.quote __FILE__
+      assert_match(/#{here}/, live.first)
+    end
+  end
+
+  def test_raise_using_only_file_override
+    3.times do
+      orig = exception_backtrace do
+        eval %{
+  
+
+          raise
+  
+        }, binding, __FILE__
+      end
+  
+      live = exception_backtrace do
+        ast_eval %{
+  
+
+          raise
+  
+        }, binding, __FILE__
+      end
+
+      assert_equal orig.first, live.first
+      here = Regexp.quote __FILE__
+      assert_match(/#{here}/, live.first)
+    end
+  end
+  
+  def test_raise_after_eval
+    raise_after_eval("raise", false)
+    raise_after_eval("1/0", false)
+
+    require 'live_ast/replace_raise'
+
+    raise_after_eval("raise", true)
+    raise_after_eval("1/0", false)
+  end
+
+  def raise_after_eval(code, will_succeed)
+    3.times do
+      orig = eval %{
+
+        lambda { #{code} }
+
+
+      }, binding, "somewhere", 1000
+  
+      live = ast_eval %{
+
+          lambda { #{code} }
+
+
+      }, binding, "somewhere", 1000
+
+      orig_top = exception_backtrace { orig.call }.first
+      live_top = exception_backtrace { live.call }.first
+
+      assert_equal orig_top, LiveAST.strip_token(live_top)
+
+      if will_succeed
+        assert_equal orig_top, live_top
+        assert_match(/somewhere:1002/, live_top)
+      else
+        assert_not_equal orig_top, live_top
+        assert_match(/somewhere.*?:1002/, live_top)
+      end
+    end
+  end
+  
+  def test_tokens_stripped
+    exception_backtrace do
+      ast_eval %{ ast_eval %{ ast_eval %{raise}, binding }, binding }, binding
+    end.each do |line|
+      assert_nil line.index(LiveAST::Linker::REVISION_TOKEN)
+    end
+  end
+end

data/test/covert_define_method_test.rb

@@ -0,0 +1,23 @@
+require_relative 'main'
+
+class CovertDefineMethodTest < RegularTest
+  DEFINE = lambda do
+    class A
+      def A.my_def(*args, &block)
+        define_method(*args, &block)
+      end
+      
+      my_def :h do |x, y|
+        x + y
+      end
+    end
+  end
+
+  def test_covert_define_method
+    DEFINE.call
+    assert_equal 77, A.new.h(33, 44)
+
+    assert_equal binop_define_method(:h, :+, :my_def),
+                 A.instance_method(:h).to_ast
+  end
+end

data/test/def_test.rb

@@ -0,0 +1,35 @@
+require_relative 'main'
+
+class DefTest < RegularTest
+  class A
+    def f
+      "A#f"
+    end
+  end
+
+  def test_def_unbound_method_a
+    expected = no_arg_def(:f, "A#f")
+    assert_equal expected, A.instance_method(:f).to_ast
+  end
+
+  def test_def_method_a
+    expected = no_arg_def(:f, "A#f")
+    assert_equal expected, A.new.method(:f).to_ast
+  end
+
+  class B
+    def f(x, y)
+      x + y
+    end
+  end
+
+  def test_def_unbound_method_b
+    expected = binop_def(:f, :+)
+    assert_equal expected, B.instance_method(:f).to_ast
+  end
+
+  def test_def_instance_method_b
+    expected = binop_def(:f, :+)
+    assert_equal expected, B.new.method(:f).to_ast
+  end
+end

data/test/define_method_test.rb

@@ -0,0 +1,67 @@
+require_relative 'main'
+
+class DefineMethodTest < RegularTest
+  WITH_BLOCKS = lambda do
+    class A
+      {
+        :f => :+,
+        :g => :*,
+        :h => :-,
+      }.each_pair do |name, op|
+        case op
+        when :+
+          define_method name do |x, y|
+            x + y
+          end
+        when :*
+          define_method name do |x, y|
+            x * y
+          end
+        when :-
+          define_method name do |x, y|
+            x - y
+          end
+        end
+      end
+    end
+  end
+
+  def test_with_block
+    WITH_BLOCKS.call
+
+    assert_equal binop_define_method_with_var(:name, :+),
+                 A.instance_method(:f).to_ast
+
+    assert_equal binop_define_method_with_var(:name, :*),
+                 A.instance_method(:g).to_ast
+
+    assert_equal binop_define_method_with_var(:name, :-),
+                 A.instance_method(:h).to_ast
+  end
+
+  WITH_PROCS = lambda do
+    class B
+      op = lambda { |x, y| x / y }
+      
+      no_arg = proc { "B#f" }
+
+      define_method :g, &no_arg ; define_method :f, &op
+    end
+  end
+
+  def test_via_block
+    WITH_PROCS.call
+
+    assert_equal binop_block(:lambda, :/),
+                 B.instance_method(:f).to_ast
+
+    assert_equal binop_block(:lambda, :/),
+                 B.new.method(:f).to_ast
+
+    assert_equal no_arg_block(:proc, "B#f"),
+                 B.instance_method(:g).to_ast
+
+    assert_equal no_arg_block(:proc, "B#f"),
+                 B.new.method(:g).to_ast
+  end
+end

data/test/define_singleton_method_test.rb

@@ -0,0 +1,15 @@
+require_relative 'main'
+
+class DefineSingletonMethodTest < RegularTest
+  def test_define_singleton_method
+    a = Object.new
+    a.define_singleton_method :f do |x, y|
+      x + y
+    end
+
+    assert_equal 77, a.f(33, 44)
+
+    assert_equal binop_define_singleton_method(:f, :+, :a),
+                 a.singleton_class.instance_method(:f).to_ast
+  end
+end

data/test/encoding_test.rb

@@ -0,0 +1,52 @@
+require_relative 'main'
+
+class AllEncodingTest < RegularTest
+  ENC_TESTS = Hash[*%w[
+    default US-ASCII
+    usascii US-ASCII
+    utf8 UTF-8
+    utf8unix UTF-8
+    utf8mac UTF-8
+    utf8mac_alt UTF8-MAC
+    utf8dos UTF-8
+    utf8bom UTF-8
+    utf8bom_only UTF-8
+    usascii_with_utf8bom US-ASCII
+    koi8_with_utf8bom KOI8-R
+    cp932 Windows-31J
+    eucjp EUC-JP
+    koi8 KOI8-R
+    koi8_shebang KOI8-R
+  ]]
+
+  ENC_TESTS.each_pair do |abbr, name|
+    define_method "test_#{abbr}" do
+      require_relative "encoding_test/#{abbr}"
+      self.class.class_eval { include EncodingTest }
+
+      str = send("#{abbr}_string")
+      assert_equal name, str.encoding.to_s
+
+      ast = EncodingTest.instance_method("#{abbr}_string").to_ast
+      assert_equal "UTF-8", no_arg_def_return(ast).encoding.to_s
+      
+      LiveAST.load "./test/encoding_test/#{abbr}.rb"
+      
+      ast = EncodingTest.instance_method("#{abbr}_string").to_ast
+      assert_equal "UTF-8", no_arg_def_return(ast).encoding.to_s
+    end
+  end
+
+  def test_bad
+    orig = assert_raises ArgumentError do
+      live_ast_original_load "./test/encoding_test/bad.rb"
+    end
+    live = assert_raises ArgumentError do
+      LiveAST.load "./test/encoding_test/bad.rb"
+    end
+    # inconsistent punctuation from Ruby
+    re = %r!\Aunknown encoding name\s*[-:]\s*feynman-diagram\Z!
+    assert_match re, orig.message
+    assert_match re, live.message
+  end
+end

data/test/encoding_test/bad.rb

@@ -0,0 +1 @@
+# encoding: feynman-diagram

data/test/encoding_test/cp932.rb

@@ -0,0 +1,6 @@
+# -*- coding: cp932 -*-
+module EncodingTest
+  def cp932_string
+    "�傫�Ȕ��Ƃ˂��B"
+  end
+end