From 1c60ff0b7ae190a5c6c1cc8c72358af6ef66c05e Mon Sep 17 00:00:00 2001
From: Kamil Trzcinski <ayufan@ayufan.eu>
Date: Fri, 10 Jun 2016 17:27:49 +0200
Subject: [PATCH] Test ExpireBuildArtifactsWorker

---
 app/workers/expire_build_artifacts_worker.rb  |  2 +-
 .../expire_build_artifacts_worker_spec.rb     | 55 +++++++++++++++++++
 2 files changed, 56 insertions(+), 1 deletion(-)
 create mode 100644 spec/workers/expire_build_artifacts_worker_spec.rb

diff --git a/app/workers/expire_build_artifacts_worker.rb b/app/workers/expire_build_artifacts_worker.rb
index 17b3b5f227f..c64ea108d52 100644
--- a/app/workers/expire_build_artifacts_worker.rb
+++ b/app/workers/expire_build_artifacts_worker.rb
@@ -1,4 +1,4 @@
-class ExpireBuildArtifacts
+class ExpireBuildArtifactsWorker
   include Sidekiq::Worker
 
   def perform
diff --git a/spec/workers/expire_build_artifacts_worker_spec.rb b/spec/workers/expire_build_artifacts_worker_spec.rb
new file mode 100644
index 00000000000..c9ccddc2a09
--- /dev/null
+++ b/spec/workers/expire_build_artifacts_worker_spec.rb
@@ -0,0 +1,55 @@
+require 'spec_helper'
+
+describe ExpireBuildArtifactsWorker do
+  include RepoHelpers
+
+  let(:worker) { ExpireBuildArtifactsWorker.new }
+
+  describe '#perform' do
+    context 'with expired artifacts' do
+      let!(:build) { create(:ci_build, :artifacts, artifacts_expire_at: Time.now - 7.days) }
+
+      it do
+        expect_any_instance_of(Ci::Build).to receive(:erase_artifacts!)
+        worker.perform
+        build.reload
+        expect(build.artifacts_expired?).to be_truthy
+      end
+    end
+
+    context 'with not yet expired artifacts' do
+      let!(:build) { create(:ci_build, :artifacts, artifacts_expire_at: Time.now + 7.days) }
+
+      it do
+        expect_any_instance_of(Ci::Build).not_to receive(:erase_artifacts!)
+        worker.perform
+        build.reload
+        expect(build.artifacts_expired?).to be_falsey
+      end
+    end
+
+    context 'without expire date' do
+      let!(:build) { create(:ci_build, :artifacts) }
+
+      it do
+        expect_any_instance_of(Ci::Build).not_to receive(:erase_artifacts!)
+        worker.perform
+      end
+    end
+
+    context 'for expired artifacts' do
+      let!(:build) { create(:ci_build, :artifacts, artifacts_expire_at: Time.now - 7.days) }
+
+      before do
+        build.erase_artifacts!
+      end
+
+      it do
+        expect_any_instance_of(Ci::Build).not_to receive(:erase_artifacts!)
+        worker.perform
+        build.reload
+        expect(build.artifacts_expired?).to be_truthy
+      end
+    end
+  end
+end