1 files changed, 18 insertions, 0 deletions
diff --git a/spec/models/ci/build_spec.rb b/spec/models/ci/build_spec.rb
index 697fe3fda06..2a1872dd7d8 100644
--- a/spec/models/ci/build_spec.rb
+++ b/spec/models/ci/build_spec.rb
@@ -776,6 +776,14 @@ describe Ci::Build do
       end
 
       it { is_expected.to be_truthy }
+
+      context 'when ci_enable_legacy_traces feature flag is disabled' do
+        before do
+          stub_feature_flags(ci_enable_legacy_traces: false)
+        end
+
+        it { is_expected.to be_falsy }
+      end
     end
 
     context 'when old trace does not exist' do
@@ -799,6 +807,16 @@ describe Ci::Build do
     it "expect to receive data from database" do
       is_expected.to eq('old trace')
     end
+
+    context 'when ci_enable_legacy_traces feature flag is disabled' do
+      before do
+        stub_feature_flags(ci_enable_legacy_traces: false)
+      end
+
+      it 'does not return old trace' do
+        is_expected.to be_nil
+      end
+    end
   end
 
   describe '#erase_old_trace!' do