Contents

module Linecook
  module Test
    module FileTest
      module ClassMethods
        attr_accessor :class_dir
        
        def self.extended(base)
          # Infers the test directory from the calling file.
          #   'some_class_test.rb' => 'some_class_test'
          calling_file = caller[2].gsub(/:\d+(:in .*)?$/, "")
          base.class_dir = calling_file.chomp(File.extname(calling_file))
        end
      end
    
      module ModuleMethods
        module_function
      
        def included(base)
          base.extend base.kind_of?(Class) ? ClassMethods : ModuleMethods
          super
        end
      end
    
      extend ModuleMethods
    
      attr_reader :user_dir
      attr_reader :method_dir
    
      def setup
        super
        @user_dir   = Dir.pwd
        @method_dir = File.expand_path(method_name, self.class.class_dir)
      
        cleanup method_dir
        FileUtils.mkdir_p method_dir
        Dir.chdir method_dir
      end
    
      def teardown
        Dir.chdir user_dir
      
        unless ENV["KEEP_OUTPUTS"] == "true"
          cleanup class_dir
        end
      
        super
      end
    
      def cleanup(dir)
        FileUtils.rm_r(dir) if File.exists?(dir)
      end
    
      def path(relative_path)
        File.expand_path(relative_path, method_dir)
      end
      
      def prepare(relative_path)
        target = path(relative_path)
        
        target_dir = File.dirname(target)
        FileUtils.mkdir_p(target_dir) unless File.exists?(target_dir)
        
        target
      end
      
      def file(relative_path, &block)
        target = prepare(relative_path)
        block ? File.open(target, 'w', &block) : FileUtils.touch(target)
        target
      end
    
      def class_dir
        self.class.class_dir
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
linecook-0.6.2	lib/linecook/test/file_test.rb