diff --git a/spec/beef/extensions/network_spec.rb b/spec/beef/extensions/network_spec.rb index a72c5d093..1d5029db3 100644 --- a/spec/beef/extensions/network_spec.rb +++ b/spec/beef/extensions/network_spec.rb @@ -3,13 +3,20 @@ require 'extensions/network/models/network_host' RSpec.describe 'BeEF Extension Network' do - it 'add good host' do + it 'add good local host' do expect { BeEF::Core::Models::NetworkHost.create(:hooked_browser_id => '1234', :ip => '127.0.0.1') }.to_not raise_error expect(BeEF::Core::Models::NetworkHost.where(hooked_browser_id: '1234', ip: '127.0.0.1')).to_not be_empty end + it 'add good not local host' do + expect { + BeEF::Core::Models::NetworkHost.create(:hooked_browser_id => '12', :ip => '192.168.1.2') + }.to_not raise_error + expect(BeEF::Core::Models::NetworkHost.where(hooked_browser_id: '12', ip: '192.168.1.2')).to_not be_empty + end + it 'add good service' do expect { BeEF::Core::Models::NetworkService.create(:hooked_browser_id => '1234', :proto => 'http', :ip => '127.0.0.1', :port => 80, :ntype => 'Apache')