diff options
author | GitLab Bot <gitlab-bot@gitlab.com> | 2023-06-06 09:08:48 +0300 |
---|---|---|
committer | GitLab Bot <gitlab-bot@gitlab.com> | 2023-06-06 09:08:48 +0300 |
commit | c77b780ee080b978bd5a63f642f741e8892383dc (patch) | |
tree | 7c50ab6450a61d27485f9557b666af46ec5adfdd /spec/workers | |
parent | 5e448ff06309854c838fb5eaa46fd05ebc5218ab (diff) |
Add latest changes from gitlab-org/gitlab@master
Diffstat (limited to 'spec/workers')
-rw-r--r-- | spec/workers/gitlab/github_import/stage/import_pull_requests_worker_spec.rb | 102 |
1 files changed, 87 insertions, 15 deletions
diff --git a/spec/workers/gitlab/github_import/stage/import_pull_requests_worker_spec.rb b/spec/workers/gitlab/github_import/stage/import_pull_requests_worker_spec.rb index 6ebf93730eb..9b2cfead684 100644 --- a/spec/workers/gitlab/github_import/stage/import_pull_requests_worker_spec.rb +++ b/spec/workers/gitlab/github_import/stage/import_pull_requests_worker_spec.rb @@ -5,38 +5,110 @@ require 'spec_helper' RSpec.describe Gitlab::GithubImport::Stage::ImportPullRequestsWorker, feature_category: :importers do let_it_be(:project) { create(:project) } let_it_be(:import_state) { create(:import_state, project: project) } + let(:options) { { state: 'all', sort: 'number', direction: 'desc', per_page: '1' } } let(:worker) { described_class.new } let(:importer) { double(:importer) } let(:client) { double(:client) } describe '#import' do - it 'imports all the pull requests' do - waiter = Gitlab::JobWaiter.new(2, '123') + context 'with pull requests' do + it 'imports all the pull requests and allocates internal iids' do + waiter = Gitlab::JobWaiter.new(2, '123') - expect(Gitlab::GithubImport::Importer::PullRequestsImporter) - .to receive(:new) - .with(project, client) - .and_return(importer) + expect(Gitlab::GithubImport::Importer::PullRequestsImporter) + .to receive(:new) + .with(project, client) + .and_return(importer) - expect(importer) - .to receive(:execute) - .and_return(waiter) + expect(importer) + .to receive(:execute) + .and_return(waiter) - expect(import_state) - .to receive(:refresh_jid_expiration) + expect(import_state) + .to receive(:refresh_jid_expiration) - expect(Gitlab::GithubImport::AdvanceStageWorker) - .to receive(:perform_async) - .with(project.id, { '123' => 2 }, :collaborators) + expect(InternalId).to receive(:exists?).and_return(false) - worker.import(client, project) + expect(client).to receive(:each_object).with( + :pulls, project.import_source, options + ).and_return([{ number: 4 }].each) + + expect(Gitlab::GithubImport::AdvanceStageWorker) + .to receive(:perform_async) + .with(project.id, { '123' => 2 }, :collaborators) + + expect(MergeRequest).to receive(:track_target_project_iid!) + + worker.import(client, project) + end + end + + context 'without pull requests' do + it 'does not allocate internal iids' do + waiter = Gitlab::JobWaiter.new(2, '123') + + expect(Gitlab::GithubImport::Importer::PullRequestsImporter) + .to receive(:new) + .with(project, client) + .and_return(importer) + + expect(importer) + .to receive(:execute) + .and_return(waiter) + + expect(import_state) + .to receive(:refresh_jid_expiration) + + expect(InternalId).to receive(:exists?).and_return(false) + + expect(client).to receive(:each_object).with( + :pulls, project.import_source, options + ).and_return([nil].each) + + expect(Gitlab::GithubImport::AdvanceStageWorker) + .to receive(:perform_async) + .with(project.id, { '123' => 2 }, :collaborators) + + expect(MergeRequest).not_to receive(:track_target_project_iid!) + + worker.import(client, project) + end + end + + context 'when retrying' do + it 'does not allocate internal iids' do + waiter = Gitlab::JobWaiter.new(2, '123') + + expect(Gitlab::GithubImport::Importer::PullRequestsImporter) + .to receive(:new) + .with(project, client) + .and_return(importer) + + expect(importer) + .to receive(:execute) + .and_return(waiter) + + expect(import_state) + .to receive(:refresh_jid_expiration) + + expect(InternalId).to receive(:exists?).and_return(true) + + expect(client).not_to receive(:each_object) + expect(MergeRequest).not_to receive(:track_target_project_iid!) + + worker.import(client, project) + end end end it 'raises an error' do exception = StandardError.new('_some_error_') + expect(client).to receive(:each_object).with( + :pulls, project.import_source, options + ).and_return([{ number: 4 }].each) + expect_next_instance_of(Gitlab::GithubImport::Importer::PullRequestsImporter) do |importer| expect(importer).to receive(:execute).and_raise(exception) end |