1 files changed, 93 insertions, 0 deletions
diff --git a/spec/features/projects_spec.rb b/spec/features/projects_spec.rb
index db64f84aa76..f6f9c7f0d3c 100644
--- a/spec/features/projects_spec.rb
+++ b/spec/features/projects_spec.rb
@@ -440,6 +440,99 @@ RSpec.describe 'Project' do
     end
   end
 
+  describe 'storage_enforcement_banner', :js do
+    let_it_be(:group) { create(:group) }
+    let_it_be_with_refind(:user) { create(:user) }
+    let_it_be(:project) { create(:project, group: group) }
+
+    before do
+      group.add_maintainer(user)
+      sign_in(user)
+    end
+
+    context 'with storage_enforcement_date set' do
+      let_it_be(:storage_enforcement_date) { Date.today + 30 }
+
+      before do
+        allow_next_found_instance_of(Group) do |grp|
+          allow(grp).to receive(:storage_enforcement_date).and_return(storage_enforcement_date)
+        end
+      end
+
+      it 'displays the banner in the project page' do
+        visit project_path(project)
+        expect_page_to_have_storage_enforcement_banner(storage_enforcement_date)
+      end
+
+      context 'when in a subgroup project page' do
+        let_it_be(:subgroup) { create(:group, parent: group) }
+        let_it_be(:project) { create(:project, namespace: subgroup) }
+
+        it 'displays the banner' do
+          visit project_path(project)
+          expect_page_to_have_storage_enforcement_banner(storage_enforcement_date)
+        end
+      end
+
+      context 'when in a user namespace project page' do
+        let_it_be(:project) { create(:project, namespace: user.namespace) }
+
+        before do
+          allow_next_found_instance_of(Namespaces::UserNamespace) do |namspace|
+            allow(namspace).to receive(:storage_enforcement_date).and_return(storage_enforcement_date)
+          end
+        end
+
+        it 'displays the banner' do
+          visit project_path(project)
+          expect_page_to_have_storage_enforcement_banner(storage_enforcement_date)
+        end
+      end
+
+      it 'does not display the banner in a paid group project page' do
+        allow_next_found_instance_of(Group) do |grp|
+          allow(grp).to receive(:paid?).and_return(true)
+        end
+        visit project_path(project)
+        expect_page_not_to_have_storage_enforcement_banner
+      end
+
+      it 'does not display the banner if user has previously closed unless threshold has changed' do
+        visit project_path(project)
+        expect_page_to_have_storage_enforcement_banner(storage_enforcement_date)
+        find('.js-storage-enforcement-banner [data-testid="close-icon"]').click
+        wait_for_requests
+        page.refresh
+        expect_page_not_to_have_storage_enforcement_banner
+
+        storage_enforcement_date = Date.today + 13
+        allow_next_found_instance_of(Group) do |grp|
+          allow(grp).to receive(:storage_enforcement_date).and_return(storage_enforcement_date)
+        end
+        page.refresh
+        expect_page_to_have_storage_enforcement_banner(storage_enforcement_date)
+      end
+    end
+
+    context 'with storage_enforcement_date not set' do
+      # This test should break and be rewritten after the implementation of the storage_enforcement_date
+      # TBD: https://gitlab.com/gitlab-org/gitlab/-/issues/350632
+      it 'does not display the banner in the group page' do
+        stub_feature_flags(namespace_storage_limit_bypass_date_check: false)
+        visit project_path(project)
+        expect_page_not_to_have_storage_enforcement_banner
+      end
+    end
+  end
+
+  def expect_page_to_have_storage_enforcement_banner(storage_enforcement_date)
+    expect(page).to have_text "From #{storage_enforcement_date} storage limits will apply to this namespace"
+  end
+
+  def expect_page_not_to_have_storage_enforcement_banner
+    expect(page).not_to have_text "storage limits will apply to this namespace"
+  end
+
   def remove_with_confirm(button_text, confirm_with, confirm_button_text = 'Confirm')
     click_button button_text
     fill_in 'confirm_name_input', with: confirm_with