From a5a970f12d49f510b0a1f0e7b5ec676108ec16d6 Mon Sep 17 00:00:00 2001
From: Sebastian Ludwig <sebastian@lurado.de>
Date: Fri, 4 Dec 2015 23:22:10 +0100
Subject: [PATCH] Added variables for STDOUT and STDERR to be able to redirect
 the output in unit tests.

---
 lib/twine.rb                     | 19 +++++++++++++++++++
 lib/twine/formatters/abstract.rb |  4 ++--
 lib/twine/runner.rb              |  6 +++---
 test/twine_test_case.rb          |  3 +++
 4 files changed, 27 insertions(+), 5 deletions(-)

diff --git a/lib/twine.rb b/lib/twine.rb
index 3f9e5e0..386d4b2 100644
--- a/lib/twine.rb
+++ b/lib/twine.rb
@@ -1,4 +1,23 @@
 module Twine
+  @@stdout = STDOUT
+  @@stderr = STDERR
+
+  def self.stdout
+    @@stdout
+  end
+
+  def self.stdout=(out)
+    @@stdout = out
+  end
+
+  def self.stderr
+    @@stderr
+  end
+
+  def self.stderr=(err)
+    @@stderr = err
+  end
+
   class Error < StandardError
   end
 
diff --git a/lib/twine/formatters/abstract.rb b/lib/twine/formatters/abstract.rb
index 4d9c361..4dd3b72 100644
--- a/lib/twine/formatters/abstract.rb
+++ b/lib/twine/formatters/abstract.rb
@@ -79,7 +79,7 @@ module Twine
             row.translations[lang] = value
           end
         elsif @options[:consume_all]
-          STDERR.puts "Adding new string '#{key}' to strings data file."
+          Twine::stderr.puts "Adding new string '#{key}' to strings data file."
           current_section = @strings.sections.find { |s| s.name == 'Uncategorized' }
           unless current_section
             current_section = StringsSection.new('Uncategorized')
@@ -95,7 +95,7 @@ module Twine
           @strings.strings_map[key] = current_row
           @strings.strings_map[key].translations[lang] = value
         else
-          STDERR.puts "Warning: '#{key}' not found in strings data file."
+          Twine::stderr.puts "Warning: '#{key}' not found in strings data file."
         end
         if !@strings.language_codes.include?(lang)
           @strings.add_language_code(lang)
diff --git a/lib/twine/runner.rb b/lib/twine/runner.rb
index 103052a..3b91d19 100644
--- a/lib/twine/runner.rb
+++ b/lib/twine/runner.rb
@@ -89,7 +89,7 @@ module Twine
           begin
             read_write_string_file(item, true, nil)
           rescue Twine::Error => e
-            STDERR.puts "#{e.message}"
+            Twine::stderr.puts "#{e.message}"
           end
         end
       end
@@ -168,7 +168,7 @@ module Twine
               begin
                 read_write_string_file(real_path, true, nil)
               rescue Twine::Error => e
-                STDERR.puts "#{e.message}"
+                Twine::stderr.puts "#{e.message}"
               end
             end
           end
@@ -218,7 +218,7 @@ module Twine
         raise Twine::Error.new errors.join("\n\n")
       end
 
-      puts "#{@options[:strings_file]} is valid."
+      Twine::stdout.puts "#{@options[:strings_file]} is valid."
     end
 
     def determine_language_given_path(path)
diff --git a/test/twine_test_case.rb b/test/twine_test_case.rb
index 2e8f593..c1657a6 100644
--- a/test/twine_test_case.rb
+++ b/test/twine_test_case.rb
@@ -2,6 +2,7 @@ require 'erb'
 require 'minitest/autorun'
 require "mocha/mini_test"
 require 'securerandom'
+require 'stringio'
 require 'twine'
 require 'twine_file_dsl'
 
@@ -12,6 +13,8 @@ class TwineTestCase < Minitest::Test
   
   def setup
     super
+    Twine::stdout = StringIO.new
+    Twine::stderr = StringIO.new
     @output_dir = Dir.mktmpdir
     @output_path = File.join @output_dir, SecureRandom.uuid
   end