raptor-io 0.0.1

data/spec/raptor-io/ruby/hash_spec.rb

@@ -0,0 +1,20 @@
+require 'spec_helper'
+
+describe Hash do
+  describe '#stringify' do
+    it 'returns a Hash with keys and values recursively converted to strings' do
+      {
+          test:         'blah',
+          another_hash: {
+              stuff: 'test'
+          }
+      }.stringify.should == {
+          'test'         => 'blah',
+          'another_hash' => {
+              'stuff' => 'test'
+          }
+      }
+
+    end
+  end
+end

data/spec/raptor-io/ruby/string_spec.rb

@@ -0,0 +1,20 @@
+require 'spec_helper'
+
+describe String do
+
+  describe '#binary?' do
+    context 'when the content is' do
+      context 'binary' do
+        it 'returns true' do
+          "\ff\ff\ff".binary?.should be_true
+        end
+      end
+      context 'text' do
+        it 'returns false' do
+          'test'.binary?.should be_false
+        end
+      end
+    end
+  end
+
+end

data/spec/raptor-io/socket/comm/local_spec.rb

@@ -0,0 +1,50 @@
+require 'spec_helper'
+
+require 'raptor-io/socket'
+
+describe RaptorIO::Socket::Comm::Local do
+  subject(:comm_local) { described_class.new }
+
+  context "with a server to connect to" do
+    let(:tcp_port) { 9999 }
+    let(:udp_port) { 9090 }
+    before do
+      @tcp_server = ::TCPServer.new(tcp_port)
+    end
+    after do
+      @tcp_server.close
+    end
+
+    subject(:comm) { described_class.new }
+    it_behaves_like "a comm"
+  end
+
+  describe '#resolve' do
+    it 'should resolve a hostname to an IP address' do
+      # OSX uses a link-local address on the loopback interface
+      # (fe80::1%lo0) to be 'localhost'
+      ['127.0.0.1', '::1', 'fe80::1%lo0'].should include(comm_local.resolve('localhost'))
+    end
+  end
+
+  describe "#support_ipv6?" do
+    # These are ghetto tests. It assumes the implementation for
+    # determining support.
+    it "should be true if we can create ipv6 sockets" do
+      socket_inst = double("sock")
+      socket_inst.stub(:close)
+
+      ::Socket.should_receive(:new).with(::Socket::AF_INET6, ::Socket::SOCK_DGRAM, ::Socket::IPPROTO_UDP).and_return(socket_inst)
+      comm_local.support_ipv6?.should be_true
+    end
+    it "should be false if ::Socket::AF_INET6 is not defined" do
+      ::Socket.should_receive(:const_defined?).and_return(false)
+      comm_local.support_ipv6?.should be_false
+    end
+    it "should be false if we cannot create ipv6 sockets" do
+      ::Socket.should_receive(:new).and_raise(RuntimeError)
+      comm_local.support_ipv6?.should be_false
+    end
+  end
+
+end

data/spec/raptor-io/socket/switch_board/route_spec.rb

@@ -0,0 +1,49 @@
+require 'spec_helper'
+
+require 'raptor-io/socket'
+
+describe RaptorIO::Socket::SwitchBoard::Route do
+
+  context "class methods" do
+    describe ".new" do
+      it "should accept an IPAddr for subnet and netmask" do
+        expect {
+          described_class.new(IPAddr.new("1.2.3.4"), IPAddr.new("255.255.0.0"), nil)
+        }.not_to raise_error
+      end
+      it "should accept a String for subnet and netmask" do
+        expect {
+          described_class.new("1.2.3.4", "255.255.0.0", nil)
+        }.not_to raise_error
+      end
+    end
+  end
+
+  context "instance methods" do
+    subject(:route) do
+      RaptorIO::Socket::SwitchBoard::Route.new("1.2.3.4", "255.255.255.0", nil)
+    end
+
+    describe "#==" do
+      it "should be equal if attributes are the same" do
+        RaptorIO::Socket::SwitchBoard::Route.new("1.2.3.4", "255.255.255.0", nil).should == route
+      end
+      it "should NOT be equal if any attributes are different" do
+        RaptorIO::Socket::SwitchBoard::Route.new("1.2.3.4", "255.0.0.0", nil).should_not == route
+        RaptorIO::Socket::SwitchBoard::Route.new("2.2.3.4", "255.255.255.0", nil).should_not == route
+      end
+    end
+
+    describe "#<=>" do
+      it "should compare the subnet" do
+        other = RaptorIO::Socket::SwitchBoard::Route.new("0.0.0.0", "255.255.0.0", nil)
+        (route.<=> other).should == 1
+        other = RaptorIO::Socket::SwitchBoard::Route.new("0.0.0.0", "255.255.255.0", nil)
+        (route.<=> other).should == 0
+        other = RaptorIO::Socket::SwitchBoard::Route.new("0.0.0.0", "255.255.255.255", nil)
+        (route.<=> other).should == -1
+      end
+    end
+  end
+
+end

data/spec/raptor-io/socket/switch_board_spec.rb

@@ -0,0 +1,87 @@
+require 'spec_helper'
+
+require 'raptor-io/socket'
+
+describe RaptorIO::Socket::SwitchBoard do
+  subject(:switch_board) do
+    described_class.new
+  end
+
+  it { should be_an Enumerable }
+
+  describe "#add_route" do
+    it "should add a route" do
+      switch_board.routes.should be_empty
+      switch_board.add_route("1.2.3.4", "255.255.255.0", nil)
+      switch_board.routes.length.should == 1
+    end
+    it "should accept IPAddrs" do
+      switch_board.routes.should be_empty
+      switch_board.add_route(IPAddr.new("1.2.3.4"), IPAddr.new("255.255.255.0"), nil)
+      switch_board.routes.length.should == 1
+    end
+  end
+
+  describe "#each" do
+    subject(:switch_board) do
+      sb = described_class.new
+      4.times do |i|
+        sb.add_route("#{i}.2.3.4", "255.255.255.0", nil)
+      end
+      sb
+    end
+
+    it "should return all routes" do
+      routes = []
+      switch_board.each do |r|
+        routes << r
+      end
+      routes.length.should == 4
+      routes.should == switch_board.routes
+    end
+  end
+
+  describe "#flush_routes" do
+    it "should empty the routes" do
+      switch_board.routes.should be_empty
+      switch_board.add_route("1.1.1.1", "255.255.255.0", nil)
+      switch_board.add_route("2.2.2.2", "255.255.255.0", nil)
+      switch_board.routes.should_not be_empty
+      switch_board.flush_routes
+      switch_board.routes.should be_empty
+    end
+  end
+
+  describe "#remove_route" do
+
+  end
+
+  describe "#best_comm" do
+    it "should return the comm for the most specific route" do
+      comm0 = double("more specific")
+      comm1 = double("less specific")
+      comm2 = double("even less specific")
+      comm3 = double("different net")
+
+      switch_board.add_route("1.1.1.0", "255.255.255.0", comm0)
+      switch_board.add_route("1.1.0.0", "255.255.0.0", comm1)
+      switch_board.add_route("1.0.0.0", "255.0.0.0", comm2)
+      switch_board.add_route("2.2.2.0", "255.255.255.255", comm3)
+
+      switch_board.best_comm("1.1.1.1").should == comm0
+    end
+
+    it "should fall back to less specific routes" do
+      comm0 = double("more specific")
+      comm1 = double("less specific")
+      comm2 = double("different net")
+
+      switch_board.add_route("1.1.1.0", "255.255.255.0", comm0)
+      switch_board.add_route("1.1.0.0", "255.255.0.0", comm1)
+
+      switch_board.best_comm("1.1.2.1").should == comm1
+
+    end
+  end
+
+end

data/spec/raptor-io/socket/tcp/ssl_spec.rb

@@ -0,0 +1,18 @@
+require 'spec_helper'
+require 'raptor-io/socket'
+
+describe RaptorIO::Socket::TCP::SSL do
+  let(:opts) do
+    {
+        ssl_version:     :TLSv1,
+        ssl_verify_mode: OpenSSL::SSL::VERIFY_NONE
+    }
+  end
+
+  #include_context 'with ssl server'
+
+  #it_behaves_like "a client socket"
+
+end
+
+

data/spec/raptor-io/socket/tcp_server/ssl_spec.rb

@@ -0,0 +1,59 @@
+=begin
+require 'spec_helper'
+require 'raptor-io/socket'
+
+describe RaptorIO::Socket::TCPServer::SSL do
+  include_context 'with tcp server'
+
+  let(:server_cert) { File.read(File.join(fixtures_path, 'raptor', 'socket', 'ssl_server.crt')) }
+  let(:server_key)  { File.read(File.join(fixtures_path, 'raptor', 'socket', 'ssl_server.key')) }
+  let(:server_context) do
+    OpenSSL::SSL::SSLContext.new.tap do |context|
+      context.cert = OpenSSL::X509::Certificate.new(server_cert)
+      context.key  = OpenSSL::PKey::RSA.new(server_key)
+    end
+  end
+
+  subject(:ssl_server) do
+    described_class.new(server_sock, context: server_context )
+  end
+
+  let(:data) { 'test'.force_encoding( 'binary' ) }
+
+  describe '#accept_nonblock' do
+    it 'returns a connected peer as a RaptorIO::Socket::TCP::SSL socket' do
+      ssl_server
+      ssl_client = OpenSSL::SSL::SSLSocket.new(client_sock)
+
+      Thread.new do
+        Thread.pass
+        #$stderr.puts "ssl_client connecting"
+        ssl_client.connect
+        #$stderr.puts "ssl_client CONNECTED"
+      end
+
+      ssl_peer = nil
+
+      begin
+        #$stderr.puts "ssl_server accepting"
+        ssl_peer = ssl_server.accept_nonblock
+        #$stderr.puts "ssl_server ACCEPTED"
+      rescue IO::WaitReadable
+        #$stderr.puts "ssl_server waiting for a client"
+        RaptorIO::Socket.select([ssl_server])
+        retry
+      end
+
+      #$stderr.puts("ssl_server accepted peer #{ssl_peer}")
+      ssl_peer.should be_kind_of RaptorIO::Socket::TCP::SSL
+
+      select( [], [ssl_peer] )
+      ssl_peer.write data
+
+      select( [ssl_client] )
+      ssl_client.read(data.size).should == data
+
+    end
+  end
+end
+=end

data/spec/raptor-io/socket/tcp_server_spec.rb

@@ -0,0 +1,19 @@
+require 'spec_helper'
+require 'raptor-io/socket'
+
+describe RaptorIO::Socket::TCPServer do
+  subject { described_class.new(io, opts) }
+  let(:opts) { {} }
+  let(:io) {
+    sio = StringIO.new
+    # Bind, listen, and connect all return 0 unless an exception is
+    # raised
+    sio.stub(:bind).and_return(0)
+    sio.stub(:listen).and_return(0)
+    sio.stub(:accept).and_return { StringIO.new }
+
+    sio
+  }
+
+  it_behaves_like 'a server socket'
+end

data/spec/raptor-io/socket/tcp_spec.rb

@@ -0,0 +1,14 @@
+require 'spec_helper'
+require 'raptor-io/socket'
+
+describe RaptorIO::Socket::TCP do
+
+  subject { described_class.new(io, opts) }
+
+  let(:io) { StringIO.new }
+  let(:opts) { {} }
+
+  it_behaves_like 'a client socket'
+
+  it { should respond_to(:to_ssl) }
+end

data/spec/raptor-io/socket_spec.rb

@@ -0,0 +1,16 @@
+require 'spec_helper'
+
+require 'raptor-io/socket'
+
+describe RaptorIO::Socket do
+  subject { described_class.new(io) }
+  let(:io) { StringIO.new }
+
+  it "should not swallow errors in method_missing" do
+    subject.should_not respond_to(:asdfjkl)
+    expect {
+      subject.asdfjkl
+    }.to raise_error(NoMethodError)
+  end
+end
+

data/spec/raptor-io/version_spec.rb

@@ -0,0 +1,10 @@
+require 'spec_helper'
+
+describe "RaptorIO::VERSION" do
+  it 'holds the version number of the library' do
+    splits = RaptorIO::VERSION.to_s.split( '.' )
+
+    splits.should be_any
+    splits.each { |number| number.should match /\d+/ }
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,56 @@
+
+require 'simplecov'
+SimpleCov.start
+
+require 'raptor-io'
+
+# This file was generated by the `rspec --init` command. Conventionally, all
+# specs live under a `spec` directory, which RSpec adds to the `$LOAD_PATH`.
+# Require this file using `require "spec_helper"` to ensure that it is only
+# loaded once.
+#
+
+# Requires supporting ruby files with custom matchers and macros, etc,
+# in spec/support/ and its subdirectories.
+Dir["#{File.dirname(__FILE__)}/support/**/*.rb"].each do |f|
+  next if f.include? '/webservers/'
+  require f
+end
+
+# See http://rubydoc.info/gems/rspec-core/RSpec/Core/Configuration
+RSpec.configure do |config|
+  config.treat_symbols_as_metadata_keys_with_true_values = true
+  config.run_all_when_everything_filtered = true
+  config.filter_run :focus
+
+  # Run specs in random order to surface order dependencies. If you find an
+  # order dependency and want to debug it, you can fix the order by providing
+  # the seed, which is printed after each run.
+  #     --seed 1234
+  config.order = 'random'
+
+  config.before( :all ) do
+    RaptorIO::Protocol::HTTP::Client.reset
+    RaptorIO::Protocol::HTTP::Request::Manipulators.reset
+    WebServers.killall
+  end
+
+  config.after( :all ) do
+    RaptorIO::Protocol::HTTP::Client.reset
+    RaptorIO::Protocol::HTTP::Request::Manipulators.reset
+  end
+
+  config.after( :suite ) do
+    RaptorIO::Protocol::HTTP::Client.reset
+    RaptorIO::Protocol::HTTP::Request::Manipulators.reset
+    WebServers.killall
+  end
+
+end
+
+EXAMPLE_PORT = 12345
+
+def example_addr; '127.0.0.1'; end
+def example_port; EXAMPLE_PORT; end
+def example_ssl_port; EXAMPLE_PORT + 1; end
+

data/spec/support/fixtures/raptor/protocol/http/request/manipulators/manifoolators/fooer.rb

@@ -0,0 +1,25 @@
+module RaptorIO
+module Protocol::HTTP
+class Request
+
+module Manipulators
+module Manifoolators
+
+#
+# Test manipulator, shows that manipulators can be namespaced.
+#
+# @author Tasos Laskos <tasos_laskos@rapid7.com>
+#
+class Fooer < Manipulator
+
+  def run
+    request.url += ('foo' * options[:times])
+  end
+
+end
+
+end
+end
+end
+end
+end