spec/integration/util/feature_spec.rb


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
50
51
52
53
54

#!/usr/bin/env ruby

Dir.chdir(File.dirname(__FILE__)) { (s = lambda { |f| File.exist?(f) ? require(f) : Dir.chdir("..") { s.call(f) } }).call("spec/spec_helper.rb") }

require 'puppet/util/feature'
require 'puppet_spec/files'

describe Puppet::Util::Feature do
    include PuppetSpec::Files

    it "should be able to load features from disk" do
        libdir = tmpfile("feature_lib")
        Dir.mkdir(libdir)

        $LOAD_PATH << libdir

        $features = Puppet::Util::Feature.new("feature_lib")

        Dir.mkdir(File.join(libdir, "feature_lib"))

        File.open(File.join(libdir, "feature_lib", "able_to_load.rb"), "w") do |f|
            f.puts "$features.add(:able_to_load) { true }"
        end

        $features.should be_able_to_load
    end

    # TODO: Make this a spec test or remove it.
    def test_dynamic_loading
        $features = @features
        cleanup { $features = nil }
        # Now create a feature and make sure it loads.
        FileUtils.mkdir_p(@path)
        nope = File.join(@path, "nope.rb")
        File.open(nope, "w") { |f|
            f.puts "$features.add(:nope, :libs => %w{nosuchlib})"
        }
        assert_nothing_raised("Failed to autoload features") do
            assert(! @features.nope?, "'nope' returned true")
        end

        # First make sure "yep?" returns false
        assert_nothing_raised("Missing feature threw an exception") do
            assert(! @features.notyep?, "'notyep' returned true before definition")
        end

        yep = File.join(@path, "yep.rb")
        File.open(yep, "w") { |f|
            f.puts "$features.add(:yep, :libs => %w{puppet})"
        }

        assert(@features.yep?, "false 'yep' is apparently cached or feature could not be loaded")
    end
end