Fix rspec

Author: wchen-r7

spec/lib/msf/core/exploit/browser_autopwnv2_spec.rb

@@ -18,7 +18,7 @@ def mock_note_destroy
     # This empties it
     notes = []
 
-    db = double('db')
+    db = double('db1')
     allow(db).to receive(:notes).and_return(notes)
     allow(framework).to receive(:db).and_return(db)
 
@@ -28,14 +28,14 @@ def mock_note_destroy
   def mock_report_note(arg)
     framework = double('Msf::Framework', datastore: {})
     notes = [create_fake_note('bap.clicks')]
-    db = double('db')
+    db = double('db2')
     allow(db).to receive(:notes).and_return(notes)
     allow(framework).to receive(:db).and_return(db)
     allow(subject).to receive(:framework).and_return(framework)
   end
 
   def create_fake_note(tag, data='')
-    note = double('note')
+    note = double('note3')
 
     allow(note).to receive(:ntype).and_return(tag)
     allow(note).to receive(:data).and_return(data)
@@ -65,6 +65,7 @@ def create_fake_exploit(opts={})
     datastore_options = opts[:datastore_options] || {}
     job_id            = opts[:job_id] || 0
     requirements      = opts[:requirements] || {}
+    workspace         = opts[:workspace] || 'default'
 
     mod = Msf::Exploit.new
     mod.extend(Msf::Exploit::Remote::BrowserExploitServer)
@@ -75,6 +76,7 @@ def create_fake_exploit(opts={})
     allow(mod).to receive(:compatible_payloads).and_return(compat_payloads)
     allow(mod).to receive(:datastore).and_return(datastore_options)
     allow(mod).to receive(:job_id).and_return(job_id)
+    allow(mod).to receive(:workspace).and_return(workspace)
     allow(mod).to receive(:exploit_simple)
     allow(mod).to receive(:vprint_status)
     allow(mod).to receive(:shortname).and_return(short_name)
@@ -139,6 +141,7 @@ def create_fake_payload(opts={})
     datastores = opts[:datastore_options]
     fullname   = opts[:fullname]
     shortname  = opts[:shortname]
+    workspace  = opts[:workspace] || 'default'
 
     p = Msf::Payload.new
 
@@ -154,8 +157,11 @@ def create_fake_payload(opts={})
       p.datastore[key] = value
     end
 
+    datastores['WORKSPACE'] = workspace
+
     allow(p).to receive(:fullname).and_return(fullname)
     allow(p).to receive(:shoftname).and_return(shortname)
+    allow(p).to receive(:workspace).and_return(workspace)
 
     p
   end
@@ -329,10 +335,14 @@ def mock_exploit_create(full_name)
     notes = [create_fake_note("#{note_type_prefix}.#{profile_tag}", profile_packed_data)]
 
     # Prepare framework.db
+    w = double('workspace')
+    allow(w).to receive(:name).and_return('WORKSPACE')
+
     db = double('db')
     allow(db).to receive(:report_note).with(kind_of(Hash)) { |arg| mock_report_note(arg) }
     allow(db).to receive(:notes).and_return(notes)
     allow(db).to receive(:active).and_return(true)
+    allow(db).to receive(:workspace).and_return(w)
     allow(framework).to receive(:db).and_return(db)
 
     # Prepare framework.exploits
@@ -710,20 +720,6 @@ def get_stdout(&block)
       # You got me, I don't know how to implement this one because the super"
     end
 
-    describe '#show_payloads' do
-      it 'shows payloads' do
-        output = get_stdout { subject.show_payloads }
-        expect(output).to include(windows_meterpreter_reverse_tcp)
-        expect(output).to include(linux_meterpreter_reverse_tcp)
-      end
-    end
-
-    describe '#set_payload' do
-      it 'shows set_payload' do
-        output = get_stdout { subject.set_payload }
-        expect(output).to include('\'set payload\' has been disabled for BrowserAutoPwn')
-      end
-    end
   end
 
   describe '#log_click' do