1 files changed, 8 insertions, 1 deletions
diff --git a/spec/unit/application/puppetd.rb b/spec/unit/application/puppetd.rb
index dcd86185c..08f2a070a 100755
--- a/spec/unit/application/puppetd.rb
+++ b/spec/unit/application/puppetd.rb
@@ -327,11 +327,18 @@ describe "puppetd" do
             end
         end
 
-        it "should inform the daemon about our agent" do
+        it "should inform the daemon about our agent if :client is set to 'true'" do
+            @puppetd.options.expects(:[]).with(:client).returns true
             @daemon.expects(:agent=).with(@agent)
             @puppetd.run_setup
         end
 
+        it "should not inform the daemon about our agent if :client is set to 'false'" do
+            @puppetd.options[:client] = false
+            @daemon.expects(:agent=).never
+            @puppetd.run_setup
+        end
+
         it "should daemonize if needed" do
             Puppet.stubs(:[]).with(:daemonize).returns(true)