First pass through initializers of {mongrel, webrick} REST handlers; hooks into Indirection to look up models from indirected names.

1 files changed, 3 insertions, 5 deletions

diff --git a/spec/unit/network/http/webrick.rb b/spec/unit/network/http/webrick.rb
index 81b2a0fa9..4bf742b6e 100644
--- a/spec/unit/network/http/webrick.rb
+++ b/spec/unit/network/http/webrick.rb
@@ -64,11 +64,9 @@ describe Puppet::Network::HTTP::WEBrick, "when turning on listening" do
         @listen_params[:protocols].each do |protocol|
             mock_handler = mock("handler instance for [#{protocol}]")
             mock_handler_class = mock("handler class for [#{protocol}]")
-            @listen_params[:handlers].each do |handler|
-                mock_handler_class.expects(:new).with {|args| 
-                    args[:server] == @mock_webrick and args[:handler] == handler
-                }.returns(mock_handler)
-            end
+            mock_handler_class.expects(:new).with {|args| 
+                args[:server] == @mock_webrick and args[:handlers] == @listen_params[:handlers]
+            }.returns(mock_handler)
             @server.expects(:class_for_protocol).with(protocol).at_least_once.returns(mock_handler_class)
         end
         @server.listen(@listen_params)