diff options
Diffstat (limited to 'spec/workers/ci')
-rw-r--r-- | spec/workers/ci/archive_trace_worker_spec.rb | 28 | ||||
-rw-r--r-- | spec/workers/ci/resource_groups/assign_resource_from_resource_group_worker_spec.rb | 2 |
2 files changed, 29 insertions, 1 deletions
diff --git a/spec/workers/ci/archive_trace_worker_spec.rb b/spec/workers/ci/archive_trace_worker_spec.rb index 889e0c92042..52723ff5823 100644 --- a/spec/workers/ci/archive_trace_worker_spec.rb +++ b/spec/workers/ci/archive_trace_worker_spec.rb @@ -16,6 +16,34 @@ RSpec.describe Ci::ArchiveTraceWorker do subject end + + it 'has preloaded the arguments for archiving' do + allow_next_instance_of(Ci::ArchiveTraceService) do |instance| + allow(instance).to receive(:execute) do |job| + expect(job.association(:project)).to be_loaded + expect(job.association(:pending_state)).to be_loaded + end + end + + subject + end + + context 'when sticky_ci_archive_trace_worker is disabled' do + before do + stub_feature_flags(sticky_ci_archive_trace_worker: false) + end + + it 'does not preload associations' do + allow_next_instance_of(Ci::ArchiveTraceService) do |instance| + allow(instance).to receive(:execute) do |job| + expect(job.association(:project)).not_to be_loaded + expect(job.association(:pending_state)).not_to be_loaded + end + end + + subject + end + end end context 'when job is not found' do diff --git a/spec/workers/ci/resource_groups/assign_resource_from_resource_group_worker_spec.rb b/spec/workers/ci/resource_groups/assign_resource_from_resource_group_worker_spec.rb index be7f7ef5c8c..785cba24f9d 100644 --- a/spec/workers/ci/resource_groups/assign_resource_from_resource_group_worker_spec.rb +++ b/spec/workers/ci/resource_groups/assign_resource_from_resource_group_worker_spec.rb @@ -25,7 +25,7 @@ RSpec.describe Ci::ResourceGroups::AssignResourceFromResourceGroupWorker do context 'when resource group exists' do it 'executes AssignResourceFromResourceGroupService' do - expect_next_instances_of(Ci::ResourceGroups::AssignResourceFromResourceGroupService, 2, resource_group.project, nil) do |service| + expect_next_instances_of(Ci::ResourceGroups::AssignResourceFromResourceGroupService, 2, false, resource_group.project, nil) do |service| expect(service).to receive(:execute).with(resource_group) end |