diff --git a/app/controllers/content_items_controller.rb b/app/controllers/content_items_controller.rb index 330bfa7b7..136d11237 100644 --- a/app/controllers/content_items_controller.rb +++ b/app/controllers/content_items_controller.rb @@ -225,6 +225,8 @@ def save_this_page_enabled? helper_method :logged_in? def user_has_saved_page? + return false unless @account_session_header + GdsApi.account_api.get_saved_page( page_path: request.path, govuk_account_session: @account_session_header, diff --git a/test/controllers/save_a_page_test.rb b/test/controllers/save_a_page_test.rb index cd2c48cee..280279d42 100644 --- a/test/controllers/save_a_page_test.rb +++ b/test/controllers/save_a_page_test.rb @@ -16,6 +16,13 @@ class ContentItemsControllerTest < ActionController::TestCase end test "logged out - shows the add_page_button when the feature flag is on" do + stub_feature_flag_on + set_up_and_visit_content_item("detailed_guide", "detailed_guide") + + assert has_save_page_button("/guidance/salary-sacrifice-and-the-effects-on-paye") + end + + test "invalid session - shows the add_page_button when the feature flag is on" do stub_feature_flag_on stub_account_api_unauthorized_get_saved_page(page_path: "/guidance/salary-sacrifice-and-the-effects-on-paye") set_up_and_visit_content_item("detailed_guide", "detailed_guide")