Add #before_store method

Author: rastating

lib/wpxf/wordpress/stored_xss.rb

@@ -40,6 +40,12 @@ def store_script_and_validate
     false
   end
 
+  # Execute all tasks required before storing the script.
+  # @return [Boolean] return true if the prerequisite actions were successfully executed.
+  def before_store
+    true
+  end
+
   # @return [Number] The status code that is expected after storing the script.
   def expected_status_code_after_store
     200
@@ -48,7 +54,7 @@ def expected_status_code_after_store
   # Run the module.
   # @return [Boolean] true if successful.
   def run
-    return false unless super
+    return false unless super && before_store
 
     emit_info 'Storing script...'
     return false unless store_script_and_validate

spec/wordpress/stored_xss_spec.rb

@@ -32,6 +32,15 @@
       subject.run
       expect(invoked).to be true
     end
+
+    it 'returns false if #before_store returns false' do
+      allow(subject).to receive(:before_store).and_return false
+      allow(subject).to receive(:puts).and_return nil
+      allow(subject).to receive(:check_wordpress_and_online).and_return true
+      allow(subject).to receive(:start_http_server).and_return true
+
+      expect(subject.run).to be false
+    end
   end
 
   describe '#store_script' do