1 files changed, 32 insertions, 0 deletions
diff --git a/spec/lib/gitlab/github_import/importer/single_endpoint_issue_events_importer_spec.rb b/spec/lib/gitlab/github_import/importer/single_endpoint_issue_events_importer_spec.rb
index 93fcda8a4f0..19142b94519 100644
--- a/spec/lib/gitlab/github_import/importer/single_endpoint_issue_events_importer_spec.rb
+++ b/spec/lib/gitlab/github_import/importer/single_endpoint_issue_events_importer_spec.rb
@@ -208,6 +208,38 @@ RSpec.describe Gitlab::GithubImport::Importer::SingleEndpointIssueEventsImporter
       end
     end
 
+    describe 'increment object counter' do
+      it 'increments counter' do
+        expect(Gitlab::GithubImport::ObjectCounter).to receive(:increment).with(project, :issue_event, :fetched)
+
+        subject.each_object_to_import { |event| event }
+      end
+
+      context 'when event should increment a mapped fetched counter' do
+        before do
+          stub_const('Gitlab::GithubImport::Importer::IssueEventImporter::EVENT_COUNTER_MAP', {
+            'closed' => 'custom_type'
+          })
+        end
+
+        it 'increments the mapped fetched counter' do
+          expect(Gitlab::GithubImport::ObjectCounter).to receive(:increment).with(project, 'custom_type', :fetched)
+
+          subject.each_object_to_import { |event| event }
+        end
+
+        context 'when extended_events is disabled' do
+          let(:extended_events) { false }
+
+          it 'increments the issue_event fetched counter' do
+            expect(Gitlab::GithubImport::ObjectCounter).to receive(:increment).with(project, :issue_event, :fetched)
+
+            subject.each_object_to_import { |event| event }
+          end
+        end
+      end
+    end
+
     describe 'save events' do
       shared_examples 'saves event' do
         it 'saves event' do