Allow stubbing stdin in msfupdate

jbarnett-r7 · Nov 15, 2013 · e6c43bf · e6c43bf
1 parent faf8099
commit e6c43bf
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/msfupdate b/msfupdate
@@ -15,11 +15,13 @@ end
 
 
 class Msfupdate
+  attr_reader :stdin
   attr_reader :stdout
   attr_reader :stderr
 
-  def initialize(msfbase_dir, stdout=$stdout, stderr=$stderr)
+  def initialize(msfbase_dir, stdin=$stdin, stdout=$stdout, stderr=$stderr)
     @msfbase_dir = msfbase_dir
+    @stdin = stdin
     @stdout = stdout
     @stderr = stderr
   end

diff --git a/spec/msfupdate_spec.rb b/spec/msfupdate_spec.rb
@@ -24,11 +24,12 @@ def dummy_apt_pathname
     dummy_git_pathname
   end
 
+  let(:stdin)  { StringIO.new("", "rb") }
   let(:stdout) { StringIO.new("", "wb") }
   let(:stderr) { StringIO.new("", "wb") }
 
   subject do
-    Msfupdate.new(msfbase_dir, stdout, stderr)
+    Msfupdate.new(msfbase_dir, stdin, stdout, stderr)
   end
 
   before(:all) do