nbudin-castronaut 0.7.5

data/spec/castronaut/models/consumeable_spec.rb

@@ -0,0 +1,39 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', 'spec_helper'))
+
+describe Castronaut::Models::Consumeable do
+  
+  class FakeModel
+    include Castronaut::Models::Consumeable
+
+    attr_accessor :consumed_at, :id
+    
+    def save!
+      
+    end
+  end
+  
+  describe "consume!" do
+  
+    it "sets consumed to the current time" do
+      fake_model = stub_model(FakeModel).as_new_record
+      fake_model.should_receive(:consumed_at=)
+      fake_model.consume!
+    end
+    
+    it "calls save!" do
+      fake_model = stub_model(FakeModel).as_new_record
+      fake_model.should_receive(:save!)
+      fake_model.consume!
+    end
+  
+  end
+  
+  it "is consumed when consumed_at has a value" do
+    fake_model = FakeModel.new
+    fake_model.should_not be_consumed
+    
+    fake_model.consumed_at = Time.now
+    fake_model.should be_consumed
+  end
+  
+end

data/spec/castronaut/models/dispenser_spec.rb

@@ -0,0 +1,30 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', 'spec_helper'))
+
+describe Castronaut::Models::Dispenser do
+  module Foo
+    module Bar
+      
+      class FakeModel
+        include Castronaut::Models::Dispenser
+    
+        attr_accessor :ticket
+    
+        def ticket_prefix
+          "WOO!"
+        end
+    
+        def write_attribute(attr, value)
+          instance_variable_set "@#{attr}", value
+        end
+      end
+      
+    end
+  end
+  
+  it "generates the ticket using the classes ticket prefix and a random string" do
+    Castronaut::Utilities::RandomString.stub!(:generate).and_return("RANDOM")
+    
+    Foo::Bar::FakeModel.new.send(:dispense_ticket).should == "WOO!-RANDOM"
+  end
+  
+end

data/spec/castronaut/models/login_ticket_spec.rb

@@ -0,0 +1,107 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', 'spec_helper'))
+include Castronaut::Models
+
+describe Castronaut::Models::LoginTicket do
+  
+  it "has no username" do
+    LoginTicket.new.username.should be_nil
+  end
+
+  it "has no proxies" do
+    LoginTicket.new.proxies.should be_nil
+  end
+  
+  it "has a ticket prefix of LT" do
+    LoginTicket.new.ticket_prefix.should == 'LT'
+  end
+  
+  it "requires a client hostname" do
+    login_ticket = LoginTicket.new
+    
+    login_ticket.should_not be_valid
+    login_ticket.errors.on(:client_hostname).should_not be_nil
+  end
+
+  it "requires a ticket" do
+    login_ticket = LoginTicket.new :client_hostname => 'http://example.com'
+    login_ticket.stub!(:dispense_ticket)
+
+    login_ticket.should_not be_valid
+    login_ticket.errors.on(:ticket).should_not be_nil
+  end
+
+  describe "generating from a client hostname" do
+
+    it "creates a new Login Ticket with given client hostname" do
+      LoginTicket.should_receive(:create!).with(:client_hostname => 'ch')
+      LoginTicket.generate_from('ch')
+    end
+
+  end
+  
+  describe "validating a ticket" do
+    
+    describe "when the ticket is missing" do
+      
+      it "returns a ticket result with the MissingMessage" do
+        Castronaut::TicketResult.should_receive(:new).with(nil, LoginTicket::MissingMessage)
+        LoginTicket.validate_ticket(nil)
+      end
+      
+    end
+    
+    it "attempts to the find the login ticket using the given ticket" do
+      LoginTicket.should_receive(:find_by_ticket).with("ticket").and_return(nil)
+      LoginTicket.validate_ticket("ticket")
+    end
+    
+    describe "when the ticket is invalid" do
+      
+      it "returns a ticket result with the InvalidMessage" do
+        Castronaut::TicketResult.should_receive(:new).with(nil, LoginTicket::InvalidMessage)
+        LoginTicket.stub!(:find_by_ticket).and_return(nil)
+        LoginTicket.validate_ticket("ticket")
+      end
+      
+    end
+    
+    describe "when the ticket is valid" do
+      
+      describe "and it has already been consumed" do
+        
+        it "returns a ticket result with the AlreadyConsumedMessage" do
+          login_ticket = stub(:consumed? => true).as_null_object
+          
+          Castronaut::TicketResult.should_receive(:new).with(login_ticket, LoginTicket::AlreadyConsumedMessage)
+          LoginTicket.stub!(:find_by_ticket).and_return(login_ticket)
+          LoginTicket.validate_ticket("ticket")
+        end
+        
+      end
+      
+      describe "and it has already expired" do
+        
+        it "returns a ticket result with the ExpiredMessage" do
+          login_ticket = stub_model(LoginTicket, :expired? => true, :consumed? => false)
+          LoginTicket.stub!(:find_by_ticket).and_return(login_ticket)
+          
+          Castronaut::TicketResult.should_receive(:new).with(login_ticket, LoginTicket::ExpiredMessage)
+          LoginTicket.validate_ticket("ticket")
+        end
+        
+      end
+      
+      it "consumes the ticket" do
+        login_ticket = stub_model(LoginTicket, :expired? => false, :consumed? => false)
+        login_ticket.should_receive(:consume!)
+        
+        LoginTicket.stub!(:find_by_ticket).and_return(login_ticket)
+
+        LoginTicket.validate_ticket("ticket")
+      end
+      
+    end
+    
+  end
+
+end

data/spec/castronaut/models/proxy_granting_ticket_spec.rb

@@ -0,0 +1,302 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', 'spec_helper'))
+include Castronaut::Models
+
+describe Castronaut::Models::ProxyGrantingTicket do
+
+  it "has no username" do
+    ProxyGrantingTicket.new.username.should be_nil
+  end
+
+  it "has no proxies" do
+    ProxyGrantingTicket.new.proxies.should be_nil
+  end
+
+  describe "dispensing a ticket" do
+    
+    it "starts with PGT" do
+      ProxyGrantingTicket.new.send(:dispense_ticket).starts_with?("PGT").should be_true
+    end
+    
+    it "is 64 characters long" do
+      ProxyGrantingTicket.new.send(:dispense_ticket).size.should == 64
+    end
+    
+    it "uses the random string generation for the remaining 60 characters" do
+      Castronaut::Utilities::RandomString.should_receive(:generate).with(60)
+      ProxyGrantingTicket.new.send(:dispense_ticket)
+    end
+    
+  end
+  
+  describe "dispensing an iou" do
+    
+    it "starts with PGTIOU" do
+      ProxyGrantingTicket.new.send(:dispense_iou).starts_with?("PGTIOU").should be_true
+    end
+    
+    it "is 64 characters long" do
+      ProxyGrantingTicket.new.send(:dispense_iou).size.should == 64
+    end
+    
+    it "uses the random string generation for the remaining 57 characters" do
+      Castronaut::Utilities::RandomString.should_receive(:generate).with(57)
+      ProxyGrantingTicket.new.send(:dispense_iou)
+    end
+    
+  end
+  
+  describe "generating a proxy granting ticket" do
+    
+    it "attempts to parse the given url with URI.parse" do
+      URI.should_receive(:parse).with('http://pgturl').and_raise(URI::InvalidURIError)
+      ProxyGrantingTicket.generate_ticket('http://pgturl', '10.1.1.1', 'service ticket')
+    end
+    
+    describe "when proxy granting ticket url is invalid it returns a ticket result" do
+      
+      it "with a message indicating it was unable to parse the given url" do
+        URI.should_receive(:parse).with('http://pgturl').and_raise(URI::InvalidURIError)
+        ProxyGrantingTicket.generate_ticket('http://pgturl', '10.1.1.1', 'service ticket').message.should == "Unable to parse pgt_url 'http://pgturl'"
+      end
+      
+      it "with a message category of warn" do
+        URI.should_receive(:parse).with('http://pgturl').and_raise(URI::InvalidURIError)
+        ProxyGrantingTicket.generate_ticket('http://pgturl', '10.1.1.1', 'service ticket').message_category.should == 'warn'
+      end
+      
+    end
+    
+    describe "when proxy granting ticket url is valid" do
+ 
+      it "creates a new net/http connection to uri host and port" do
+        https = stub({}).as_null_object
+        Net::HTTP.stub!(:new).and_return(https)  
+        
+        uri = stub('uri', :host => 'example.com', :port => 443, :path => '')
+        URI.stub!(:parse).and_return(uri)
+        
+        Net::HTTP.should_receive(:new).with('example.com', 443).and_return(https)
+        ProxyGrantingTicket.generate_ticket('http://example.com', '10.1.1.1', 'service ticket')
+      end
+      
+      it "sets the new net/http connection to use ssl" do
+        https = stub({}).as_null_object
+        Net::HTTP.stub!(:new).and_return(https)  
+        
+        uri = stub(:host => 'example.com', :port => 443, :path => '').as_null_object
+        URI.stub!(:parse).and_return(uri)
+        
+        https.should_receive(:use_ssl=).with(true)
+
+        Net::HTTP.stub!(:new).with('example.com', 443).and_return(https)
+
+        ProxyGrantingTicket.generate_ticket('http://example.com', '10.1.1.1', 'service ticket')
+      end
+      
+      it "starts the connection" do
+        https = stub({}).as_null_object
+        Net::HTTP.stub!(:new).and_return(https)  
+        
+        uri = stub(:host => 'example.com', :port => 443, :path => '').as_null_object
+        URI.stub!(:parse).and_return(uri)
+        
+        https.should_receive(:start)
+        
+        ProxyGrantingTicket.generate_ticket('http://example.com', '10.1.1.1', 'service ticket')
+      end
+      
+      describe "inside the http connection block" do
+        
+        it "intializes a new proxy granting ticket" do
+          https = stub({}).as_null_object
+          Net::HTTP.stub!(:new).and_return(https)  
+
+          uri = stub(:host => 'example.com', :port => 443, :path => '').as_null_object
+          URI.stub!(:parse).and_return(uri)
+          
+          https.stub!(:start).and_yield(stub(:request_get => stub({}).as_null_object).as_null_object)
+          
+          ProxyGrantingTicket.should_receive(:new).with(:service_ticket => 'service ticket', :client_hostname => '10.1.1.1').and_return(stub({}).as_null_object)
+          ProxyGrantingTicket.generate_ticket('http://example.com', '10.1.1.1', 'service ticket')
+        end
+        
+        it "dispenses a new ticket for the proxy granting ticket" do
+          https = stub({}).as_null_object
+          Net::HTTP.stub!(:new).and_return(https)  
+
+          uri = stub(:host => 'example.com', :port => 443, :path => '').as_null_object
+          URI.stub!(:parse).and_return(uri)
+
+          https.stub!(:start).and_yield(stub(:request_get => stub({}).as_null_object).as_null_object)
+
+          proxy_granting_ticket = stub({}).as_null_object
+          proxy_granting_ticket.should_receive(:dispense_ticket)
+          ProxyGrantingTicket.stub!(:new).and_return(proxy_granting_ticket)
+          
+          ProxyGrantingTicket.generate_ticket('http://example.com', '10.1.1.1', 'service ticket')
+        end
+
+        it "dispenses a new iou for the proxy granting ticket" do
+          https = stub({}).as_null_object
+          Net::HTTP.stub!(:new).and_return(https)  
+
+          uri = stub(:host => 'example.com', :port => 443, :path => '').as_null_object
+          URI.stub!(:parse).and_return(uri)
+
+          https.stub!(:start).and_yield(stub(:request_get => stub({}).as_null_object).as_null_object)
+
+          proxy_granting_ticket = stub({}).as_null_object
+          proxy_granting_ticket.should_receive(:dispense_iou)
+          ProxyGrantingTicket.stub!(:new).and_return(proxy_granting_ticket)
+          
+          ProxyGrantingTicket.generate_ticket('http://example.com', '10.1.1.1', 'service ticket')
+        end
+        
+        it "requests the proxy granting ticket path via GET" do
+          https = stub({}).as_null_object
+          Net::HTTP.stub!(:new).and_return(https)  
+
+          uri = stub(:host => 'example.com', :port => 443, :path => '').as_null_object
+          URI.stub!(:parse).and_return(uri)
+          http_connection = mock('httpconnection')
+          http_connection.should_receive(:request_get).with('/?pgtId=PGT-RANDOM&pgtIou=PGTIOU-RANDOM').and_return(stub({}).as_null_object)
+          
+          https.stub!(:start).and_yield(http_connection)
+          
+          ProxyGrantingTicket.stub!(:new).and_return(stub('proxy granting ticket', :iou => 'PGTIOU-RANDOM', :ticket => 'PGT-RANDOM', :dispense_ticket => nil, :dispense_iou => nil))
+          
+          ProxyGrantingTicket.generate_ticket('http://example.com', '10.1.1.1', 'service ticket')
+        end
+        
+        describe "when the request to the proxy granting ticket path is successful" do
+          
+          it "saves the proxy granting ticket" do
+            https = stub({}).as_null_object
+            Net::HTTP.stub!(:new).and_return(https)  
+
+            uri = stub('uri', :host => 'example.com', :port => 443, :path => '', :query => '')
+            URI.stub!(:parse).and_return(uri)
+            http_connection = mock('httpconnection')
+            http_connection.stub!(:request_get).and_return(stub('response', :code => '200'))
+            https.stub!(:start).and_yield(http_connection)
+
+            proxy_granting_ticket = mock_model(ProxyGrantingTicket, :ticket => 'PGT-RANDOM', :iou => 'PGTIOU-RANDOM', :dispense_ticket => nil, :dispense_iou => nil)
+            proxy_granting_ticket.should_receive(:save!)
+            
+            ProxyGrantingTicket.stub!(:new).and_return(proxy_granting_ticket)
+
+            ProxyGrantingTicket.generate_ticket('http://example.com', '10.1.1.1', 'service ticket')
+          end
+          
+          describe "returns a ticket result" do
+            
+            it "with the proxy granting ticket" do
+              https = stub({}).as_null_object
+              Net::HTTP.stub!(:new).and_return(https)  
+
+              uri = stub('uri', :host => 'example.com', :port => 443, :path => '', :query => '')
+              URI.stub!(:parse).and_return(uri)
+              http_connection = mock('httpconnection')
+              http_connection.stub!(:request_get).and_return(stub('request', :code => '200'))
+              https.stub!(:start).and_yield(http_connection)
+
+              proxy_granting_ticket = stub('ProxyGrantingTicket', :ticket => 'PGT-RANDOM', :iou => 'PGTIOU-RANDOM', :dispense_ticket => nil, :dispense_iou => nil, :save! => nil)
+              ProxyGrantingTicket.stub!(:new).and_return(proxy_granting_ticket)              
+              
+              ProxyGrantingTicket.generate_ticket('http://example.com', '10.1.1.1', 'service ticket').ticket.should == proxy_granting_ticket
+            end
+            
+            it "with a proxy granting ticket generated message" do
+              https = stub({}).as_null_object
+              Net::HTTP.stub!(:new).and_return(https)  
+
+              uri = stub(:host => 'example.com', :port => 443, :path => '').as_null_object
+              URI.stub!(:parse).and_return(uri)
+              http_connection = mock('httpconnection')
+              http_connection.stub!(:request_get).and_return(stub('request', :code => '200'))
+              https.stub!(:start).and_yield(http_connection)
+
+              proxy_granting_ticket = stub('proxy granting ticket', :iou => 'PGTIOU-RANDOM', :ticket => 'PGT-RANDOM', :dispense_ticket => nil, :dispense_iou => nil, :save! => true, :inspect => 'PGT-INSPECT')
+              ProxyGrantingTicket.stub!(:new).and_return(proxy_granting_ticket)              
+              
+              ProxyGrantingTicket.generate_ticket('http://example.com', '10.1.1.1', 'service ticket').message.should == "PGT generated for pgt_url 'http://example.com': PGT-INSPECT"
+            end
+            
+            it "with a message category of success" do
+              https = stub({}).as_null_object
+              Net::HTTP.stub!(:new).and_return(https)  
+
+              uri = stub(:host => 'example.com', :port => 443, :path => '').as_null_object
+              URI.stub!(:parse).and_return(uri)
+              http_connection = mock('httpconnection')
+              http_connection.stub!(:request_get).and_return(stub('request', :code => '200'))
+              https.stub!(:start).and_yield(http_connection)
+
+              proxy_granting_ticket = stub(:iou => 'PGTIOU-RANDOM', :ticket => 'PGT-RANDOM', :save! => true).as_null_object
+              ProxyGrantingTicket.stub!(:new).and_return(proxy_granting_ticket)              
+              
+              ProxyGrantingTicket.generate_ticket('http://example.com', '10.1.1.1', 'service ticket').message_category.should == 'success'
+            end
+                      
+          end
+
+        end
+        
+        describe "when the request to the proxy granting ticket path has failed" do
+          
+           describe "returns a ticket result" do
+
+              it "with a proxy granting ticket generated message" do
+                https = stub({}).as_null_object
+                Net::HTTP.stub!(:new).and_return(https)  
+
+                uri = stub(:host => 'example.com', :port => 443, :path => '').as_null_object
+                URI.stub!(:parse).and_return(uri)
+                http_connection = mock('httpconnection')
+                http_connection.stub!(:request_get).and_return(stub('request', :code => '404'))
+                https.stub!(:start).and_yield(http_connection)
+
+                proxy_granting_ticket = stub(:iou => 'PGTIOU-RANDOM', :ticket => 'PGT-RANDOM', :save! => true, :inspect => 'PGT-INSPECT').as_null_object
+                ProxyGrantingTicket.stub!(:new).and_return(proxy_granting_ticket)              
+
+                ProxyGrantingTicket.generate_ticket('http://example.com', '10.1.1.1', 'service ticket').message.should == "PGT callback server responded with a bad result code '404'. PGT will not be stored."
+              end
+
+              it "with a message category of success" do
+                https = stub({}).as_null_object
+                Net::HTTP.stub!(:new).and_return(https)  
+
+                uri = stub(:host => 'example.com', :port => 443, :path => '').as_null_object
+                URI.stub!(:parse).and_return(uri)
+                http_connection = mock('httpconnection')
+                http_connection.stub!(:request_get).and_return(stub(:code => '404').as_null_object)
+                https.stub!(:start).and_yield(http_connection)
+
+                proxy_granting_ticket = stub(:iou => 'PGTIOU-RANDOM', :ticket => 'PGT-RANDOM', :save! => true).as_null_object
+                ProxyGrantingTicket.stub!(:new).and_return(proxy_granting_ticket)              
+
+                ProxyGrantingTicket.generate_ticket('http://example.com', '10.1.1.1', 'service ticket').message_category.should == 'warn'
+              end
+            
+            end
+          
+        end
+        
+      end
+      
+    end
+   
+  end
+
+  describe "when processing a logout request" do
+
+    it "destroys the proxy granting ticket" do
+      proxy_granting_ticket = stub('proxy granting ticket')
+      ProxyGrantingTicket.stub!(:all).and_return([proxy_granting_ticket])
+      proxy_granting_ticket.should_receive(:destroy)
+      ProxyGrantingTicket.clean_up_proxy_granting_tickets_for('bob')
+    end
+
+  end
+
+end