spec/unit/file_serving/mount/plugins.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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70

#!/usr/bin/env ruby

require File.dirname(__FILE__) + '/../../../spec_helper'
require 'puppet/file_serving/mount/plugins'

describe Puppet::FileServing::Mount::Plugins, "when finding files" do
    before do
        @mount = Puppet::FileServing::Mount::Plugins.new("modules")

        @environment = stub 'environment', :module => nil
        @mount.stubs(:environment).returns @environment
    end

    it "should use the node's environment to find the modules" do
        env = mock 'env'
        @mount.expects(:environment).with("mynode").returns env
        env.expects(:modules).returns []

        @mount.find("foo", :node => "mynode")
    end

    it "should return nil if no module can be found with a matching plugin" do
        mod = mock 'module'
        mod.stubs(:plugin).with("foo/bar").returns nil

        @environment.expects(:modules).returns [mod]
        @mount.find("foo/bar").should be_nil
    end

    it "should return the file path from the module" do
        mod = mock 'module'
        mod.stubs(:plugin).with("foo/bar").returns "eh"

        @environment.expects(:modules).returns [mod]
        @mount.find("foo/bar").should == "eh"
    end
end

describe Puppet::FileServing::Mount::Plugins, "when searching for files" do
    before do
        @mount = Puppet::FileServing::Mount::Plugins.new("modules")

        @environment = stub 'environment', :module => nil
        @mount.stubs(:environment).returns @environment
    end

    it "should use the node's environment to find the modules" do
        env = mock 'env'
        @mount.expects(:environment).with("mynode").returns env
        env.expects(:modules).returns []

        @mount.search("foo", :node => "mynode")
    end

    it "should return nil if no modules can be found that have plugins" do
        mod = mock 'module'
        mod.stubs(:plugins?).returns false

        @environment.expects(:modules).returns [mod]
        @mount.search("foo/bar").should be_nil
    end

    it "should return the plugin paths for each module that has plugins" do
        one = stub 'module', :plugins? => true, :plugin_directory => "/one"
        two = stub 'module', :plugins? => true, :plugin_directory => "/two"

        @environment.expects(:modules).returns [one, two]
        @mount.search("foo/bar").should == %w{/one /two}
    end
end