diff options
Diffstat (limited to 'spec/support/helpers/filter_spec_helper.rb')
-rw-r--r-- | spec/support/helpers/filter_spec_helper.rb | 87 |
1 files changed, 87 insertions, 0 deletions
diff --git a/spec/support/helpers/filter_spec_helper.rb b/spec/support/helpers/filter_spec_helper.rb new file mode 100644 index 00000000000..721d359c2ee --- /dev/null +++ b/spec/support/helpers/filter_spec_helper.rb @@ -0,0 +1,87 @@ +# Helper methods for Banzai filter specs +# +# Must be included into specs manually +module FilterSpecHelper + extend ActiveSupport::Concern + + # Perform `call` on the described class + # + # Automatically passes the current `project` value, if defined, to the context + # if none is provided. + # + # html - HTML String to pass to the filter's `call` method. + # context - Hash context for the filter. (default: {project: project}) + # + # Returns a Nokogiri::XML::DocumentFragment + def filter(html, context = {}) + if defined?(project) + context.reverse_merge!(project: project) + end + + render_context = Banzai::RenderContext + .new(context[:project], context[:current_user]) + + context = context.merge(render_context: render_context) + + described_class.call(html, context) + end + + # Run text through HTML::Pipeline with the current filter and return the + # result Hash + # + # body - String text to run through the pipeline + # context - Hash context for the filter. (default: {project: project}) + # + # Returns the Hash + def pipeline_result(body, context = {}) + context.reverse_merge!(project: project) if defined?(project) + + pipeline = HTML::Pipeline.new([described_class], context) + pipeline.call(body) + end + + def reference_pipeline(context = {}) + context.reverse_merge!(project: project) if defined?(project) + + filters = [ + Banzai::Filter::AutolinkFilter, + described_class + ] + + HTML::Pipeline.new(filters, context) + end + + def reference_pipeline_result(body, context = {}) + reference_pipeline(context).call(body) + end + + def reference_filter(html, context = {}) + reference_pipeline(context).to_document(html) + end + + # Modify a String reference to make it invalid + # + # Commit SHAs get reversed, IDs get incremented by 1, all other Strings get + # their word characters reversed. + # + # reference - String reference to modify + # + # Returns a String + def invalidate_reference(reference) + if reference =~ /\A(.+)?[^\d]\d+\z/ + # Integer-based reference with optional project prefix + reference.gsub(/\d+\z/) { |i| i.to_i + 10_000 } + elsif reference =~ /\A(.+@)?(\h{7,40}\z)/ + # SHA-based reference with optional prefix + reference.gsub(/\h{7,40}\z/) { |v| v.reverse } + else + reference.gsub(/\w+\z/) { |v| v.reverse } + end + end + + # Shortcut to Rails' auto-generated routes helpers, to avoid including the + # module + def urls + Gitlab::Routing.url_helpers + end +end |