summaryrefslogtreecommitdiffstats
path: root/test/network/handler/handler.rb
blob: b812ca3d320e2d55de1f207530dfcd1bce9219f1 (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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
#!/usr/bin/env ruby

$:.unshift("../../lib") if __FILE__ =~ /\.rb$/

require 'puppettest'
require 'puppet/network/handler'

class TestHandler < Test::Unit::TestCase
    include PuppetTest::ServerTest

    def test_load_handlers
        # Make sure we don't get a failure but that we also get nothing back
        assert_nothing_raised do
            assert_nil(Puppet::Network::Handler.handler(:fake),
                "Got something back from a missing handler")
        end
        # Make a fake handler
        dir = tempfile()
        libdir = File.join([dir, %w{puppet network handler}].flatten)
        FileUtils.mkdir_p(libdir)

        file = File.join(libdir, "fake.rb")
        File.open(file, "w") do |f|
            f.puts %{class Puppet::Network::Handler
                class Fake < Handler
                end
            end
            }
        end

        $: << dir
        cleanup { $:.delete(dir) if $:.include?(dir) }

        handler = nil
        assert_nothing_raised do
            handler = Puppet::Network::Handler.handler(:fake)
        end
        assert(handler, "did not load handler")

        # Now make sure the handler behaves correctly
        assert_equal(:Fake, handler.name, "name was not calculated correctly")

        Puppet[:trace] = false
        assert_raise(Puppet::DevError,
            "did not throw an error on missing interface") do
                handler.interface
        end
    end

    def test_handlers_by_name
        %w{ca filebucket fileserver master report resource runner status}.each do |name|
            handler = nil
            assert_nothing_raised do
                handler = Puppet::Network::Handler.handler(name)
            end
            assert(handler, "did not get handler for %s" % name)
            assert(handler.name, "did not get name for %s" % name)
            assert(handler.interface, "did not get interface for %s" % name)
            assert(handler.interface.prefix, "did not get interface prefix for %s" % name)
        end
    end
end