1 files changed, 8 insertions, 2 deletions
diff --git a/spec/services/resource_events/change_milestone_service_spec.rb b/spec/services/resource_events/change_milestone_service_spec.rb
index 9c0f9420f7a..3a9dadbd40e 100644
--- a/spec/services/resource_events/change_milestone_service_spec.rb
+++ b/spec/services/resource_events/change_milestone_service_spec.rb
@@ -3,9 +3,15 @@
 require 'spec_helper'
 
 RSpec.describe ResourceEvents::ChangeMilestoneService do
+  let_it_be(:timebox) { create(:milestone) }
+
+  let(:created_at_time) { Time.utc(2019, 12, 30) }
+  let(:add_timebox_args) { { created_at: created_at_time, old_milestone: nil } }
+  let(:remove_timebox_args) { { created_at: created_at_time, old_milestone: timebox } }
+
   [:issue, :merge_request].each do |issuable|
-    it_behaves_like 'a milestone events creator' do
-      let(:resource) { create(issuable) }
+    it_behaves_like 'timebox(milestone or iteration) resource events creator', ResourceMilestoneEvent do
+      let_it_be(:resource) { create(issuable) }
     end
   end
 end