From 52bb32d6b71365cb24273de3eed5a712206815f3 Mon Sep 17 00:00:00 2001
From: Alan Wu <XrXr@users.noreply.github.com>
Date: Mon, 30 Dec 2019 18:13:55 -0500
Subject: Fix Proc#<< spec

[Bug #16406]
---
 spec/ruby/core/proc/compose_spec.rb | 46 +++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)

(limited to 'spec')

diff --git a/spec/ruby/core/proc/compose_spec.rb b/spec/ruby/core/proc/compose_spec.rb
index 35e949a779..ef9c125ae9 100644
--- a/spec/ruby/core/proc/compose_spec.rb
+++ b/spec/ruby/core/proc/compose_spec.rb
@@ -38,6 +38,36 @@ ruby_version_is "2.6" do
         (f << g).lambda?.should == false
       end
 
+      ruby_version_is(''...'2.8') do
+        it "is a Proc when other is lambda" do
+          f = proc { |x| x * x }
+          g = -> x { x + x }
+
+          (f << g).is_a?(Proc).should == true
+          (f << g).lambda?.should == false
+        end
+
+        it "is a lambda when self is lambda" do
+          f = -> x { x * x }
+          g = proc { |x| x + x }
+
+          (f << g).is_a?(Proc).should == true
+          (f << g).lambda?.should == true
+        end
+      end
+
+      ruby_version_is('2.8') do
+        it "is a lambda when parameter is lambda" do
+          f = -> x { x * x }
+          g = proc { |x| x + x }
+          lambda_proc = -> x { x }
+
+          (f << g).is_a?(Proc).should == true
+          (f << g).lambda?.should == false
+          (f << lambda_proc).lambda?.should == true
+        end
+      end
+
       it "may accept multiple arguments" do
         inc = proc { |n| n + 1 }
         mul = proc { |n, m| n * m }
@@ -91,6 +121,22 @@ ruby_version_is "2.6" do
         (f >> g).lambda?.should == false
       end
 
+      it "is a Proc when other is lambda" do
+        f = proc { |x| x * x }
+        g = -> x { x + x }
+
+        (f >> g).is_a?(Proc).should == true
+        (f >> g).lambda?.should == false
+      end
+
+      it "is a lambda when self is lambda" do
+        f = -> x { x * x }
+        g = proc { |x| x + x }
+
+        (f >> g).is_a?(Proc).should == true
+        (f >> g).lambda?.should == true
+      end
+
       it "may accept multiple arguments" do
         inc = proc { |n| n + 1 }
         mul = proc { |n, m| n * m }
-- 
cgit v1.2.3