1 files changed, 17 insertions, 0 deletions
diff --git a/spec/workers/concerns/worker_attributes_spec.rb b/spec/workers/concerns/worker_attributes_spec.rb
index ac9d3fa824e..959cb62c6fb 100644
--- a/spec/workers/concerns/worker_attributes_spec.rb
+++ b/spec/workers/concerns/worker_attributes_spec.rb
@@ -37,6 +37,7 @@ RSpec.describe WorkerAttributes, feature_category: :shared do
       :worker_has_external_dependencies? | :worker_has_external_dependencies! | false            | []                                 | true
       :idempotent?                       | :idempotent!                       | false            | []                                 | true
       :big_payload?                      | :big_payload!                      | false            | []                                 | true
+      :database_health_check_attrs       | :defer_on_database_health_signal   | nil              | [:gitlab_main, 1.minute, [:users]] | { gitlab_schema: :gitlab_main, delay_by: 1.minute, tables: [:users] }
     end
     # rubocop: enable Layout/LineLength
 
@@ -141,4 +142,20 @@ RSpec.describe WorkerAttributes, feature_category: :shared do
       end
     end
   end
+
+  describe '#defer_on_database_health_signal?' do
+    subject(:defer_on_database_health_signal?) { worker.defer_on_database_health_signal? }
+
+    context 'when defer_on_database_health_signal is set' do
+      before do
+        worker.defer_on_database_health_signal(:gitlab_main, 1.minute, [:users])
+      end
+
+      it { is_expected.to be(true) }
+    end
+
+    context 'when defer_on_database_health_signal is not set' do
+      it { is_expected.to be(false) }
+    end
+  end
 end