test_application.rb in rubikon-0.5.0

- old
+ new

@@ -41,12 +41,16 @@
         @app.run(%w{unknown})
       rescue Exception => e
         assert_instance_of SystemExit, e
         assert_equal 1, e.status
       end
-      @ostream.rewind
-      assert_equal "Error:\n", @ostream.gets
+      if @app.instance.instance_eval { @settings[:colors] }
+        error_text = "\e[0;31mError:\e[0m\n"
+      else
+        error_text = "Error:\n"
+      end
+      assert_equal error_text, @ostream.gets
       assert_equal "    Unknown command: unknown\n", @ostream.gets
     end
 
     should 'run its default command without arguments' do
       assert_equal 'default command', @app.run([])
@@ -70,19 +74,15 @@
 
       input_string = 'test'
       @istream.puts input_string
       @istream.rewind
       assert_equal input_string, @app.run(%w{input})
-      @ostream.rewind
       assert_equal 'input: ', @ostream.gets
     end
 
     should 'not break output while displaying a throbber or progress bar' do
       @app.run(%w{throbber})
       assert_match (/ \x08(?:(?:-|\\|\/|\|)\x08){4,}don't\nbreak\n/), @ostream.string
-
-      @ostream.rewind
-
       @app.run(%w{progressbar})
       assert_equal "#" * 20 << "\n" << "test\n" * 4, @ostream.string
     end
 
     should 'have working command aliases' do