Move to_raw_source to CompositeParser

Author: sue445

lib/rspec/parameterized/core.rb

@@ -1,9 +1,8 @@
 require "rspec/parameterized/core/version"
-require 'parser'
-require 'unparser'
-require 'proc_to_ast'
 require 'rspec/parameterized/core/helper_methods'
 require 'rspec/parameterized/core/example_helper_methods'
+require 'rspec/parameterized/core/errors'
+require 'rspec/parameterized/core/composite_parser'
 
 module RSpec
   module Parameterized
@@ -143,11 +142,9 @@ def define_cases(parameter, *args, &block)
         end
 
         def params_inspect(obj)
-          begin
-            obj.is_a?(Proc) ? obj.to_raw_source : obj.inspect
-          rescue Parser::SyntaxError
-            return obj.inspect
-          end
+          RSpec::Parameterized::Core::CompositeParser.to_raw_source(obj)
+        rescue ParserSyntaxError
+          return obj.inspect
         end
 
         def set_verbose_parameters(&block)

lib/rspec/parameterized/core/composite_parser.rb

@@ -0,0 +1,51 @@
+module RSpec
+  module Parameterized
+    module Core
+      # Proxy class for parser and prism
+      module CompositeParser
+        # @param obj [Object]
+        # @return [String]
+        # @raise [RSpec::Parameterized::Core::ParserSyntaxError]
+        def self.to_raw_source(obj)
+          return to_raw_source_with_prism(obj) if use_prism?
+
+          to_raw_source_with_parser(obj)
+        end
+
+        # Whether use parser or prism
+        #
+        # @return [true] Use prism
+        # @return [false] Use parser
+        def self.use_prism?
+          Gem::Version.new(RUBY_VERSION) >= Gem::Version.new("3.4.0")
+        end
+
+        # @param obj [Object]
+        # @return [String]
+        # @raise [RSpec::Parameterized::Core::ParserSyntaxError]
+        def self.to_raw_source_with_parser(obj)
+          obj.is_a?(Proc) ? obj.to_raw_source : obj.inspect
+        rescue Parser::SyntaxError => e
+          raise ParserSyntaxError
+        end
+        private_class_method :to_raw_source_with_parser
+
+        # @param obj [Object]
+        # @return [String]
+        def self.to_raw_source_with_prism(obj)
+          # TODO: Impl
+          raise "TODO: Impl"
+        end
+        private_class_method :to_raw_source_with_prism
+      end
+    end
+  end
+end
+
+if RSpec::Parameterized::Core::CompositeParser.use_prism?
+  require 'prism'
+else
+  require 'parser'
+  require 'unparser'
+  require 'proc_to_ast'
+end

lib/rspec/parameterized/core/errors.rb

@@ -0,0 +1,9 @@
+module RSpec
+  module Parameterized
+    module Core
+      class Error < StandardError; end
+
+      class ParserSyntaxError < Error; end
+    end
+  end
+end

lib/rspec/parameterized/core/lazy_arg.rb

@@ -11,8 +11,8 @@ def apply(obj)
         end
 
         def inspect
-          "#{@block.to_raw_source}"
-        rescue Parser::SyntaxError
+          CompositeParser.to_raw_source(@block)
+        rescue ParserSyntaxError
           super.inspect
         end
       end

spec/rspec/parameterized/core/composite_parser_spec.rb

@@ -0,0 +1,33 @@
+describe RSpec::Parameterized::Core::CompositeParser do
+  describe ".to_raw_source" do
+    subject { RSpec::Parameterized::Core::CompositeParser.to_raw_source(arg) }
+
+    context "arg is not proc" do
+      let(:arg) do
+        123
+      end
+
+      it { should eq "123" }
+    end
+
+    context "arg is proc" do
+      context "simple case" do
+        let(:arg) do
+          ->(a) { a + 1 }
+        end
+
+        it { should eq "->(a) { a + 1 }" }
+        its(:encoding) { should eq Encoding::UTF_8 }
+      end
+
+      context "arg is multibyte characters" do
+        let(:arg) do
+          ->(a) { a + "ほげほげ" }
+        end
+
+        it { should eq '->(a) { a + "ほげほげ" }' }
+        its(:encoding) { should eq Encoding::UTF_8 }
+      end
+    end
+  end
+end