spec/spec_helper.rb in memoized-1.1.0 vs spec/spec_helper.rb in memoized-1.1.1

@@ -1,9 +1,88 @@
 require 'date'
 require 'timecop'
 require 'memoized'
+if Gem::Version.new(RUBY_VERSION) > Gem::Version.new('2.5.3')
+  require 'prop_check'
+end
 
 RSpec.configure do |config|
   config.warnings = true
   config.order = :random
   Kernel.srand config.seed
 end
+
+module Memoized
+  class Parameters
+    attr_accessor :args_count, :kwargs_count
+
+    def test_body
+      params.map(&Parameters.method(:to_test_body)).join(" * ")
+    end
+
+    def self.to_test_body((param_type, param_name))
+      case param_type
+      when :req
+        "#{param_name}"
+      when :opt
+        "#{param_name}"
+      when :rest
+        "#{param_name}.inject(&:*)"
+      when :keyreq
+        "#{param_name}"
+      when :key
+        "#{param_name}"
+      when :keyrest
+        "#{param_name}.values.inject(&:*)"
+      else raise "unknown parameter type"
+      end
+    end
+
+    def test_arguments
+      params.map(&Parameters.new([], @args_count, @kwargs_count).method(:to_test_arguments)).join(", ")
+    end
+
+    def to_test_arguments((param_type, param_name))
+      case param_type
+      when :req
+        "2"
+      when :opt
+        "3"
+      when :rest
+        if @args_count.nil?
+          "5, 5, 5"
+        else
+          (["5"] * @args_count).join(', ')
+        end
+      when :keyreq
+        "#{param_name}: 7"
+      when :key
+        "#{param_name}: 11"
+      when :keyrest
+        if @kwargs_count.nil?
+          "**{ first: 13, second: 13 }"
+        else
+          kwargs_list = (1..@kwargs_count).map.with_index do |counter|
+            "kwarg_#{counter}: 13"
+          end
+          "**{ #{kwargs_list.join(', ')} }"
+        end
+      else raise "unknown parameter type"
+      end
+    end
+
+    def debug_info
+      "#{@req_params.size} - #{@opt_params.size} - #{@args_count || @rest_params.size} " \
+        "| #{@keyreq_params.size} - #{@key_params.size} - #{@kwargs_count || @keyrest_params.size}"
+    end
+  end
+end
+
+module ParametersTestExtension
+  def initialize(parameters = [], args_count = nil, kwargs_count = nil)
+    super(parameters)
+    @args_count = args_count
+    @kwargs_count = kwargs_count
+  end
+end
+
+Memoized::Parameters.prepend(ParametersTestExtension)