1 files changed, 167 insertions, 0 deletions
diff --git a/spec/unit/application/ca.rb b/spec/unit/application/ca.rb
new file mode 100644
index 000000000..b9a716e3f
--- /dev/null
+++ b/spec/unit/application/ca.rb
@@ -0,0 +1,167 @@
+#!/usr/bin/env ruby
+
+require File.dirname(__FILE__) + '/../../spec_helper'
+
+require 'puppet/application/ca'
+
+describe "PuppetCA" do
+    before :each do
+        @ca_app = Puppet::Application[:ca]
+        Puppet::Util::Log.stubs(:newdestination)
+        Puppet::Util::Log.stubs(:level=)
+    end
+
+    it "should ask Puppet::Application to parse Puppet configuration file" do
+        @ca_app.should_parse_config?.should be_true
+    end
+
+    it "should declare a main command" do
+        @ca_app.should respond_to(:main)
+    end
+
+    Puppet::SSL::CertificateAuthority::Interface::INTERFACE_METHODS.reject{ |m| m == :destroy }.each do |method|
+        it "should declare option --#{method}" do
+            @ca_app.should respond_to("handle_#{method}".to_sym)
+        end
+    end
+
+    it "should set log level to info with the --verbose option" do
+
+        Puppet::Log.expects(:level=).with(:info)
+
+        @ca_app.handle_verbose(0)
+    end
+
+    it "should set log level to debug with the --debug option" do
+
+        Puppet::Log.expects(:level=).with(:debug)
+
+        @ca_app.handle_debug(0)
+    end
+
+    it "should set the fingerprint digest with the --digest option" do
+        @puppetca.handle_digest(:digest)
+
+        @puppetca.digest.should == :digest
+    end
+
+    it "should set mode to :destroy for --clean" do
+        @ca_app.handle_clean(0)
+        @ca_app.mode.should == :destroy
+    end
+
+    it "should set all to true for --all" do
+        @ca_app.handle_all(0)
+        @ca_app.all.should be_true
+    end
+
+    Puppet::SSL::CertificateAuthority::Interface::INTERFACE_METHODS.reject { |m| m == :destroy }.each do |method|
+        it "should set mode to #{method} with option --#{method}" do
+            @ca_app.send("handle_#{method}".to_sym, nil)
+
+            @ca_app.mode.should == method
+        end
+    end
+
+    describe "during setup" do
+
+        before :each do
+            Puppet::Log.stubs(:newdestination)
+            Puppet::SSL::Host.stubs(:ca_location=)
+            Puppet::SSL::CertificateAuthority.stubs(:new)
+        end
+
+        it "should set console as the log destination" do
+            Puppet::Log.expects(:newdestination).with(:console)
+
+            @ca_app.run_setup
+        end
+
+        it "should print puppet config if asked to in Puppet config" do
+            @ca_app.stubs(:exit)
+            Puppet.settings.stubs(:print_configs?).returns(true)
+
+            Puppet.settings.expects(:print_configs)
+
+            @ca_app.run_setup
+        end
+
+        it "should exit after printing puppet config if asked to in Puppet config" do
+            Puppet.settings.stubs(:print_configs?).returns(true)
+
+            lambda { @ca_app.run_setup }.should raise_error(SystemExit)
+        end
+
+        it "should set the CA location to 'only'" do
+            Puppet::SSL::Host.expects(:ca_location=).with(:only)
+
+            @ca_app.run_setup
+        end
+
+        it "should create a new certificate authority" do
+            Puppet::SSL::CertificateAuthority.expects(:new)
+
+            @ca_app.run_setup
+        end
+    end
+
+    describe "when running" do
+        before :each do
+            @ca_app.all = false
+            @ca = stub_everything 'ca'
+            @ca_app.ca = @ca
+            ARGV.stubs(:collect).returns([])
+        end
+
+        it "should delegate to the CertificateAuthority" do
+            @ca.expects(:apply)
+
+            @ca_app.main
+        end
+
+        it "should delegate with :all if option --all was given" do
+            @ca_app.handle_all(0)
+
+            @ca.expects(:apply).with { |mode,to| to[:to] == :all }
+
+            @ca_app.main
+        end
+
+        it "should delegate to ca.apply with the hosts given on command line" do
+            ARGV.stubs(:collect).returns(["host"])
+
+            @ca.expects(:apply).with { |mode,to| to[:to] == ["host"]}
+
+            @ca_app.main
+        end
+
+        it "should send the currently set digest" do
+            ARGV.stubs(:collect).returns(["host"])
+            @puppetca.handle_digest(:digest)
+
+            @ca.expects(:apply).with { |mode,to| to[:digest] == :digest}
+
+            @puppetca.main
+        end
+
+        it "should delegate to ca.apply with current set mode" do
+            @ca_app.mode = "currentmode"
+            ARGV.stubs(:collect).returns(["host"])
+
+            @ca.expects(:apply).with { |mode,to| mode == "currentmode" }
+
+            @ca_app.main
+        end
+
+        it "should revoke cert if mode is clean" do
+            @puppetca.mode = :destroy
+            ARGV.stubs(:collect).returns(["host"])
+
+            @ca.expects(:apply).with { |mode,to| mode == :revoke }
+            @ca.expects(:apply).with { |mode,to| mode == :destroy }
+
+            @puppetca.main
+        end
+
+    end
+end