@@ -60,7 +60,8 @@ class Navigator @Inject() () {
6060 _ => controllers.add.company.routes.CompanyAddressYesNoController .onPageLoad(NormalMode )
6161 case CompanyContactOptionsPage =>
6262 userAnswers => navigatorFromCompanyContactOptionsPage(NormalMode )(userAnswers)
63- case CompanyEmailAddressPage => _ => controllers.add.company.routes.CompanyEmailAddressController .onPageLoad(NormalMode )
63+ case CompanyEmailAddressPage =>
64+ _ => controllers.add.company.routes.CompanyEmailAddressController .onPageLoad(NormalMode )
6465 case PartnershipNamePage =>
6566 _ => controllers.add.partnership.routes.PartnershipHasUtrYesNoController .onPageLoad(NormalMode )
6667 case PartnershipHasUtrYesNoPage => userAnswers => navigatorFromPartnershipHasUtrYesNoPage(NormalMode )(userAnswers)
@@ -94,6 +95,8 @@ class Navigator @Inject() () {
9495 userAnswers => navigatorFromPartnershipNominatedPartnerUtrYesNoPage(NormalMode )(userAnswers)
9596 case CompanyAddressPage =>
9697 _ => controllers.add.company.routes.CompanyAddressController .onPageLoad(NormalMode )
98+ case CompanyUtrYesNoPage =>
99+ userAnswers => navigatorFromCompanyUtrYesNoPage(NormalMode )(userAnswers)
97100 case PartnershipMobileNumberPage =>
98101 _ => controllers.add.partnership.routes.PartnershipMobileNumberController .onPageLoad(NormalMode )
99102 case PartnershipPhoneNumberPage =>
@@ -129,13 +132,15 @@ class Navigator @Inject() () {
129132 userAnswers => navigatorFromPartnershipNominatedPartnerUtrYesNoPage(CheckMode )(userAnswers)
130133 case CompanyAddressPage =>
131134 _ => controllers.add.company.routes.CompanyAddressController .onPageLoad(CheckMode )
132- case CompanyEmailAddressPage =>
135+ case CompanyEmailAddressPage =>
133136 _ => controllers.add.company.routes.CompanyEmailAddressController .onPageLoad(CheckMode )
134137 case PartnershipMobileNumberPage =>
135138 _ => controllers.add.partnership.routes.PartnershipMobileNumberController .onPageLoad(CheckMode )
136139 case PartnershipPhoneNumberPage => _ => controllers.add.routes.CheckYourAnswersController .onPageLoad()
137140 case CompanyAddressYesNoPage =>
138141 _ => controllers.add.company.routes.CompanyAddressYesNoController .onPageLoad(CheckMode )
142+ case CompanyUtrYesNoPage =>
143+ userAnswers => navigatorFromCompanyUtrYesNoPage(CheckMode )(userAnswers)
139144 case CompanyPhoneNumberPage =>
140145 _ => controllers.add.company.routes.CompanyPhoneNumberController .onPageLoad(CheckMode )
141146 case _ => _ => controllers.add.routes.CheckYourAnswersController .onPageLoad()
@@ -325,6 +330,15 @@ class Navigator @Inject() () {
325330 case (Some (false ), CheckMode ) => controllers.add.routes.CheckYourAnswersController .onPageLoad()
326331 case (None , _) => routes.JourneyRecoveryController .onPageLoad()
327332 }
333+ private def navigatorFromCompanyUtrYesNoPage (mode : Mode )(userAnswers : UserAnswers ): Call =
334+ (userAnswers.get(CompanyUtrYesNoPage ), mode) match {
335+ case (Some (true ), _) =>
336+ controllers.add.company.routes.CompanyUtrController .onPageLoad(mode)
337+ case (Some (false ), NormalMode ) =>
338+ controllers.add.company.routes.CompanyCrnYesNoController .onPageLoad(NormalMode )
339+ case (Some (false ), CheckMode ) => controllers.add.routes.CheckYourAnswersController .onPageLoad()
340+ case (None , _) => routes.JourneyRecoveryController .onPageLoad()
341+ }
328342
329343 private def navigatorFromPartnershipNominatedPartnerCrnYesNoPage (mode : Mode )(userAnswers : UserAnswers ): Call =
330344 (userAnswers.get(PartnershipNominatedPartnerCrnYesNoPage ), mode) match {
0 commit comments