Updated test for notice-page to reflect new logic

cabinetoffice · Jul 17, 2023 · b4c9bc8 · b4c9bc8
1 parent 74f7ef7
commit b4c9bc8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/test/java/gov/cabinetofice/gapuserservice/web/LoginControllerV2Test.java b/src/test/java/gov/cabinetofice/gapuserservice/web/LoginControllerV2Test.java
@@ -22,6 +22,7 @@
 import org.springframework.web.servlet.view.RedirectView;
 import org.springframework.web.util.WebUtils;
 
+import java.sql.Ref;
 import java.util.List;
 import java.util.Optional;
 
@@ -119,9 +120,12 @@ void shouldReturnDefaultRedirectUrl_IfRedirectUrlNotProvided_AndTokenIsValid() {
 
         @Test
         void showNoticePage_ShowsNoticePage_WithLoginUrl() {
+            when(oneLoginService.getOneLoginAuthorizeUrl())
+                    .thenReturn("loginUrl");
+
             final ModelAndView methodResponse = loginController.showNoticePage();
             assertThat(methodResponse.getViewName()).isEqualTo(LoginControllerV2.NOTICE_PAGE_VIEW);
-            assertThat(methodResponse.getModel().get("loginUrl")).isEqualTo("oneLoginBaseUrl");
+            assertThat(methodResponse.getModel().get("loginUrl")).isEqualTo("loginUrl");
 
         }
     }