From 946081b3c0bf24dc68a3a08fbcb6b5347e01de8f Mon Sep 17 00:00:00 2001
From: Andrew Shafer <andrew@reductivelabs.com>
Date: Tue, 10 Jun 2008 23:19:37 -0600
Subject: Try again

Add the class to handle certs, specs and modify puppetd

should fix 1190, 1199, 1200
---
 spec/unit/executables/client/certhandler.rb | 80 +++++++++++++++++++++++++++++
 1 file changed, 80 insertions(+)
 create mode 100755 spec/unit/executables/client/certhandler.rb

(limited to 'spec/unit/executables/client')

diff --git a/spec/unit/executables/client/certhandler.rb b/spec/unit/executables/client/certhandler.rb
new file mode 100755
index 000000000..d12c51c1b
--- /dev/null
+++ b/spec/unit/executables/client/certhandler.rb
@@ -0,0 +1,80 @@
+#!/usr/bin/env ruby
+
+Dir.chdir(File.dirname(__FILE__)) { (s = lambda { |f| File.exist?(f) ? require(f) : Dir.chdir("..") { s.call(f) } }).call("spec/spec_helper.rb") }
+
+require 'puppet/executables/client/certhandler'
+
+cert_handler = Puppet::Executables::Client::CertHandler
+
+describe cert_handler, "when handling certificates" do
+    before do 
+        @caclient = mock('caclient')
+        caclient_class = mock('caclient_class')
+        caclient_class.stubs(:new).returns(@caclient)
+        Puppet::Network::Client.stubs(:ca).returns(caclient_class)
+    end
+
+    it "should return true if the certificate exists" do
+        Puppet::Network::HttpPool.expects(:read_cert).returns(true)
+        cert_handler.new(1,true).read_retrieve.should be_true
+    end
+    
+    it "should return false when getting a new cert" do
+        Puppet::Network::HttpPool.expects(:read_cert).returns(true)
+        @caclient.stubs(:request_cert).returns(true)
+        ch = cert_handler.new(1,true)
+        ch.stubs(:read_cert).returns(false)
+        ch.read_retrieve.should be_false
+    end
+
+    describe "when waiting for cert" do
+        it "should loop when the cert request fails" do
+            @caclient.stubs(:request_cert).times(2).returns(false).then.returns(true)
+            ch = cert_handler.new(1,false)
+            ch.expects(:sleep)
+            ch.expects(:read_new_cert).returns(true)
+            ch.read_retrieve
+        end
+
+        it "should loop when the cert request raises an Error" do
+            @caclient.stubs(:request_cert).times(2).raises(StandardError, 'Testing').then.returns(true)
+            ch = cert_handler.new(1,false)
+            ch.expects(:sleep)
+            ch.expects(:read_new_cert).returns(true)
+            ch.read_retrieve
+        end
+        
+        it "should loop when the new cert can't be read" do
+            @caclient.stubs(:request_cert).returns(true)
+            ch = cert_handler.new(1,false)
+            ch.expects(:sleep)
+            ch.expects(:read_new_cert).times(2).returns(false).then.returns(true)
+            ch.read_retrieve
+        end
+    end
+
+    describe "when in one time mode" do
+        it "should exit if the cert request fails" do
+            @caclient.stubs(:request_cert).returns(false)
+            ch = cert_handler.new(1,true)
+            ch.expects(:exit).with(1).raises(SystemExit)
+            lambda { ch.read_retrieve }.should raise_error(SystemExit)
+        end
+
+
+        it "should exit if the cert request raises an exception" do
+            @caclient.stubs(:request_cert).raises(StandardError, 'Testing')
+            ch = cert_handler.new(1,true)
+            ch.expects(:exit).with(23).raises(SystemExit)
+            lambda { ch.read_retrieve }.should raise_error(SystemExit)
+        end
+        
+        it "should exit if the new cert can't be read" do
+            @caclient.stubs(:request_cert).returns(true)
+            Puppet::Network::HttpPool.stubs(:read_cert).returns(false)
+            ch = cert_handler.new(1,true)
+            ch.expects(:exit).with(34).raises(SystemExit)
+            lambda { ch.read_retrieve }.should raise_error(SystemExit)
+        end
+    end
+end
-- 
cgit 


From 6764af3c196a526647eb4a270ec737e2b0f20d36 Mon Sep 17 00:00:00 2001
From: Andrew Shafer <andrew@reductivelabs.com>
Date: Tue, 10 Jun 2008 23:37:00 -0600
Subject: Change description of spec to make baby jesus happy

FAIL
---
 spec/unit/executables/client/certhandler.rb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'spec/unit/executables/client')

diff --git a/spec/unit/executables/client/certhandler.rb b/spec/unit/executables/client/certhandler.rb
index d12c51c1b..0a7b77f15 100755
--- a/spec/unit/executables/client/certhandler.rb
+++ b/spec/unit/executables/client/certhandler.rb
@@ -28,7 +28,7 @@ describe cert_handler, "when handling certificates" do
     end
 
     describe "when waiting for cert" do
-        it "should loop when the cert request fails" do
+        it "should loop when the cert request does not return a certificate" do
             @caclient.stubs(:request_cert).times(2).returns(false).then.returns(true)
             ch = cert_handler.new(1,false)
             ch.expects(:sleep)
@@ -54,7 +54,7 @@ describe cert_handler, "when handling certificates" do
     end
 
     describe "when in one time mode" do
-        it "should exit if the cert request fails" do
+        it "should exit if the cert request does not return a certificate" do
             @caclient.stubs(:request_cert).returns(false)
             ch = cert_handler.new(1,true)
             ch.expects(:exit).with(1).raises(SystemExit)
-- 
cgit