diff --git a/extensions/social_engineering/models/web_cloner.rb b/extensions/social_engineering/models/web_cloner.rb index 0287868c1..1f645d3c8 100644 --- a/extensions/social_engineering/models/web_cloner.rb +++ b/extensions/social_engineering/models/web_cloner.rb @@ -6,7 +6,7 @@ module BeEF module Core module Models - class Webcloner < BeEF::Core::Model + class WebCloner < BeEF::Core::Model has_many :interceptors diff --git a/extensions/social_engineering/web_cloner/web_cloner.rb b/extensions/social_engineering/web_cloner/web_cloner.rb index 478df6906..8de4cfd75 100644 --- a/extensions/social_engineering/web_cloner/web_cloner.rb +++ b/extensions/social_engineering/web_cloner/web_cloner.rb @@ -206,7 +206,7 @@ module BeEF end def persist_page(uri, mount) - webcloner_db = BeEF::Core::Models::Webcloner.new( + webcloner_db = BeEF::Core::Models::WebCloner.new( :uri => uri, :mount => mount ) diff --git a/spec/beef/extensions/social_engineering_spec.rb b/spec/beef/extensions/social_engineering_spec.rb new file mode 100644 index 000000000..ab19d8475 --- /dev/null +++ b/spec/beef/extensions/social_engineering_spec.rb @@ -0,0 +1,22 @@ +require 'extensions/social_engineering/models/web_cloner' +require 'extensions/social_engineering/web_cloner/web_cloner' +require 'extensions/social_engineering/web_cloner/interceptor' +require 'extensions/social_engineering/models/interceptor' +require 'fileutils' + +RSpec.describe 'BeEF Extension Social Engineering' do + + it 'persistence web cloner' do + expect { + BeEF::Core::Models::WebCloner.create(uri: "example.com", mount: "/") + }.to_not raise_error + end + + it 'clone web page' do + expect { + BeEF::Extension::SocialEngineering::WebCloner.instance.clone_page("https://www.google.com", "/", nil, nil) + }.to_not raise_error + FileUtils.rm(Dir['./extensions/social_engineering/web_cloner/cloned_pages/www.google.com']) + FileUtils.rm(Dir['./extensions/social_engineering/web_cloner/cloned_pages/www.google.com_mod']) + end +end