HELLO v3!!!

spec/app/controllers/custom_directories_spec.rb

@@ -0,0 +1,45 @@
+require 'spec_helper'
+
+describe WPScan::Controller::CustomDirectories do
+  subject(:controller) { described_class.new }
+  let(:target_url)     { 'http://ex.lo/' }
+  let(:parsed_options) { rspec_parsed_options(cli_args) }
+  let(:cli_args)       { "--url #{target_url}" }
+
+  before do
+    WPScan::Browser.reset
+    described_class.parsed_options = parsed_options
+  end
+
+  describe '#cli_options' do
+    its(:cli_options) { should_not be_empty }
+    its(:cli_options) { should be_a Array }
+
+    it 'contains to correct options' do
+      expect(controller.cli_options.map(&:to_sym)).to eq %i[wp_content_dir wp_plugins_dir]
+    end
+  end
+
+  describe '#before_scan' do
+    context 'when the content_dir is not found and not supply' do
+      before { expect(controller.target).to receive(:content_dir) }
+
+      let(:exception) do
+        'Unable to identify the wp-content dir, please supply it with --wp-content-dir'
+      end
+
+      it 'raises an exception' do
+        expect { controller.before_scan }.to raise_error(exception)
+      end
+    end
+
+    context 'when content_dir found/supplied' do
+      let(:cli_args) { "#{super()} --wp-content-dir wp-content" }
+
+      it 'does not raise any error' do
+        expect { controller.before_scan }.to_not raise_error
+        expect(controller.target.content_dir).to eq parsed_options[:wp_content_dir]
+      end
+    end
+  end
+end