From 47b01f592134d7a4e55d6832633436e2d098d4cc Mon Sep 17 00:00:00 2001
From: Shinya Maeda <shinya@gitlab.com>
Date: Fri, 6 Apr 2018 21:29:39 +0900
Subject: [PATCH] Add test for after_destroy :redis_delete_data hook

---
 spec/models/ci/job_trace_chunk_spec.rb | 61 ++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)

diff --git a/spec/models/ci/job_trace_chunk_spec.rb b/spec/models/ci/job_trace_chunk_spec.rb
index cb993885fa1..0daf5ca25c3 100644
--- a/spec/models/ci/job_trace_chunk_spec.rb
+++ b/spec/models/ci/job_trace_chunk_spec.rb
@@ -316,4 +316,65 @@ describe Ci::JobTraceChunk, :clean_gitlab_redis_shared_state do
       end
     end
   end
+
+  describe 'deletes data in redis after chunk record destroyed' do
+    let(:project) { create(:project) }
+
+    before do
+      pipeline = create(:ci_pipeline, project: project)
+      create(:ci_build, :running, :trace_live, pipeline: pipeline, project: project)
+      create(:ci_build, :running, :trace_live, pipeline: pipeline, project: project)
+      create(:ci_build, :running, :trace_live, pipeline: pipeline, project: project)
+    end
+
+    shared_examples_for 'deletes all job_trace_chunk and data in redis' do
+      it do
+        project.builds.each do |build|
+          Gitlab::Redis::SharedState.with do |redis|
+            redis.scan_each(match: "gitlab:ci:trace:#{build.id}:chunks:?") do |key|
+              expect(redis.exists(key)).to be_truthy
+            end
+          end
+        end
+
+        expect(described_class.count).not_to eq(0)
+
+        subject
+
+        expect(described_class.count).to eq(0)
+
+        project.builds.each do |build|
+          Gitlab::Redis::SharedState.with do |redis|
+            redis.scan_each(match: "gitlab:ci:trace:#{build.id}:chunks:?") do |key|
+              expect(redis.exists(key)).to be_falsey
+            end
+          end
+        end
+      end
+    end
+
+    context 'when job_trace_chunk is destroyed' do
+      let(:subject) do
+        project.builds.each { |build| build.chunks.destroy_all }
+      end
+
+      it_behaves_like 'deletes all job_trace_chunk and data in redis'
+    end
+
+    context 'when job is destroyed' do
+      let(:subject) do
+        project.builds.destroy_all
+      end
+
+      it_behaves_like 'deletes all job_trace_chunk and data in redis'
+    end
+
+    context 'when project is destroyed' do
+      let(:subject) do
+        project.destroy!
+      end
+
+      it_behaves_like 'deletes all job_trace_chunk and data in redis'
+    end
+  end
 end