1 files changed, 39 insertions, 0 deletions
diff --git a/spec/lib/gitlab/memory/watchdog/monitor/rss_memory_limit_spec.rb b/spec/lib/gitlab/memory/watchdog/monitor/rss_memory_limit_spec.rb
new file mode 100644
index 00000000000..9e25cfda782
--- /dev/null
+++ b/spec/lib/gitlab/memory/watchdog/monitor/rss_memory_limit_spec.rb
@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+require 'fast_spec_helper'
+require 'support/shared_examples/lib/gitlab/memory/watchdog/monitor_result_shared_examples'
+
+RSpec.describe Gitlab::Memory::Watchdog::Monitor::RssMemoryLimit do
+  let(:memory_limit) { 2048 }
+  let(:worker_memory) { 1024 }
+
+  subject(:monitor) do
+    described_class.new(memory_limit: memory_limit)
+  end
+
+  before do
+    allow(Gitlab::Metrics::System).to receive(:memory_usage_rss).and_return({ total: worker_memory })
+  end
+
+  describe '#call' do
+    context 'when process exceeds threshold' do
+      let(:worker_memory) { memory_limit + 1 }
+      let(:payload) do
+        {
+          message: 'rss memory limit exceeded',
+          memwd_rss_bytes: worker_memory,
+          memwd_max_rss_bytes: memory_limit
+        }
+      end
+
+      include_examples 'returns Watchdog Monitor result', threshold_violated: true
+    end
+
+    context 'when process does not exceed threshold' do
+      let(:worker_memory) { memory_limit - 1 }
+      let(:payload) { {} }
+
+      include_examples 'returns Watchdog Monitor result', threshold_violated: false
+    end
+  end
+end