test/mgmt-iostat: allow RUBY env to override "/usr/bin/env ruby"

The ruby executable name is "ruby19" on some systems, so
"/usr/bin/env ruby" will do bad things to us.

1 files changed, 7 insertions, 2 deletions

diff --git a/test/mgmt-iostat.rb b/test/mgmt-iostat.rb
index a6365ec..944661c 100644
--- a/test/mgmt-iostat.rb
+++ b/test/mgmt-iostat.rb
@@ -6,6 +6,11 @@ require 'test/test_helper'
 
 class TestMgmtIostat < Test::Unit::TestCase
   TEST_PATH = File.dirname(__FILE__) + ":#{ENV['PATH']}"
+  RUBY = ENV["RUBY"] || "ruby"
+
+  def iostat_mock
+    "#{RUBY} #{File.dirname(__FILE__)}/iostat-mock.rb"
+  end
 
   def setup
     @iostat_pid = Tempfile.new('testt-iostat-pid')
@@ -46,7 +51,7 @@ class TestMgmtIostat < Test::Unit::TestCase
     Dir.mkdir "#@tmpdir/dev666"
     @pid = fork do
       ENV["PATH"] = TEST_PATH
-      ENV["MOG_IOSTAT_CMD"] = "iostat-mock.rb #{@iostat_pid.path} fast"
+      ENV["MOG_IOSTAT_CMD"] = "#{iostat_mock} #{@iostat_pid.path} fast"
       exec(*@cmd)
     end
 
@@ -93,7 +98,7 @@ class TestMgmtIostat < Test::Unit::TestCase
     Dir.mkdir "#@tmpdir/dev666"
     @pid = fork do
       ENV["PATH"] = TEST_PATH
-      ENV["MOG_IOSTAT_CMD"] = "iostat-mock.rb #{@iostat_pid.path} #{type}"
+      ENV["MOG_IOSTAT_CMD"] = "#{iostat_mock} #{@iostat_pid.path} #{type}"
       exec(*@cmd)
     end