Commit 562f357b authored by Lin Jen-Shin's avatar Lin Jen-Shin

Implement `expect_next_instance_of` and use it

We need this because `expect_any_instance_of` doesn't work on
prepended models. Now we could use the same code between CE/EE
parent 7c11ed8c
...@@ -25,7 +25,9 @@ describe Gitlab::BitbucketImport::ProjectCreator do ...@@ -25,7 +25,9 @@ describe Gitlab::BitbucketImport::ProjectCreator do
end end
it 'creates project' do it 'creates project' do
allow_any_instance_of(Project).to receive(:add_import_job) expect_next_instance_of(Project) do |project|
expect(project).to receive(:add_import_job)
end
project_creator = described_class.new(repo, 'vim', namespace, user, access_params) project_creator = described_class.new(repo, 'vim', namespace, user, access_params)
project = project_creator.execute project = project_creator.execute
......
...@@ -21,7 +21,9 @@ describe Gitlab::GitlabImport::ProjectCreator do ...@@ -21,7 +21,9 @@ describe Gitlab::GitlabImport::ProjectCreator do
end end
it 'creates project' do it 'creates project' do
allow_any_instance_of(Project).to receive(:add_import_job) expect_next_instance_of(Project) do |project|
expect(project).to receive(:add_import_job)
end
project_creator = described_class.new(repo, namespace, user, access_params) project_creator = described_class.new(repo, namespace, user, access_params)
project = project_creator.execute project = project_creator.execute
......
...@@ -16,7 +16,9 @@ describe Gitlab::GoogleCodeImport::ProjectCreator do ...@@ -16,7 +16,9 @@ describe Gitlab::GoogleCodeImport::ProjectCreator do
end end
it 'creates project' do it 'creates project' do
allow_any_instance_of(Project).to receive(:add_import_job) expect_next_instance_of(Project) do |project|
expect(project).to receive(:add_import_job)
end
project_creator = described_class.new(repo, namespace, user) project_creator = described_class.new(repo, namespace, user)
project = project_creator.execute project = project_creator.execute
......
...@@ -17,7 +17,10 @@ describe Gitlab::LegacyGithubImport::ProjectCreator do ...@@ -17,7 +17,10 @@ describe Gitlab::LegacyGithubImport::ProjectCreator do
before do before do
namespace.add_owner(user) namespace.add_owner(user)
allow_any_instance_of(Project).to receive(:add_import_job)
expect_next_instance_of(Project) do |project|
expect(project).to receive(:add_import_job)
end
end end
describe '#execute' do describe '#execute' do
......
...@@ -69,6 +69,7 @@ RSpec.configure do |config| ...@@ -69,6 +69,7 @@ RSpec.configure do |config|
config.include StubFeatureFlags config.include StubFeatureFlags
config.include StubGitlabCalls config.include StubGitlabCalls
config.include StubGitlabData config.include StubGitlabData
config.include ExpectNextInstanceOf
config.include TestEnv config.include TestEnv
config.include Devise::Test::ControllerHelpers, type: :controller config.include Devise::Test::ControllerHelpers, type: :controller
config.include Devise::Test::IntegrationHelpers, type: :feature config.include Devise::Test::IntegrationHelpers, type: :feature
......
module ExpectNextInstanceOf
def expect_next_instance_of(klass, *new_args)
receive_new = receive(:new)
receive_new.with(*new_args) if new_args.any?
expect(klass).to receive_new
.and_wrap_original do |method, *original_args|
method.call(*original_args).tap do |instance|
yield(instance)
end
end
end
end
...@@ -18,12 +18,8 @@ describe UpdateMergeRequestsWorker do ...@@ -18,12 +18,8 @@ describe UpdateMergeRequestsWorker do
end end
it 'executes MergeRequests::RefreshService with expected values' do it 'executes MergeRequests::RefreshService with expected values' do
expect(MergeRequests::RefreshService).to receive(:new) expect_next_instance_of(MergeRequests::RefreshService, project, user) do |refresh_service|
.with(project, user).and_wrap_original do |method, *args| expect(refresh_service).to receive(:execute).with(oldrev, newrev, ref)
method.call(*args).tap do |refresh_service|
expect(refresh_service)
.to receive(:execute).with(oldrev, newrev, ref)
end
end end
perform perform
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment