From 238c6def85a7253b2eed7fe686705257b43139c5 Mon Sep 17 00:00:00 2001
From: chriswaugh <238270+chriswaugh@users.noreply.github.com>
Date: Mon, 3 Jun 2024 16:11:31 +0100
Subject: [PATCH] Added isFIUser check after FI name page - routing differently
 if so

---
 app/navigation/Navigator.scala      | 7 ++++++-
 test/navigation/NavigatorSpec.scala | 8 +++++++-
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/app/navigation/Navigator.scala b/app/navigation/Navigator.scala
index 5f2901ca..79da37f3 100644
--- a/app/navigation/Navigator.scala
+++ b/app/navigation/Navigator.scala
@@ -31,7 +31,12 @@ class Navigator @Inject() () {
   private val normalRoutes: Page => UserAnswers => Call = {
 
     case NameOfFinancialInstitutionPage =>
-      _ => routes.HaveUniqueTaxpayerReferenceController.onPageLoad(NormalMode)
+      userAnswers =>
+        isFiUser(
+          userAnswers,
+          routes.SendReportsController.onPageLoad(NormalMode),
+          routes.HaveUniqueTaxpayerReferenceController.onPageLoad(NormalMode)
+        )
     case WhatIsUniqueTaxpayerReferencePage =>
       _ => routes.SendReportsController.onPageLoad(NormalMode)
     case SendReportsPage =>
diff --git a/test/navigation/NavigatorSpec.scala b/test/navigation/NavigatorSpec.scala
index cbce20e4..3f9b7d57 100644
--- a/test/navigation/NavigatorSpec.scala
+++ b/test/navigation/NavigatorSpec.scala
@@ -110,7 +110,13 @@ class NavigatorSpec extends SpecBase {
           routes.CheckYourAnswersController.onPageLoad
       }
 
-      "must go from NameOfFinancialInstitutionPage to HaveUniqueTaxpayerReferencePage" in {
+      "must go from NameOfFinancialInstitutionPage to SendReportsPage when user is FI" in {
+        val userAnswers = emptyUserAnswers.withPage(ReportForRegisteredBusinessPage, true)
+        navigator.nextPage(NameOfFinancialInstitutionPage, NormalMode, userAnswers) mustBe
+          routes.SendReportsController.onPageLoad(NormalMode)
+      }
+
+      "must go from NameOfFinancialInstitutionPage to HaveUniqueTaxpayerReferencePage when user is not FI" in {
         val userAnswers = emptyUserAnswers.withPage(NameOfFinancialInstitutionPage, "FI")
         navigator.nextPage(NameOfFinancialInstitutionPage, NormalMode, userAnswers) mustBe
           routes.HaveUniqueTaxpayerReferenceController.onPageLoad(NormalMode)