aboutsummaryrefslogtreecommitdiff
path: root/spec/puppetlabs_spec/fixtures.rb
blob: 9ce0a6b097e3e7534bd3e1a5652107a4f05c6a7c (plain)
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
44
45
46
47
48
49
# This module provides some helper methods to assist with fixtures. It's
# methods are designed to help when you have a conforming fixture layout so we
# get project consistency.
module PuppetlabsSpec::Fixtures

  # Returns the joined path of the global FIXTURE_DIR plus any path given to it
  def fixtures(*rest)
    File.join(PuppetlabsSpec::FIXTURE_DIR, *rest)
  end

  # Returns the path to your relative fixture dir. So if your spec test is
  # <project>/spec/unit/facter/foo_spec.rb then your relative dir will be
  # <project>/spec/fixture/unit/facter/foo
  def my_fixture_dir
    callers = caller
    while line = callers.shift do
      next unless found = line.match(%r{/spec/(.*)_spec\.rb:})
      return fixtures(found[1])
    end
    fail "sorry, I couldn't work out your path from the caller stack!"
  end

  # Given a name, returns the full path of a file from your relative fixture
  # dir as returned by my_fixture_dir.
  def my_fixture(name)
    file = File.join(my_fixture_dir, name)
    unless File.readable? file then
      fail "fixture '#{name}' for #{my_fixture_dir} is not readable"
    end
    return file
  end

  # Return the contents of the file using read when given a name. Uses
  # my_fixture to work out the relative path.
  def my_fixture_read(name)
    File.read(my_fixture(name))
  end

  # Provides a block mechanism for iterating across the files in your fixture
  # area.
  def my_fixtures(glob = '*', flags = 0)
    files = Dir.glob(File.join(my_fixture_dir, glob), flags)
    unless files.length > 0 then
      fail "fixture '#{glob}' for #{my_fixture_dir} had no files!"
    end
    block_given? and files.each do |file| yield file end
    files
  end
end