1 files changed, 34 insertions, 0 deletions
diff --git a/spec/lib/banzai/filter/timeout_html_pipeline_filter_spec.rb b/spec/lib/banzai/filter/timeout_html_pipeline_filter_spec.rb
new file mode 100644
index 00000000000..cdb40ef5b04
--- /dev/null
+++ b/spec/lib/banzai/filter/timeout_html_pipeline_filter_spec.rb
@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Banzai::Filter::TimeoutHtmlPipelineFilter do
+  include FilterSpecHelper
+
+  it_behaves_like 'filter timeout' do
+    let(:text) { '<p>some text</p>' }
+  end
+
+  it 'raises NotImplementedError' do
+    expect { filter('test') }.to raise_error NotImplementedError
+  end
+
+  context 'when markup_rendering_timeout is disabled' do
+    it 'waits until the execution completes' do
+      text = '<p>some text</p>'
+
+      stub_feature_flags(markup_rendering_timeout: false)
+      allow_next_instance_of(described_class) do |instance|
+        allow(instance).to receive(:call_with_timeout) do
+          text
+        end
+      end
+
+      expect(Gitlab::RenderTimeout).not_to receive(:timeout)
+
+      result = filter(text)
+
+      expect(result).to eq text
+    end
+  end
+end