Merge pull request #138 from paviliondev/maybe_later_bug

FIX: clear submission if skipped by user

controllers/custom_wizard/wizard.rb

@@ -68,7 +68,8 @@ class CustomWizard::WizardController < ::ApplicationController
         result.merge!(redirect_to: submission.redirect_to)
       end
 
-      wizard.final_cleanup!
+      submission.remove if submission.present?
+      wizard.reset
     end
 
     render json: result

lib/custom_wizard/submission.rb

@@ -97,6 +97,17 @@ class CustomWizard::Submission
     new(wizard, data, user_id)
   end
 
+  def remove
+    if present?
+      user_id = @user.id
+      wizard_id = @wizard.id
+      submission_id = @id
+      data = PluginStore.get("#{wizard_id}_#{KEY}", user_id)
+      data.delete_if { |sub| sub["id"] == submission_id }
+      PluginStore.set("#{wizard_id}_#{KEY}", user_id, data)
+    end
+  end
+
   def self.cleanup_incomplete_submissions(wizard)
     user_id = wizard.user.id
     all_submissions = list(wizard, user_id: user_id)

spec/requests/custom_wizard/wizard_controller_spec.rb

@@ -50,6 +50,8 @@ describe CustomWizard::WizardController do
     expect(response.parsed_body["error"]).to eq("We couldn't find a wizard at that address.")
   end
 
+  context 'when user skips the wizard' do
+
     it 'skips a wizard if user is allowed to skip' do
       put '/w/super-mega-fun-wizard/skip.json'
       expect(response.status).to eq(200)
@@ -76,4 +78,27 @@ describe CustomWizard::WizardController do
       put '/w/super-mega-fun-wizard/skip.json'
       expect(response.parsed_body['redirect_to']).to eq('/t/2')
     end
+
+    it "deletes the submission if user has filled up some data" do
+      @wizard = CustomWizard::Wizard.create(@template["id"], user)
+      CustomWizard::Submission.new(@wizard, step_1_field_1: "Hello World").save
+      current_submission = @wizard.current_submission
+      put '/w/super-mega-fun-wizard/skip.json'
+      list = CustomWizard::Submission.list(@wizard)
+
+      expect(list.any? { |submission| submission.id == current_submission.id }).to eq(false)
+    end
+
+    it "starts from the first step if user visits after skipping the wizard" do
+      put '/w/super-mega-fun-wizard/steps/step_1.json', params: {
+        fields: {
+          step_1_field_1: "Text input"
+        }
+      }
+      put '/w/super-mega-fun-wizard/skip.json'
+      get '/w/super-mega-fun-wizard.json'
+
+      expect(response.parsed_body["start"]).to eq('step_1')
+    end
+  end
 end