add helpers for formula tests

1 files changed, 23 insertions, 0 deletions

diff --git a/Library/Homebrew/formula_assertions.rb b/Library/Homebrew/formula_assertions.rb
new file mode 100644
index 000000000..845c5480d
--- /dev/null
+++ b/Library/Homebrew/formula_assertions.rb
@@ -0,0 +1,23 @@
+require 'test/unit/assertions'
+
+module Homebrew
+  module Assertions
+    include Test::Unit::Assertions
+
+    # Returns the output of running cmd, and asserts the exit status
+    def shell_output(cmd, result=0)
+      output = `#{cmd}`
+      assert_equal result, $?.exitstatus
+      output
+    end
+
+    # Returns the output of running the cmd, with the optional input
+    def pipe_output(cmd, input=nil)
+      IO.popen(cmd, "w+") do |pipe|
+        pipe.write(input) unless input.nil?
+        pipe.close_write
+        pipe.read
+      end
+    end
+  end
+end