Make Console more testable with some overrides

Author: acook

lib/remedy/console.rb

@@ -23,6 +23,14 @@ def output
       @output ||= $stdout
     end
 
+    def input= new_input
+      @input = new_input
+    end
+
+    def output= new_output
+      @output = new_output
+    end
+
     def raw
       raw!
       result = yield
@@ -52,14 +60,20 @@ def rows
     alias_method :height, :rows
 
     def size
+      return @size_override if @size_override
+
       str = [0, 0, 0, 0].pack('SSSS')
-      if input.ioctl(TIOCGWINSZ, str) >= 0 then
+      if input.respond_to?(:ioctl) && input.ioctl(TIOCGWINSZ, str) >= 0 then
         Size.new str.unpack('SSSS').first 2
       else
         raise UnknownConsoleSize, "Unable to get console size"
       end
     end
 
+    def size_override= new_size
+      @size_override = new_size
+    end
+
     def interactive?
       input.isatty
     end

spec/viewport_spec.rb

@@ -1,19 +1,34 @@
-require_relative 'spec_helper'
-require 'remedy/viewport'
+require_relative "spec_helper"
+require "remedy/viewport"
 
 describe Remedy::Viewport do
+  let(:console){ ::Remedy::Console }
+  let(:size_override){ Size 20, 40 }
+  let(:stringio){ StringIO.new }
+
+  before(:each) do
+    console.input = stringio
+    console.output = stringio
+    console.size_override = size_override
+  end
+
+  after(:each) do
+    console.input = $stdin
+    console.output = $stdout
+    console.size_override = nil
+  end
+
   it 'should be able to execute the example code from the readme' do
-    @stdout = $stdout
+    expected = "\\e[H\\e[JQ: What's the difference between a duck?\\e[1B\\e[0GA: Purple, because ice cream has no bone\\e[1B\\e[0G"
+
     joke = ::Remedy::Partial.new
     joke << "Q: What's the difference between a duck?"
     joke << "A: Purple, because ice cream has no bones!"
 
     screen = ::Remedy::Viewport.new
-    sio = StringIO.new
-    $stdout = sio
-    screen.draw joke unless ENV['CI']
-    expect(sio.string).to include("ice cream")
-  ensure
-    $stdout = @stdout
+    screen.draw joke
+
+    actual = stringio.string.inspect[1..-2]
+    expect(actual).to eq expected
   end
 end