summaryrefslogtreecommitdiff
path: root/spec/workers
diff options
context:
space:
mode:
authorGitLab Bot <gitlab-bot@gitlab.com>2020-03-24 09:09:25 +0000
committerGitLab Bot <gitlab-bot@gitlab.com>2020-03-24 09:09:25 +0000
commit6f7881ee9dcec34141a8f34fc814b56b366d2b48 (patch)
tree25f72a06874b32b1049b79a9d7f4f1b7bca43b9b /spec/workers
parent8c8bf44fa64f98114f7439f751c92d59a44b3218 (diff)
downloadgitlab-ce-6f7881ee9dcec34141a8f34fc814b56b366d2b48.tar.gz
Add latest changes from gitlab-org/gitlab@master
Diffstat (limited to 'spec/workers')
-rw-r--r--spec/workers/cluster_update_app_worker_spec.rb98
-rw-r--r--spec/workers/cluster_wait_for_app_update_worker_spec.rb27
2 files changed, 125 insertions, 0 deletions
diff --git a/spec/workers/cluster_update_app_worker_spec.rb b/spec/workers/cluster_update_app_worker_spec.rb
new file mode 100644
index 00000000000..e540ede4bc0
--- /dev/null
+++ b/spec/workers/cluster_update_app_worker_spec.rb
@@ -0,0 +1,98 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+describe ClusterUpdateAppWorker do
+ include ExclusiveLeaseHelpers
+
+ let_it_be(:project) { create(:project) }
+
+ let(:prometheus_update_service) { spy }
+
+ subject { described_class.new }
+
+ around do |example|
+ Timecop.freeze(Time.now) { example.run }
+ end
+
+ before do
+ allow(::Clusters::Applications::PrometheusUpdateService).to receive(:new).and_return(prometheus_update_service)
+ end
+
+ describe '#perform' do
+ context 'when the application last_update_started_at is higher than the time the job was scheduled in' do
+ it 'does nothing' do
+ application = create(:clusters_applications_prometheus, :updated, last_update_started_at: Time.now)
+
+ expect(prometheus_update_service).not_to receive(:execute)
+
+ expect(subject.perform(application.name, application.id, project.id, Time.now - 5.minutes)).to be_nil
+ end
+ end
+
+ context 'when another worker is already running' do
+ it 'returns nil' do
+ application = create(:clusters_applications_prometheus, :updating)
+
+ expect(subject.perform(application.name, application.id, project.id, Time.now)).to be_nil
+ end
+ end
+
+ it 'executes PrometheusUpdateService' do
+ application = create(:clusters_applications_prometheus, :installed)
+
+ expect(prometheus_update_service).to receive(:execute)
+
+ subject.perform(application.name, application.id, project.id, Time.now)
+ end
+
+ context 'with exclusive lease' do
+ let(:application) { create(:clusters_applications_prometheus, :installed) }
+ let(:lease_key) { "#{described_class.name.underscore}-#{application.id}" }
+
+ before do
+ allow(Gitlab::ExclusiveLease).to receive(:new)
+ stub_exclusive_lease_taken(lease_key)
+ end
+
+ it 'does not allow same app to be updated concurrently by same project' do
+ expect(Clusters::Applications::PrometheusUpdateService).not_to receive(:new)
+
+ subject.perform(application.name, application.id, project.id, Time.now)
+ end
+
+ it 'does not allow same app to be updated concurrently by different project' do
+ project1 = create(:project)
+
+ expect(Clusters::Applications::PrometheusUpdateService).not_to receive(:new)
+
+ subject.perform(application.name, application.id, project1.id, Time.now)
+ end
+
+ it 'allows different app to be updated concurrently by same project' do
+ application2 = create(:clusters_applications_prometheus, :installed)
+ lease_key2 = "#{described_class.name.underscore}-#{application2.id}"
+
+ stub_exclusive_lease(lease_key2)
+
+ expect(Clusters::Applications::PrometheusUpdateService).to receive(:new)
+ .with(application2, project)
+
+ subject.perform(application2.name, application2.id, project.id, Time.now)
+ end
+
+ it 'allows different app to be updated by different project' do
+ application2 = create(:clusters_applications_prometheus, :installed)
+ lease_key2 = "#{described_class.name.underscore}-#{application2.id}"
+ project2 = create(:project)
+
+ stub_exclusive_lease(lease_key2)
+
+ expect(Clusters::Applications::PrometheusUpdateService).to receive(:new)
+ .with(application2, project2)
+
+ subject.perform(application2.name, application2.id, project2.id, Time.now)
+ end
+ end
+ end
+end
diff --git a/spec/workers/cluster_wait_for_app_update_worker_spec.rb b/spec/workers/cluster_wait_for_app_update_worker_spec.rb
new file mode 100644
index 00000000000..f1206bd85cb
--- /dev/null
+++ b/spec/workers/cluster_wait_for_app_update_worker_spec.rb
@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+describe ClusterWaitForAppUpdateWorker do
+ let(:check_upgrade_progress_service) { spy }
+
+ before do
+ allow(::Clusters::Applications::CheckUpgradeProgressService).to receive(:new).and_return(check_upgrade_progress_service)
+ end
+
+ it 'runs CheckUpgradeProgressService when application is found' do
+ application = create(:clusters_applications_prometheus)
+
+ expect(check_upgrade_progress_service).to receive(:execute)
+
+ subject.perform(application.name, application.id)
+ end
+
+ it 'does not run CheckUpgradeProgressService when application is not found' do
+ expect(check_upgrade_progress_service).not_to receive(:execute)
+
+ expect do
+ subject.perform("prometheus", -1)
+ end.to raise_error(ActiveRecord::RecordNotFound)
+ end
+end