1 files changed, 72 insertions, 0 deletions
diff --git a/spec/lib/gitlab/github_import/importer/events/changed_milestone_spec.rb b/spec/lib/gitlab/github_import/importer/events/changed_milestone_spec.rb
new file mode 100644
index 00000000000..2687627fc23
--- /dev/null
+++ b/spec/lib/gitlab/github_import/importer/events/changed_milestone_spec.rb
@@ -0,0 +1,72 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Gitlab::GithubImport::Importer::Events::ChangedMilestone do
+  subject(:importer) { described_class.new(project, client) }
+
+  let_it_be(:project) { create(:project, :repository) }
+  let_it_be(:user) { create(:user) }
+
+  let(:client) { instance_double('Gitlab::GithubImport::Client') }
+  let(:issue) { create(:issue, project: project) }
+  let!(:milestone) { create(:milestone, project: project) }
+
+  let(:issue_event) do
+    Gitlab::GithubImport::Representation::IssueEvent.from_json_hash(
+      'id' => 6501124486,
+      'actor' => { 'id' => user.id, 'login' => user.username },
+      'event' => event_type,
+      'commit_id' => nil,
+      'milestone_title' => milestone.title,
+      'issue_db_id' => issue.id,
+      'created_at' => '2022-04-26 18:30:53 UTC',
+      'issue' => { 'number' => issue.iid }
+    )
+  end
+
+  let(:event_attrs) do
+    {
+      user_id: user.id,
+      issue_id: issue.id,
+      milestone_id: milestone.id,
+      state: 'opened',
+      created_at: issue_event.created_at
+    }.stringify_keys
+  end
+
+  shared_examples 'new event' do
+    it 'creates a new milestone event' do
+      expect { importer.execute(issue_event) }.to change { issue.resource_milestone_events.count }
+        .from(0).to(1)
+      expect(issue.resource_milestone_events.last)
+        .to have_attributes(expected_event_attrs)
+    end
+  end
+
+  describe '#execute' do
+    before do
+      allow(Gitlab::Cache::Import::Caching).to receive(:read_integer).and_return(milestone.id)
+      allow_next_instance_of(Gitlab::GithubImport::IssuableFinder) do |finder|
+        allow(finder).to receive(:database_id).and_return(issue.id)
+      end
+      allow_next_instance_of(Gitlab::GithubImport::UserFinder) do |finder|
+        allow(finder).to receive(:find).with(user.id, user.username).and_return(user.id)
+      end
+    end
+
+    context 'when importing a milestoned event' do
+      let(:event_type) { 'milestoned' }
+      let(:expected_event_attrs) { event_attrs.merge(action: 'add') }
+
+      it_behaves_like 'new event'
+    end
+
+    context 'when importing demilestoned event' do
+      let(:event_type) { 'demilestoned' }
+      let(:expected_event_attrs) { event_attrs.merge(action: 'remove') }
+
+      it_behaves_like 'new event'
+    end
+  end
+end