diff --git a/lib/draftjs_exporter/wrapper_state.rb b/lib/draftjs_exporter/wrapper_state.rb index 0e9c86a..f4396cb 100644 --- a/lib/draftjs_exporter/wrapper_state.rb +++ b/lib/draftjs_exporter/wrapper_state.rb @@ -41,10 +41,7 @@ def parent_for(type) new_options = nokogiri_options(*options.fetch(:wrapper)) return wrapper_element if new_options == wrapper_options - document.create_element(*new_options).tap do |new_element| - reset_wrapper.add_child(new_element) - set_wrapper(new_element, new_options) - end + create_wrapper(new_options) end def reset_wrapper @@ -62,5 +59,12 @@ def nokogiri_options(element_name, element_attributes) def block_options(type) block_map.fetch(type).fetch(:element) end + + def create_wrapper(options) + document.create_element(*options).tap do |new_element| + reset_wrapper.add_child(new_element) + set_wrapper(new_element, options) + end + end end end