From 45c26343a159758cccb50ea5d92442bc092c3bfa Mon Sep 17 00:00:00 2001
From: James Lee <egypt@metasploit.com>
Date: Thu, 5 Jun 2014 11:51:11 -0500
Subject: [PATCH] Add spec for pass_file

---
 .../framework/credential_collection_spec.rb   | 23 +++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/spec/lib/metasploit/framework/credential_collection_spec.rb b/spec/lib/metasploit/framework/credential_collection_spec.rb
index ae8d715a23..80cba3097d 100644
--- a/spec/lib/metasploit/framework/credential_collection_spec.rb
+++ b/spec/lib/metasploit/framework/credential_collection_spec.rb
@@ -9,12 +9,16 @@ describe Metasploit::Framework::CredentialCollection do
         username: username,
         password: password,
         user_file: user_file,
+        pass_file: pass_file,
+        userpass_file: userpass_file,
       )
     end
 
     let(:username) { "user" }
     let(:password) { "pass" }
     let(:user_file) { nil }
+    let(:pass_file) { nil }
+    let(:userpass_file) { nil }
 
     specify do
       expect { |b| collection.each(&b) }.to yield_with_args(Metasploit::Framework::Credential)
@@ -37,6 +41,25 @@ describe Metasploit::Framework::CredentialCollection do
         )
       end
     end
+
+    context "when given a pass_file and username" do
+      let(:password) { nil }
+      let(:pass_file) do
+        filename = "foo"
+        stub_file = StringIO.new("asdf\njkl\n")
+        File.stub(:open).with(filename,/^r/).and_return stub_file
+
+        filename
+      end
+
+      specify  do
+        expect { |b| collection.each(&b) }.to yield_successive_args(
+          Metasploit::Framework::Credential.new(public: username, private: "asdf"),
+          Metasploit::Framework::Credential.new(public: username, private: "jkl"),
+        )
+      end
+    end
+
   end
 
 end