ronin 1.0.0.pre3 → 1.0.0.pre4

data/spec/ip_address.rb

@@ -5,6 +5,8 @@ describe IPAddress do
   let(:example_domain) { 'www.example.com' }
   let(:example_ip) { '192.0.32.10' }
 
+  subject { IPAddress.new(:address => example_ip) }
+
   it "should require an address" do
     ip = IPAddress.new
 
@@ -23,6 +25,14 @@ describe IPAddress do
       ips[0].address.should == example_ip
     end
 
+    it "should associate the IP addresses with the original host name" do
+      ips = subject.lookup(example_domain)
+
+      ips.each do |ip|
+        ip.host_names[0].address.should == example_domain
+      end
+    end
+
     it "should return an empty Array for unknown domain names" do
       ips = subject.lookup(bad_domain)
       
@@ -34,13 +44,20 @@ describe IPAddress do
     let(:bad_ip) { '0.0.0.0' }
 
     it "should reverse lookup the host-name for an IP Address" do
-      ip = IPAddress.new(:address => example_ip)
-      host_names = ip.lookup!
+      host_names = subject.lookup!
       
       host_names.should_not be_empty
       host_names[0].address.should == example_domain
     end
 
+    it "should associate the host names with the original IP address" do
+      host_names = subject.lookup!
+
+      host_names.each do |host_name|
+        host_name.ip_addresses[0].address.should == subject
+      end
+    end
+
     it "should return an empty Array for unknown domain names" do
       ip = IPAddress.new(:address => bad_ip)
       host_names = ip.lookup!
@@ -49,7 +66,7 @@ describe IPAddress do
     end
   end
 
-  describe "version" do
+  describe "#version" do
     let(:ipv4) { IPAddress.new(:address => '192.168.1.1') }
     let(:ipv6) { IPAddress.new(:address => '::1') }
 

data/spec/model/cacheable_spec.rb

@@ -56,7 +56,7 @@ describe Model::Cacheable do
     end
   end
 
-  describe "cached" do
+  context "previously cached" do
     subject { CacheableModel.first }
 
     it "should have a cached_file resource" do

data/spec/model/has_description_spec.rb

@@ -14,7 +14,7 @@ describe Model::HasDescription do
     subject.properties.should be_named(:description)
   end
 
-  describe "description" do
+  describe "#description" do
     let(:resource) { DescribedModel.new }
 
     it "should allow the setting of the description" do

data/spec/os_spec.rb

@@ -2,6 +2,10 @@ require 'spec_helper'
 require 'ronin/os'
 
 describe OS do
+  subject do
+    OS.new(:name => 'Linux', :version => '2.6.11')
+  end
+
   it "should require a name" do
     os = OS.new
     os.should_not be_valid
@@ -17,4 +21,26 @@ describe OS do
   it "should provide methods for creating OSes with versions" do
     OS.linux('2.6.11').should be_valid
   end
+
+  describe "#to_s" do
+    it "should convert both the name and version" do
+      os = OS.new(:name => 'Linux', :version => '2.6.23')
+
+      os.to_s.should == 'Linux 2.6.23'
+    end
+
+    it "should convert just the name if there is no version" do
+      os = OS.new(:name => 'Linux')
+
+      os.to_s.should == os.name
+    end
+  end
+
+  it "should implicitly splat the name and version" do
+    os = OS.new(:name => 'Linux', :version => '2.6.23')
+    name, version = os
+
+    name.should == os.name
+    version.should == os.version
+  end
 end

data/spec/password_spec.rb

@@ -32,4 +32,8 @@ describe Password do
       digest.should == Digest::SHA1.hexdigest(password + salt)
     end
   end
+
+  it "should be convertable to a String" do
+    subject.to_s.should == password
+  end
 end

data/spec/port_spec.rb

@@ -0,0 +1,43 @@
+require 'spec_helper'
+require 'ronin/port'
+
+describe Port do
+  let(:protocol) { 'tcp' }
+  let(:number) { 80 }
+
+  before(:all) do
+    @port = Port.create(:protocol => protocol, :number => number)
+  end
+
+  it "should require a protocol" do
+    port = Port.new(:number => 1111)
+
+    port.should_not be_valid
+  end
+
+  it "should require a port number" do
+    port = Port.new(:protocol => 'tcp')
+
+    port.should_not be_valid
+  end
+
+  it "should only allow 'tcp' and 'udp' as protocols" do
+    port = Port.new(:protocol => 'foo', :number => 1111)
+
+    port.should_not be_valid
+  end
+
+  it "should require unique protocol/port-number combinations" do
+    port = Port.new(:protocol => protocol, :number => number)
+
+    port.should_not be_valid
+  end
+
+  it "should be convertable to an Integer" do
+    @port.to_i.should == number
+  end
+
+  it "should be convertable to a String" do
+    @port.to_s.should == "#{number}/#{protocol}"
+  end
+end

data/spec/repository_spec.rb

@@ -82,7 +82,7 @@ describe Repository do
     end
   end
 
-  describe "domain" do
+  describe "#domain" do
     it "should be considered local for 'localhost' domains" do
       hello = repository('hello')
 
@@ -98,7 +98,7 @@ describe Repository do
     end
   end
 
-  describe "initialize" do
+  describe "#initialize" do
     it "should default the 'name' property to the name of the Repository directory" do
       repo = subject.new(
         :path => File.join(Helpers::Repositories::DIR,'hello')
@@ -117,7 +117,7 @@ describe Repository do
     end
   end
 
-  describe "initialize_metadata" do
+  describe "#initialize_metadata" do
     subject { repository('hello') }
 
     it "should load the title" do
@@ -149,7 +149,7 @@ describe Repository do
     end
   end
 
-  describe "activate!" do
+  describe "#activate!" do
     subject { repository('hello') }
 
     before(:all) do
@@ -165,7 +165,7 @@ describe Repository do
     end
   end
 
-  describe "deactivate!" do
+  describe "#deactivate!" do
     subject { repository('hello') }
 
     before(:all) do
@@ -179,7 +179,7 @@ describe Repository do
     end
   end
 
-  describe "cache_paths" do
+  describe "#cache_paths" do
     subject { repository('test1') }
 
     it "should list the contents of the 'cache/' directory" do
@@ -195,7 +195,7 @@ describe Repository do
     end
   end
 
-  describe "cached_files" do
+  describe "#cached_files" do
     before(:all) do
       CacheableModel.auto_migrate!
     end
@@ -203,7 +203,7 @@ describe Repository do
     let(:test1) { repository('test1') }
     let(:test2) { repository('test2') }
 
-    describe "cache_files!" do
+    describe "#cache_files!" do
       before(:all) do
         test1.cache_files!
         test2.cache_files!
@@ -251,7 +251,7 @@ describe Repository do
       end
     end
 
-    describe "sync_cached_files!" do
+    describe "#sync_cached_files!" do
       before(:all) do
         test1.cache_files!
         test2.cache_files!
@@ -278,7 +278,7 @@ describe Repository do
       end
     end
 
-    describe "clean_cached_files!" do
+    describe "#clean_cached_files!" do
       before(:all) do
         test1.clean_cached_files!
       end

data/spec/service_spec.rb

@@ -0,0 +1,18 @@
+require 'spec_helper'
+require 'ronin/service'
+
+describe Service do
+  before(:all) { @service = Service.create(:name => 'Apache') }
+
+  it "should require a name" do
+    service = Service.new
+
+    service.should_not be_valid
+  end
+
+  it "should require a unique name" do
+    service = Service.new(:name => 'Apache')
+
+    service.should_not be_valid
+  end
+end

data/spec/software_spec.rb

@@ -13,11 +13,21 @@ describe Software do
     software.should be_valid
   end
 
-  it "should be convertable to a String" do
-    software = Software.new(:name => 'Test', :version => '0.1.0')
-    software.to_s.should == 'Test 0.1.0'
+  describe "#to_s" do
+    it "should be convertable to a String" do
+      software = Software.new(
+        :name => 'Test', 
+        :version => '0.1.0',
+        :vendor => {:name => 'TestCo'}
+      )
 
-    software.vendor = Vendor.new(:name => 'TestCo')
-    software.to_s.should == 'TestCo Test 0.1.0'
+      software.to_s.should == 'TestCo Test 0.1.0'
+    end
+
+    it "should ignore the missing vendor information" do
+      software = Software.new(:name => 'Test', :version => '0.1.0')
+
+      software.to_s.should == 'Test 0.1.0'
+    end
   end
 end

data/spec/url_query_param_spec.rb

@@ -0,0 +1,30 @@
+require 'spec_helper'
+require 'ronin/url_query_param'
+
+describe URLQueryParam do
+  it "should require a name" do
+    param = URLQueryParam.new
+
+    param.should_not be_valid
+  end
+
+  describe "#to_s" do
+    it "should dump a name and a value into a String" do
+      param = URLQueryParam.new(:name => 'foo', :value => 'bar')
+
+      param.to_s.should == "foo=bar"
+    end
+
+    it "should ignore empty or nil values" do
+      param = URLQueryParam.new(:name => 'foo')
+
+      param.to_s.should == "foo="
+    end
+
+    it "should escape special characters" do
+      param = URLQueryParam.new(:name => 'foo', :value => 'bar baz')
+
+      param.to_s.should == "foo=bar%20baz"
+    end
+  end
+end

data/spec/url_scheme_spec.rb

@@ -0,0 +1,18 @@
+require 'spec_helper'
+require 'ronin/url_scheme'
+
+describe URLScheme do
+  before(:all) { @scheme = URLScheme.create(:name => 'http') }
+
+  it "should require a name" do
+    scheme = URLScheme.new
+
+    scheme.should_not be_valid
+  end
+
+  it "should require a unique name" do
+    scheme = URLScheme.new(:name => 'http')
+
+    scheme.should_not be_valid
+  end
+end

data/spec/url_spec.rb

@@ -11,8 +11,8 @@ describe URL do
   let(:query_string) { 'q=1' }
   let(:fragment) { 'frag' }
 
-  let(:uri) do
-    URI::HTTPS.build(
+  before(:all) do
+    @uri = URI::HTTPS.build(
       :scheme => scheme,
       :host => host_name,
       :port => port,
@@ -20,12 +20,15 @@ describe URL do
       :query => query_string,
       :fragment => fragment
     )
-  end
-
-  it "should have a default path" do
-    url = URL.new
 
-    url.path.should == ''
+    @url = URL.create(
+      :scheme => {:name => scheme},
+      :host_name => {:address => host_name},
+      :port => {:number => port},
+      :path => path,
+      :query_string => query_string,
+      :fragment => fragment
+    )
   end
 
   it "should have a host String" do
@@ -35,77 +38,91 @@ describe URL do
   end
 
   it "should be convertable to a String" do
-    url = URL.new(
-      :scheme => {:name => scheme},
-      :host_name => {:address => host_name},
-      :port => {:number => port},
-      :path => path,
-      :query_string => query_string,
-      :fragment => fragment
-    )
+    @url.to_s.should == @uri.to_s
+  end
+
+  describe "[]" do
+    it "should query URLs using URIs" do
+      URL[@uri].should == @url
+    end
 
-    url.to_s.should == uri.to_s
+    it "should query URLs using Strings" do
+      URL[@uri.to_s].should == @url
+    end
+
+    it "should still treat Integer arguments as indexes" do
+      URL[0].should == @url
+    end
   end
 
   describe "from" do
-    before(:all) do
-      @url = URL.from(uri)
-    end
+    subject { URL.from(@uri) }
 
     it "should parse URL schemes" do
-      @url.scheme.should_not be_nil
-      @url.scheme.name.should == scheme
+      subject.scheme.should_not be_nil
+      subject.scheme.name.should == scheme
     end
 
     it "should parse host names" do
-      @url.host_name.address.should == host_name
+      subject.host_name.address.should == host_name
     end
 
     it "should parse port numbers" do
-      @url.port.number.should == port
+      subject.port.number.should == port
     end
 
     it "should parse paths" do
-      @url.path.should == path
+      subject.path.should == path
     end
 
     it "should parse query strings" do
-      @url.query_string.should == query_string
+      subject.query_string.should == query_string
     end
 
     it "should parse URL fragments" do
-      @url.fragment.should == fragment
+      subject.fragment.should == fragment
     end
-  end
 
-  describe "to_uri" do
-    before(:all) do
-      @url = URL.parse('https://www.example.com:8080/path?q=1#frag')
-      @uri = @url.to_uri
+    it "should normalize the paths of HTTP URIs" do
+      uri = URI('http://www.example.com')
+      url = URL.from(uri)
+
+      url.path.should == '/'
     end
+  end
+
+  describe "#to_uri" do
+    subject { @url.to_uri }
 
     it "should convert the scheme" do
-      @uri.scheme.should == scheme
+      subject.scheme.should == scheme
     end
     
     it "should convert the host name" do
-      @uri.host.should == host_name
+      subject.host.should == host_name
     end
 
     it "should convert the port number" do
-      @uri.port.should == port
+      subject.port.should == port
     end
 
     it "should convert the path" do
-      @uri.path.should == path
+      subject.path.should == path
     end
 
     it "should convert the query string" do
-      @uri.query.should == query_string
+      subject.query.should == query_string
+    end
+
+    it "should omit the query string if there are no query params" do
+      new_url = URL.parse('https://www.example.com:8080/path')
+      new_uri = new_url.to_uri
+
+      new_uri.query.should be_nil
     end
 
     it "should convert the fragment" do
-      @uri.fragment.should == fragment
+      subject.fragment.should == fragment
     end
   end
 end