1 files changed, 18 insertions, 5 deletions
diff --git a/spec/models/clusters/agents/activity_event_spec.rb b/spec/models/clusters/agents/activity_event_spec.rb
index 18b9c82fa6a..2e3833898fd 100644
--- a/spec/models/clusters/agents/activity_event_spec.rb
+++ b/spec/models/clusters/agents/activity_event_spec.rb
@@ -16,11 +16,10 @@ RSpec.describe Clusters::Agents::ActivityEvent do
     let_it_be(:agent) { create(:cluster_agent) }
 
     describe '.in_timeline_order' do
-      let(:recorded_at) { 1.hour.ago }
-
-      let!(:event1) { create(:agent_activity_event, agent: agent, recorded_at: recorded_at) }
-      let!(:event2) { create(:agent_activity_event, agent: agent, recorded_at: Time.current) }
-      let!(:event3) { create(:agent_activity_event, agent: agent, recorded_at: recorded_at) }
+      let_it_be(:recorded_at) { 1.hour.ago }
+      let_it_be(:event1) { create(:agent_activity_event, agent: agent, recorded_at: recorded_at) }
+      let_it_be(:event2) { create(:agent_activity_event, agent: agent, recorded_at: Time.current) }
+      let_it_be(:event3) { create(:agent_activity_event, agent: agent, recorded_at: recorded_at) }
 
       subject { described_class.in_timeline_order }
 
@@ -28,5 +27,19 @@ RSpec.describe Clusters::Agents::ActivityEvent do
         is_expected.to eq([event2, event3, event1])
       end
     end
+
+    describe '.recorded_before' do
+      let_it_be(:event1) { create(:agent_activity_event, agent: agent, recorded_at: 1.hour.ago) }
+      let_it_be(:event2) { create(:agent_activity_event, agent: agent, recorded_at: 2.hours.ago) }
+      let_it_be(:event3) { create(:agent_activity_event, agent: agent, recorded_at: 3.hours.ago) }
+
+      let(:cutoff) { event2.recorded_at }
+
+      subject { described_class.recorded_before(cutoff) }
+
+      it 'returns only events recorded before the cutoff' do
+        is_expected.to contain_exactly(event3)
+      end
+    end
   end
 end