Refactor component_with_doc, add test

lib/komponent/komponent_helper.rb

@@ -21,19 +21,22 @@ def komponent_components
   def component_with_doc(component_name, locals = {}, options = {}, &block)
     captured_output = component(component_name, locals, options, &block)
 
-    if locals.present?
-      pretty_locals = JSON.pretty_generate(locals).gsub(/^(\s+)"(\w+)":/, "\\1\\2:")
-    end
-
     captured_doc = capture do
       content_tag :pre, class: "komponent-code" do
         content_tag :code do
-          "= component \"#{component_name}\"" + (pretty_locals ? ", #{pretty_locals}" : "")
+          "= component \"#{component_name}\"" + (locals.present? ? ", #{pretty_locals(locals)}" : "")
         end
       end
     end
 
     captured_output + captured_doc
   end
   alias :cdoc :component_with_doc
+
+  private
+
+  def pretty_locals(locals)
+    return nil if locals.blank?
+    JSON.pretty_generate(locals).gsub(/^(\s+)"(\w+)":/, "\\1\\2:")
+  end
 end

test/komponent/komponent_helper_test.rb

@@ -62,4 +62,14 @@ def test_helper_renders_yield_args
       %(<div class="foo-bar">Foo Bar</div>),
       component('foo_bar') { |x| x }.chomp
   end
+
+  def test_helper_renders_with_doc
+    assert_equal \
+      %(<div class="all">🌎 😎</div>
+<pre class="komponent-code"><code>= component &quot;all&quot;, {
+  world: &quot;🌎&quot;,
+  sunglasses: &quot;😎&quot;
+}</code></pre>),
+      component_with_doc('all', world: "🌎", sunglasses: "😎").chomp
+  end
 end