ezmobius-nanite 0.4.0 → 0.4.1.1

data/lib/nanite/util.rb

@@ -0,0 +1,51 @@
+class String
+  ##
+  # Convert to snake case.
+  #
+  #   "FooBar".snake_case           #=> "foo_bar"
+  #   "HeadlineCNNNews".snake_case  #=> "headline_cnn_news"
+  #   "CNN".snake_case              #=> "cnn"
+  #
+  # @return [String] Receiver converted to snake case.
+  #
+  # @api public
+  def snake_case
+    return self.downcase if self =~ /^[A-Z]+$/
+    self.gsub(/([A-Z]+)(?=[A-Z][a-z]?)|\B[A-Z]/, '_\&') =~ /_*(.*)/
+    return $+.downcase
+  end
+
+  ##
+  # Convert a constant name to a path, assuming a conventional structure.
+  #
+  #   "FooBar::Baz".to_const_path # => "foo_bar/baz"
+  #
+  # @return [String] Path to the file containing the constant named by receiver
+  #   (constantized string), assuming a conventional structure.
+  #
+  # @api public
+  def to_const_path
+    snake_case.gsub(/::/, "/")
+  end
+end
+
+class Object
+  module InstanceExecHelper; end
+  include InstanceExecHelper
+  def instance_exec(*args, &block)
+    begin
+      old_critical, Thread.critical = Thread.critical, true
+      n = 0
+      n += 1 while respond_to?(mname="__instance_exec#{n}")
+      InstanceExecHelper.module_eval{ define_method(mname, &block) }
+    ensure
+      Thread.critical = old_critical
+    end
+    begin
+      ret = send(mname, *args)
+    ensure
+      InstanceExecHelper.module_eval{ remove_method(mname) } rescue nil
+    end
+    ret
+  end
+end

data/spec/actor_registry_spec.rb

@@ -0,0 +1,62 @@
+require File.join(File.dirname(__FILE__), 'spec_helper')
+
+describe Nanite::ActorRegistry do
+  
+  before(:all) do
+    class WebDocumentImporter
+      include Nanite::Actor
+      expose :import, :cancel
+
+      def import
+        1
+      end
+      def cancel
+        0
+      end
+    end
+
+    module Actors
+      class ComedyActor
+        include Nanite::Actor
+        expose :fun_tricks
+        def fun_tricks
+          :rabbit_in_the_hat
+        end
+      end
+    end
+  end
+  
+  before(:each) do
+    Nanite::Log.stub!(:info)
+    @registry = Nanite::ActorRegistry.new
+  end
+
+  it "should know about all services" do
+    @registry.register(WebDocumentImporter.new, nil)
+    @registry.register(Actors::ComedyActor.new, nil)
+    @registry.services.sort.should == ["/actors/comedy_actor/fun_tricks", "/web_document_importer/cancel", "/web_document_importer/import"]
+  end
+
+  it "should not register anything except Nanite::Actor" do
+    lambda { @registry.register(String.new, nil) }.should raise_error(ArgumentError)
+  end
+
+  it "should register an actor" do
+    importer = WebDocumentImporter.new
+    @registry.register(importer, nil)
+    @registry.actors['web_document_importer'].should == importer
+  end
+
+  it "should log info message that actor was registered" do
+    importer = WebDocumentImporter.new
+    Nanite::Log.should_receive(:info).with("Registering #{importer.inspect} with prefix nil")
+    @registry.register(importer, nil)
+  end
+
+  it "should handle actors registered with a custom prefix" do
+    importer = WebDocumentImporter.new
+    @registry.register(importer, 'monkey')
+    @registry.actors['monkey'].should == importer
+  end
+  
+end # Nanite::ActorRegistry

data/spec/actor_spec.rb

@@ -0,0 +1,59 @@
+require File.join(File.dirname(__FILE__), 'spec_helper')
+
+class WebDocumentImporter
+  include Nanite::Actor
+  expose :import, :cancel
+
+  def import
+    1
+  end
+  def cancel
+    0
+  end
+  def continue
+    1
+  end
+end
+
+module Actors
+  class ComedyActor
+    include Nanite::Actor
+    expose :fun_tricks
+    def fun_tricks
+      :rabbit_in_the_hat
+    end
+  end
+end
+
+describe Nanite::Actor do
+  
+  describe ".expose" do
+    it "should single expose method only once" do
+      3.times { WebDocumentImporter.expose(:continue) }
+      WebDocumentImporter.provides_for("webfiles").should == ["/webfiles/import", "/webfiles/cancel", "/webfiles/continue"]
+    end
+  end
+  
+  describe ".default_prefix" do
+    it "is calculated as default prefix as const path of class name" do
+      Actors::ComedyActor.default_prefix.should == "actors/comedy_actor"
+      WebDocumentImporter.default_prefix.should == "web_document_importer"
+    end
+  end
+
+  describe ".provides_for(prefix)" do
+    before :each do
+      @provides = Actors::ComedyActor.provides_for("money")
+    end
+    it "returns an array" do
+      @provides.should be_kind_of(Array)
+    end
+
+    it "maps exposed service methods to prefix" do
+      @provides.should == ["/money/fun_tricks"]
+      wdi_provides = WebDocumentImporter.provides_for("webfiles")
+      wdi_provides.should include("/webfiles/import")
+      wdi_provides.should include("/webfiles/cancel")
+    end
+  end
+end

data/spec/agent_spec.rb

@@ -0,0 +1,235 @@
+require File.join(File.dirname(__FILE__), 'spec_helper')
+
+describe "Agent:" do
+
+  describe "Default Option" do
+
+    before(:all) do
+      EM.stub!(:add_periodic_timer)
+      AMQP.stub!(:connect)
+      @amq = mock("AMQueue", :queue => mock("queue", :subscribe => {}), :fanout => mock("fanout", :publish => nil))
+      MQ.stub!(:new).and_return(@amq)
+      @agent = Nanite::Agent.start
+    end
+
+    it "for daemonize is false" do
+      @agent.options.should include(:daemonize)
+      @agent.options[:daemonize].should == false
+    end
+
+    it "for format is marshal" do
+      @agent.options.should include(:format)
+      @agent.options[:format].should == :marshal
+    end
+
+    it "for console is false" do
+      @agent.options.should include(:console)
+      @agent.options[:console].should == false
+    end
+
+    it "for user is nanite" do
+      @agent.options.should include(:user)
+      @agent.options[:user].should == "nanite"
+    end
+
+    it "for pass(word) is testing" do
+      @agent.options.should include(:pass)
+      @agent.options[:pass].should == "testing"
+    end
+
+    it "for secure is false" do
+      @agent.options.should include(:secure)
+      @agent.options[:secure].should == false
+    end
+
+    it "for host is 0.0.0.0" do
+      @agent.options.should include(:host)
+      @agent.options[:host].should == "0.0.0.0"
+    end
+
+    it "for log_level is info" do
+      @agent.options.should include(:log_level)
+      @agent.options[:log_level].should == :info
+    end
+
+    it "for vhost is /nanite" do
+      @agent.options.should include(:vhost)
+      @agent.options[:vhost].should == "/nanite"
+    end
+
+    it "for ping_time is 15" do
+      @agent.options.should include(:ping_time)
+      @agent.options[:ping_time].should == 15
+    end
+
+    it "for default_services is []" do
+      @agent.options.should include(:default_services)
+      @agent.options[:default_services].should == []
+    end
+
+    it "for root is #{File.expand_path(File.join(File.dirname(__FILE__), '..'))}" do
+      @agent.options.should include(:root)
+      @agent.options[:root].should == File.expand_path(File.join(File.dirname(__FILE__), '..'))
+    end
+
+    it "for file_root is #{File.expand_path(File.join(File.dirname(__FILE__), '..', 'files'))}" do
+      @agent.options.should include(:file_root)
+      @agent.options[:file_root].should == File.expand_path(File.join(File.dirname(__FILE__), '..', 'files'))
+    end
+
+  end
+
+  describe "Options from config.yml" do
+
+    before(:all) do
+      @agent = Nanite::Agent.start
+    end
+
+  end
+
+  describe "Passed in Options" do
+
+    before(:each) do
+      EM.stub!(:add_periodic_timer)
+      AMQP.stub!(:connect)
+      @amq = mock("AMQueue", :queue => mock("queue", :subscribe => {}), :fanout => mock("fanout", :publish => nil))
+      MQ.stub!(:new).and_return(@amq)
+    end
+
+    # TODO figure out how to stub call to daemonize
+    # it "for daemonize should override default (false)" do
+    #   agent = Nanite::Agent.start(:daemonize => true)
+    #   agent.options.should include(:daemonize)
+    #   agent.options[:daemonize].should == true
+    # end
+
+    it "for format should override default (marshal)" do
+      agent = Nanite::Agent.start(:format => :json)
+      agent.options.should include(:format)
+      agent.options[:format].should == :json
+    end
+
+    # TODO figure out how to avoid console output
+    # it "for console should override default (false)" do
+    #   agent = Nanite::Agent.start(:console => true)
+    #   agent.options.should include(:console)
+    #   agent.options[:console].should == true
+    # end
+
+    it "for user should override default (nanite)" do
+      agent = Nanite::Agent.start(:user => "me")
+      agent.options.should include(:user)
+      agent.options[:user].should == "me"
+    end
+
+    it "for pass(word) should override default (testing)" do
+      agent = Nanite::Agent.start(:pass => "secret")
+      agent.options.should include(:pass)
+      agent.options[:pass].should == "secret"
+    end
+
+    it "for secure should override default (false)" do
+      agent = Nanite::Agent.start(:secure => true)
+      agent.options.should include(:secure)
+      agent.options[:secure].should == true
+    end
+
+    it "for host should override default (0.0.0.0)" do
+      agent = Nanite::Agent.start(:host => "127.0.0.1")
+      agent.options.should include(:host)
+      agent.options[:host].should == "127.0.0.1"
+    end
+
+    it "for log_level should override default (info)" do
+      agent = Nanite::Agent.start(:log_level => :debug)
+      agent.options.should include(:log_level)
+      agent.options[:log_level].should == :debug
+    end
+
+    it "for vhost should override default (/nanite)" do
+      agent = Nanite::Agent.start(:vhost => "/virtual_host")
+      agent.options.should include(:vhost)
+      agent.options[:vhost].should == "/virtual_host"
+    end
+
+    it "for ping_time should override default (15)" do
+      agent = Nanite::Agent.start(:ping_time => 5)
+      agent.options.should include(:ping_time)
+      agent.options[:ping_time].should == 5
+    end
+
+    it "for default_services should override default ([])" do
+      agent = Nanite::Agent.start(:default_services => [:test])
+      agent.options.should include(:default_services)
+      agent.options[:default_services].should == [:test]
+    end
+
+    it "for root should override default (#{File.expand_path(File.join(File.dirname(__FILE__), '..'))})" do
+      agent = Nanite::Agent.start(:root => File.expand_path(File.dirname(__FILE__)))
+      agent.options.should include(:root)
+      agent.options[:root].should == File.expand_path(File.dirname(__FILE__))
+    end
+
+    it "for file_root should override default (#{File.expand_path(File.join(File.dirname(__FILE__), '..', 'files'))})" do
+      agent = Nanite::Agent.start(:file_root => File.expand_path(File.dirname(__FILE__)))
+      agent.options.should include(:file_root)
+      agent.options[:file_root].should == File.expand_path(File.dirname(__FILE__))
+    end
+
+    it "for a single tag should result in the agent's tags being set" do
+      agent = Nanite::Agent.start(:tag => "sample_tag")
+      agent.tags.should include("sample_tag")
+    end
+
+    it "for multiple tags should result in the agent's tags being set" do
+      agent = Nanite::Agent.start(:tag => ["sample_tag_1", "sample_tag_2"])
+      agent.tags.should include("sample_tag_1")
+      agent.tags.should include("sample_tag_2")
+    end
+
+  end
+  
+  describe "Security" do
+
+    before(:each) do
+      EM.stub!(:add_periodic_timer)
+      AMQP.stub!(:connect)
+      @amq = mock("AMQueue", :queue => mock("queue", :subscribe => {}, :publish => {}), :fanout => mock("fanout", :publish => nil))
+      MQ.stub!(:new).and_return(@amq)
+      serializer = Nanite::Serializer.new
+      @request = Nanite::Request.new('/foo/bar', '')
+      @push = Nanite::Push.new('/foo/bar', '')
+      @agent = Nanite::Agent.start
+    end
+    
+    it 'should correctly deny requests' do
+      security = mock("Security")
+      @agent.register_security(security)
+      
+      security.should_receive(:authorize).twice.and_return(false)
+      @agent.dispatcher.should_not_receive(:dispatch)
+      @agent.__send__(:receive, @request)
+      @agent.__send__(:receive, @push)
+    end
+
+    it 'should correctly authorize requests' do
+      security = mock("Security")
+      @agent.register_security(security)
+      
+      security.should_receive(:authorize).twice.and_return(true)
+      @agent.dispatcher.stub!(:dispatch)
+      @agent.dispatcher.should_receive(:dispatch).twice
+      @agent.__send__(:receive, @request)
+      @agent.__send__(:receive, @push)
+    end
+
+    it 'should be ignored when not specified' do
+      @agent.dispatcher.stub!(:dispatch)
+      @agent.dispatcher.should_receive(:dispatch).twice
+      @agent.__send__(:receive, @request)
+      @agent.__send__(:receive, @push)
+    end    
+
+  end
+
+end

data/spec/cached_certificate_store_proxy_spec.rb

@@ -0,0 +1,34 @@
+require File.join(File.dirname(__FILE__), 'spec_helper')
+
+describe Nanite::CachedCertificateStoreProxy do
+  
+  include SpecHelpers
+
+  before(:all) do
+    @signer, key = issue_cert
+    @recipient, key = issue_cert
+    @store = mock("Store")
+    @proxy = Nanite::CachedCertificateStoreProxy.new(@store)
+  end
+
+  it 'should not raise and return nil for non existent certificates' do
+    res = nil
+    @store.should_receive(:get_recipients).with(nil).and_return(nil)
+    lambda { res = @proxy.get_recipients(nil) }.should_not raise_error
+    res.should == nil
+    @store.should_receive(:get_signer).with(nil).and_return(nil)
+    lambda { res = @proxy.get_signer(nil) }.should_not raise_error
+    res.should == nil
+  end
+
+  it 'should return recipient certificates' do
+    @store.should_receive(:get_recipients).with('anything').and_return(@recipient)
+    @proxy.get_recipients('anything').should == @recipient
+  end
+  
+  it 'should return signer certificates' do
+    @store.should_receive(:get_signer).with('anything').and_return(@signer)
+    @proxy.get_signer('anything').should == @signer
+  end
+  
+end

data/spec/certificate_cache_spec.rb

@@ -0,0 +1,49 @@
+require File.join(File.dirname(__FILE__), 'spec_helper')
+
+describe Nanite::CertificateCache do
+
+  before(:each) do
+    @cache = Nanite::CertificateCache.new(2)
+  end
+
+  it 'should allow storing and retrieving objects' do
+    @cache['some_id'].should be_nil
+    @cache['some_id'] = 'some_value'
+    @cache['some_id'].should == 'some_value'
+  end
+
+  it 'should not store more than required' do
+    @cache[1] = 'oldest'
+    @cache[2] = 'older'
+    @cache[1].should == 'oldest'
+    @cache[2].should == 'older'
+  
+    @cache[3] = 'new'
+    @cache[3].should == 'new'
+
+    @cache[1].should be_nil
+    @cache[2].should == 'older'
+  end
+
+  it 'should use LRU to remove entries' do
+    @cache[1] = 'oldest'
+    @cache[2] = 'older'
+    @cache[1].should == 'oldest'
+    @cache[2].should == 'older'
+  
+    @cache[1] = 'new'
+    @cache[3] = 'newer'
+    @cache[1].should == 'new'
+    @cache[3].should == 'newer'
+
+    @cache[2].should be_nil
+  end
+
+  it 'should store items returned by block' do
+    @cache[1].should be_nil
+    item = @cache.get(1) { 'item' }
+    item.should == 'item'
+    @cache[1].should == 'item'
+  end
+
+end