Relocate PlantUML config and disable SVG support

Merge branch 'security-416902-config-16-0' into '16-0-stable-ee'

See merge request gitlab-org/security/gitlab!3440

Changelog: security

lib/banzai/filter/plantuml_filter.rb

@@ -11,7 +11,7 @@ class PlantumlFilter < HTML::Pipeline::Filter
       def call
         return doc unless settings.plantuml_enabled? && doc.at_xpath(lang_tag)
-        plantuml_setup
+        Gitlab::Plantuml.configure
         doc.xpath(lang_tag).each do |node|
           img_tag = Nokogiri::HTML::DocumentFragment.parse(
@@ -38,15 +38,6 @@ def lang_tag
       def settings
         Gitlab::CurrentSettings.current_application_settings
       end
-
-      def plantuml_setup
-        Asciidoctor::PlantUml.configure do |conf|
-          conf.url = settings.plantuml_url
-          conf.png_enable = settings.plantuml_enabled
-          conf.svg_enable = false
-          conf.txt_enable = false
-        end
-      end
     end
   end
 end

lib/gitlab/asciidoc.rb

@@ -77,20 +77,11 @@ def self.render(input, context)
       context[:pipeline] = :ascii_doc
       context[:max_includes] = [MAX_INCLUDES, context[:max_includes]].compact.min
-      plantuml_setup
+      Gitlab::Plantuml.configure
       html = ::Asciidoctor.convert(input, asciidoc_opts)
       html = Banzai.render(html, context)
       html.html_safe
     end
-
-    def self.plantuml_setup
-      Asciidoctor::PlantUml.configure do |conf|
-        conf.url = Gitlab::CurrentSettings.plantuml_url
-        conf.svg_enable = Gitlab::CurrentSettings.plantuml_enabled
-        conf.png_enable = Gitlab::CurrentSettings.plantuml_enabled
-        conf.txt_enable = false
-      end
-    end
   end
 end

lib/gitlab/plantuml.rb

+# frozen_string_literal: true
+
+require "asciidoctor_plantuml/plantuml"
+
+module Gitlab
+  module Plantuml
+    class << self
+      def configure
+        Asciidoctor::PlantUml.configure do |conf|
+          conf.url        = Gitlab::CurrentSettings.plantuml_url
+          conf.png_enable = Gitlab::CurrentSettings.plantuml_enabled
+          conf.svg_enable = false
+          conf.txt_enable = false
+
+          conf
+        end
+      end
+    end
+  end
+end

spec/lib/gitlab/plantuml_spec.rb

+# frozen_string_literal: true
+
+require "spec_helper"
+
+RSpec.describe Gitlab::Plantuml, feature_category: :shared do
+  describe ".configure" do
+    subject { described_class.configure }
+
+    let(:plantuml_url) { "http://plantuml.foo.bar" }
+
+    before do
+      allow(Gitlab::CurrentSettings).to receive(:plantuml_url).and_return(plantuml_url)
+    end
+
+    context "when PlantUML is enabled" do
+      before do
+        allow(Gitlab::CurrentSettings).to receive(:plantuml_enabled).and_return(true)
+      end
+
+      it "configures the endpoint URL" do
+        expect(subject.url).to eq(plantuml_url)
+      end
+
+      it "enables PNG support" do
+        expect(subject.png_enable).to be_truthy
+      end
+
+      it "disables SVG support" do
+        expect(subject.svg_enable).to be_falsey
+      end
+
+      it "disables TXT support" do
+        expect(subject.txt_enable).to be_falsey
+      end
+    end
+
+    context "when PlantUML is disabled" do
+      before do
+        allow(Gitlab::CurrentSettings).to receive(:plantuml_enabled).and_return(false)
+      end
+
+      it "configures the endpoint URL" do
+        expect(subject.url).to eq(plantuml_url)
+      end
+
+      it "enables PNG support" do
+        expect(subject.png_enable).to be_falsey
+      end
+
+      it "disables SVG support" do
+        expect(subject.svg_enable).to be_falsey
+      end
+
+      it "disables TXT support" do
+        expect(subject.txt_enable).to be_falsey
+      end
+    end
+  end
+end