patch address routing for no user=fi

hmrc · Aug 29, 2024 · 347dff8 · 347dff8
1 parent 4cafbcc
commit 347dff8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/app/viewmodels/checkAnswers/CheckYourAnswersViewModel.scala b/app/viewmodels/checkAnswers/CheckYourAnswersViewModel.scala
@@ -55,9 +55,8 @@ object CheckYourAnswersViewModel {
   def getAddressChangeRoute(answers: UserAnswers): String =
     answers
       .get(ReportForRegisteredBusinessPage) match {
-      case Some(true)  => controllers.addFinancialInstitution.registeredBusiness.routes.IsTheAddressCorrectController.onPageLoad(CheckMode).url
-      case Some(false) => controllers.addFinancialInstitution.routes.WhereIsFIBasedController.onPageLoad(CheckMode).url
-      case None        => controllers.routes.IndexController.onPageLoad().url
+      case Some(true) => controllers.addFinancialInstitution.registeredBusiness.routes.IsTheAddressCorrectController.onPageLoad(CheckMode).url
+      case _          => controllers.addFinancialInstitution.routes.WhereIsFIBasedController.onPageLoad(CheckMode).url
     }
 
   private def getGIINRows(ua: UserAnswers)(implicit messages: Messages): Seq[SummaryListRow] = {

diff --git a/test/viewmodels/checkAnswers/CheckYourAnswersViewModelSpec.scala b/test/viewmodels/checkAnswers/CheckYourAnswersViewModelSpec.scala
@@ -87,6 +87,9 @@ class CheckYourAnswersViewModelSpec extends SpecBase {
     "getAddressChangeRoute" - {
       "must be WhereIsFIBased when adding another business" in {
         val answers = ua.withPage(ReportForRegisteredBusinessPage, false)
+
+        sut.getAddressChangeRoute(ua) mustBe
+          controllers.addFinancialInstitution.routes.WhereIsFIBasedController.onPageLoad(CheckMode).url
         sut.getAddressChangeRoute(answers) mustBe
           controllers.addFinancialInstitution.routes.WhereIsFIBasedController.onPageLoad(CheckMode).url
       }