[ruby/irb] Stub a screen size for test_context

http://ci.rvm.jp/logfiles/brlog.trunk-random1.20210119-074232

https://github.com/ruby/irb/commit/ea87592d4a

2 files changed, 11 insertions, 2 deletions

diff --git a/test/irb/test_color.rb b/test/irb/test_color.rb
index 459802181e..50dfc22568 100644
--- a/test/irb/test_color.rb
+++ b/test/irb/test_color.rb
@@ -19,7 +19,6 @@ module TestIRB
     CYAN      = "\e[36m"
 
     def setup
-      super
       @get_screen_size = Reline.method(:get_screen_size)
       Reline.instance_eval { undef :get_screen_size }
       def Reline.get_screen_size
@@ -30,7 +29,6 @@ module TestIRB
     def teardown
       Reline.instance_eval { undef :get_screen_size }
       Reline.define_singleton_method(:get_screen_size, @get_screen_size)
-      super
     end
 
     def test_colorize_code
diff --git a/test/irb/test_context.rb b/test/irb/test_context.rb
index 2dba42d321..63e71e5867 100644
--- a/test/irb/test_context.rb
+++ b/test/irb/test_context.rb
@@ -42,6 +42,17 @@ module TestIRB
       IRB.conf[:VERBOSE] = false
       workspace = IRB::WorkSpace.new(Object.new)
       @context = IRB::Context.new(nil, workspace, TestInputMethod.new)
+
+      @get_screen_size = Reline.method(:get_screen_size)
+      Reline.instance_eval { undef :get_screen_size }
+      def Reline.get_screen_size
+        [36, 80]
+      end
+    end
+
+    def teardown
+      Reline.instance_eval { undef :get_screen_size }
+      Reline.define_singleton_method(:get_screen_size, @get_screen_size)
     end
 
     def test_last_value