Refactor Generator to initialize with destination and content

Author: elct9620

lib/boxing/generator.rb

@@ -7,22 +7,32 @@ module Boxing
   class Generator
     include Utils
 
+    # @since 0.11.0
+    def initialize(destination, content)
+      pp current_path
+      @destination = current_path.join(destination)
+      @content = content
+    end
+
     # Generate file
     #
     # @since 0.11.0
-    def execute(destination, content = nil)
-      content = yield if block_given?
-      destination = current_path.join(destination)
-      write(destination, content)
+    def execute
+      FileUtils.mkdir_p(File.dirname(@destination))
+      File.write(@destination, render)
     end
 
-    # Write file to relative path
+    # Render content
     #
-    # @param [String] destination
-    # @param [String] content
-    def write(destination, content)
-      FileUtils.mkdir_p(File.dirname(destination))
-      File.write(destination, content)
+    # @return [String]
+    #
+    # @since 0.11.0
+    def render
+      @render ||= if @content.is_a?(Proc)
+                    @content.call
+                  else
+                    @content
+                  end
     end
   end
 end

lib/boxing/utils.rb

@@ -27,9 +27,7 @@ def current_path
     #
     # @since 0.11.0
     def template(destination, template, context: nil)
-      Generator.new.execute(destination) do
-        Template.new(template).render(context)
-      end
+      Generator.new(destination, -> { Template.new(template).render(context) }).execute
     end
   end
 end

spec/boxing/generator_spec.rb

@@ -3,17 +3,15 @@
 require 'tmpdir'
 
 RSpec.describe Boxing::Generator do
-  subject(:generator) { described_class.new }
+  subject(:generator) { described_class.new('Dockerfile', 'test') }
 
   let(:tmpdir) { Pathname.new(Dir.mktmpdir) }
 
-  before do
-    allow(generator).to receive(:current_path).and_return(tmpdir)
-
-    generator.execute('Dockerfile', 'test')
-  end
+  around { |example| Dir.chdir(tmpdir) { example.run } }
   after { FileUtils.remove_entry(tmpdir) }
 
+  before { generator.execute }
+
   it 'creates a file' do
     expect(File.exist?(File.join(tmpdir, 'Dockerfile'))).to be true
   end