aboutsummaryrefslogtreecommitdiffstats
path: root/activerecord/lib/active_record/fixtures/file.rb
blob: 9e95007f40c5fcfa551d499a68b8ce34df9c4885 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
begin
  require 'psych'
rescue LoadError
end

require 'erb'
require 'yaml'

module ActiveRecord
  class Fixtures
    class File
      include Enumerable

      ##
      # Open a fixture file named +file+.  When called with a block, the block
      # is called with the filehandle and the filehandle is automatically closed
      # when the block finishes.
      def self.open(file)
        x = new file
        block_given? ? yield(x) : x
      end

      def initialize(file)
        @file = file
        @rows = nil
      end

      def each(&block)
        rows.each(&block)
      end

      private
      def rows
        return @rows if @rows
        @rows = YAML.load(render(IO.read(@file))).to_a
      end

      def render(content)
        ERB.new(content).result
      end
    end
  end
end