lib/specinfra/backend/dockerfile.rb in specinfra-2.5.0 vs lib/specinfra/backend/dockerfile.rb in specinfra-2.6.0

@@ -2,10 +2,14 @@
 module Specinfra::Backend
   class Dockerfile < Specinfra::Backend::Base
     def initialize
       @lines = []
       ObjectSpace.define_finalizer(self) {
-        puts @lines
+        if Specinfra.configuration.dockerfile_finalizer.nil?
+          puts @lines
+        else
+          Specinfra.configuration.dockerfile_finalizer.call(@lines)
+        end
       }
     end
 
     def run_command(cmd, opts={})
       @lines << "RUN #{cmd}"