Update state to archived_with_draft when archived form is updated

app/controllers/api/v1/forms_controller.rb

@@ -29,6 +29,7 @@ def update
       # TODO: https://trello.com/c/dg9CFPgp/1503-user-triggers-state-change-from-live-to-livewithdraft
       # Will not be needed when users can trigger this event themselves through the UI
       form.create_draft_from_live_form! if form.live?
+      form.create_draft_from_archived_form! if form.archived?
 
       render json: form.to_json, status: :ok
     else

spec/factories/forms.rb

@@ -55,6 +55,11 @@
       after(:create, &:make_live!)
     end
 
+    trait :archived do
+      live
+      after(:create, &:archive_live_form!)
+    end
+
     trait :with_support do
       support_email { Faker::Internet.email(domain: "example.gov.uk") }
       support_phone { Faker::Lorem.paragraph(sentence_count: 2, supplemental: true, random_sentences_to_add: 4) }

spec/requests/api/v1/forms_controller_spec.rb

@@ -215,6 +215,8 @@
   end
 
   describe "#update" do
+    let(:form) { create :form }
+
     it "when no forms exists for an id, returns 404 an error" do
       put form_path(id: 999), as: :json
       expect(response).to have_http_status(:not_found)
@@ -223,22 +225,37 @@
     end
 
     it "when given an valid id and params, updates DB and returns 200" do
-      form1 = create :form
-      put form_path(form1), params: { submission_email: "[email protected]" }, as: :json
+      put form_path(form), params: { submission_email: "[email protected]" }, as: :json
       expect(response).to have_http_status(:ok)
       expect(response.headers["Content-Type"]).to eq("application/json")
       expect(json_body).to include(submission_email: "[email protected]")
-      expect(form1.reload.submission_email).to eq("[email protected]")
+      expect(form.reload.submission_email).to eq("[email protected]")
     end
 
     it "ignores created_at" do
-      form1 = create :form
-      expect { put form_path(form1), params: { created_at: "2023-01-11T16:22:22.661+00:00" }, as: :json }.not_to(change { form1.reload.created_at })
+      expect { put form_path(form), params: { created_at: "2023-01-11T16:22:22.661+00:00" }, as: :json }.not_to(change { form.reload.created_at })
     end
 
     it "ignores updated_at" do
-      form1 = create :form
-      expect { put form_path(form1), params: { updated_at: "2023-01-11T16:22:22.661+00:00" }, as: :json }.not_to(change { form1.reload.updated_at })
+      expect { put form_path(form), params: { updated_at: "2023-01-11T16:22:22.661+00:00" }, as: :json }.not_to(change { form.reload.updated_at })
+    end
+
+    context "when form is live" do
+      let(:form) { create(:form, :live) }
+
+      it "updates form state to live_with_draft" do
+        put form_path(form), params: { submission_email: "[email protected]" }, as: :json
+        expect(form.reload.state).to eq("live_with_draft")
+      end
+    end
+
+    context "when form is archived" do
+      let(:form) { create(:form, :archived) }
+
+      it "updates form state to archived_with_draft" do
+        put form_path(form), params: { submission_email: "[email protected]" }, as: :json
+        expect(form.reload.state).to eq("archived_with_draft")
+      end
     end
   end