spec/unit/error_spec.rb in tty-prompt-0.1.0 vs spec/unit/error_spec.rb in tty-prompt-0.2.0

@@ -1,30 +1,26 @@
 # encoding: utf-8
 
 require 'spec_helper'
 
 RSpec.describe TTY::Prompt, '.error' do
-  let(:input)  { StringIO.new }
-  let(:output) { StringIO.new }
   let(:color)  { Pastel.new(enabled: true) }
 
-  subject(:prompt) { described_class.new(input, output) }
+  subject(:prompt) { TTY::TestPrompt.new }
 
   before { allow(Pastel).to receive(:new).and_return(color) }
 
-  after { output.rewind }
-
   it 'displays one message' do
     prompt.error "Nothing is fine!"
-    expect(output.string).to eql "\e[31mNothing is fine!\e[0m\n"
+    expect(prompt.output.string).to eql "\e[31mNothing is fine!\e[0m\n"
   end
 
   it 'displays many messages' do
     prompt.error "Nothing is fine!", "All is broken!"
-    expect(output.string).to eql "\e[31mNothing is fine!\e[0m\n\e[31mAll is broken!\e[0m\n"
+    expect(prompt.output.string).to eql "\e[31mNothing is fine!\e[0m\n\e[31mAll is broken!\e[0m\n"
   end
 
   it 'displays message with option' do
     prompt.error "Nothing is fine!", newline: false
-    expect(output.string).to eql "\e[31mNothing is fine!\e[0m"
+    expect(prompt.output.string).to eql "\e[31mNothing is fine!\e[0m"
   end
 end