summaryrefslogtreecommitdiffstats
path: root/spec/integration/network/server/webrick_spec.rb
blob: 81c35af4f4ba0196629a686c9a5c74da8bfebc6e (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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
#!/usr/bin/env rspec
require 'spec_helper'
require 'puppet/network/server'
require 'puppet/ssl/certificate_authority'
require 'socket'

describe Puppet::Network::Server do
  describe "when using webrick" do
    before :each do
      Puppet[:servertype] = 'webrick'
      Puppet[:server] = '127.0.0.1'
      @params = { :port => 34343, :handlers => [ :node ], :xmlrpc_handlers => [ :status ] }

      # Get a safe temporary file
      @tmpfile = Tempfile.new("webrick_integration_testing")
      @dir = @tmpfile.path + "_dir"

      Puppet.settings[:confdir] = @dir
      Puppet.settings[:vardir] = @dir
      Puppet.settings[:group] = Process.gid

      Puppet::SSL::Host.ca_location = :local

      ca = Puppet::SSL::CertificateAuthority.new
      ca.generate(Puppet[:certname]) unless Puppet::SSL::Certificate.indirection.find(Puppet[:certname])
    end

    after do
      @tmpfile.delete
      Puppet.settings.clear

      system("rm -rf #{@dir}")

      Puppet::SSL::Host.ca_location = :none
      Puppet::Util::Cacher.expire
    end

    describe "before listening" do
      it "should not be reachable at the specified address and port" do
        lambda { TCPSocket.new('127.0.0.1', 34343) }.should raise_error
      end
    end

    describe "when listening" do
      it "should be reachable on the specified address and port" do
        @server = Puppet::Network::Server.new(@params.merge(:port => 34343))
        @server.listen
        lambda { TCPSocket.new('127.0.0.1', 34343) }.should_not raise_error
      end

      it "should default to '0.0.0.0' as its bind address" do
        Puppet.settings.clear
        Puppet[:servertype] = 'webrick'
        Puppet[:bindaddress].should == '0.0.0.0'
      end

      it "should use any specified bind address" do
        Puppet[:bindaddress] = "127.0.0.1"
        @server = Puppet::Network::Server.new(@params.merge(:port => 34343))
        @server.stubs(:unlisten) # we're breaking listening internally, so we have to keep it from unlistening
        @server.send(:http_server).expects(:listen).with { |args| args[:address] == "127.0.0.1" }
        @server.listen
      end

      it "should not allow multiple servers to listen on the same address and port" do
        @server = Puppet::Network::Server.new(@params.merge(:port => 34343))
        @server.listen
        @server2 = Puppet::Network::Server.new(@params.merge(:port => 34343))
        lambda { @server2.listen }.should raise_error
      end

      after :each do
        @server.unlisten if @server && @server.listening?
      end
    end

    describe "after unlistening" do
      it "should not be reachable on the port and address assigned" do
        @server = Puppet::Network::Server.new(@params.merge(:port => 34343))
        @server.listen
        @server.unlisten
        lambda { TCPSocket.new('127.0.0.1', 34343) }.should raise_error(Errno::ECONNREFUSED)
      end
    end
  end
end