ronin 0.2.4 → 0.3.0

data/spec/license_spec.rb

@@ -4,14 +4,14 @@ require 'helpers/database'
 
 describe License do
   it "should require name and description attributes" do
-    @license = License.new
-    @license.should_not be_valid
+    license = License.new
+    license.should_not be_valid
 
-    @license.name = 'joke'
-    @license.should_not be_valid
+    license.name = 'joke'
+    license.should_not be_valid
 
-    @license.description = "yep, it's a joke."
-    @license.should be_valid
+    license.description = "yep, it's a joke."
+    license.should be_valid
   end
 
   it "should provide built-in licenses"do

data/spec/model/classes/basic_model.rb

@@ -0,0 +1,13 @@
+require 'ronin/model/model'
+
+class BasicModel
+
+  include Ronin::Model
+
+  property :id, Serial
+
+  property :name, String
+
+  property :age, Integer
+
+end

data/spec/model/classes/custom_model.rb

@@ -0,0 +1,17 @@
+require 'ronin/model/model'
+
+require 'model/classes/basic_model'
+
+class CustomModel < BasicModel
+
+  include Ronin::Model
+
+  attr_reader :var
+
+  def initialize(attributes={})
+    super(attributes)
+
+    @var = 2
+  end
+
+end

data/spec/model/classes/described_model.rb

@@ -0,0 +1,10 @@
+require 'ronin/model/has_description'
+
+class DescribedModel
+
+  include Ronin::Model
+  include Ronin::Model::HasDescription
+
+  property :id, Serial
+
+end

data/spec/model/has_description_spec.rb

@@ -0,0 +1,42 @@
+require 'ronin/model/has_description'
+
+require 'spec_helper'
+require 'model/classes/described_model'
+
+describe Model::HasDescription do
+  before(:each) do
+    @model = DescribedModel.new
+  end
+
+  it "should allow the setting of the description" do
+    @model.description = 'test one'
+    @model.description.should == 'test one'
+  end
+
+  it "should strip leading and tailing white-space" do
+    @model.description = %{
+test two
+    }
+
+    @model.description.should == 'test two'
+  end
+
+  it "should strip leading and tailing white-space from each line" do
+    @model.description = %{
+      test
+      three
+    }
+
+    @model.description.should == "test\nthree"
+  end
+
+  it "should preserve non-bordering empty lines" do
+    @model.description = %{
+      test
+
+      four
+    }
+
+    @model.description.should == "test\n\nfour"
+  end
+end

data/spec/model/has_license_spec.rb

@@ -8,22 +8,36 @@ describe Model::HasLicense do
     LicensedModel.auto_migrate!
   end
 
+  it "should define a license relationship" do
+    relationship = LicensedModel.relationships['license']
+
+    relationship.should_not be_nil
+    relationship.parent_model.should == License
+  end
+
+  it "should define relationships with License" do
+    relationship = License.relationships['licensed_models']
+    
+    relationship.should_not be_nil
+    relationship.child_model.should == LicensedModel
+  end
+
   it "should have a license" do
     model = LicensedModel.new(:content => 'bla')
-    model.license = License.gpl_2
+    model.license = License.gpl2
     model.save
 
-    model.license.should == License.gpl_2
+    model.license.should == License.gpl2
   end
 
   it "should provide helper methods for querying licensed models" do
     model = LicensedModel.new(:content => 'stuff here')
-    model.license = License.gpl_2
+    model.license = License.gpl2
     model.save
     model.reload
 
     LicensedModel.all(
       :content.like => '%stuff%'
-    ).licensed_under(:gpl_2).first.should == model
+    ).licensed_under(:gpl2).first.should == model
   end
 end

data/spec/model/model_spec.rb

@@ -0,0 +1,68 @@
+require 'ronin/model/model'
+
+require 'spec_helper'
+require 'model/classes/basic_model'
+require 'model/classes/custom_model'
+
+describe Model do
+  before(:all) do
+    BasicModel.auto_migrate!
+    CustomModel.auto_migrate!
+  end
+
+  it "should have a default repository name" do
+    BasicModel.default_repository_name.should == Model::REPOSITORY_NAME
+  end
+
+  it "should allow creating new instances of the model" do
+    resource = BasicModel.new(:name => 'joe')
+
+    resource.name.should == 'joe'
+  end
+
+  it "should call initialize when creating new instances of the model" do
+    resource = CustomModel.new(:name => 'joe')
+
+    resource.name.should == 'joe'
+    resource.var.should == 2
+  end
+
+  it "should still call initialize when loading from the database" do
+    CustomModel.create(:name => 'bob')
+
+    resource = CustomModel.first
+    resource.name.should == 'bob'
+    resource.var.should == 2
+  end
+
+  describe "humanize_attributes" do
+    before(:all) do
+      @resource = BasicModel.new(:name => 'joe', :age => 21)
+    end
+
+    it "should humanize the attributes of a model" do
+      @resource.humanize_attributes.should == {
+        'Name' => 'joe',
+        'Age' => '21'
+      }
+    end
+
+    it "should exclude certain attributes to humanize" do
+      @resource.humanize_attributes(:exclude => [:name]).should == {
+        'Age' => '21'
+      }
+    end
+
+    it "should filter out nil values" do
+      resource = BasicModel.new(:name => 'joe')
+
+      resource.humanize_attributes.should == {'Name' => 'joe'}
+    end
+
+    it "should filter out empty values" do
+      resource = BasicModel.new(:name => '', :age => 21)
+
+      resource.humanize_attributes.should == {'Age' => '21'}
+    end
+  end
+end

data/spec/network/helpers/classes/test_helper.rb

@@ -0,0 +1,11 @@
+require 'ronin/network/helpers/helper'
+
+module TestHelper
+  include Network::Helpers::Helper
+
+  def connect
+    require_variable :host
+
+    return @host
+  end
+end

data/spec/network/helpers/classes/uses_test_helper.rb

@@ -0,0 +1,10 @@
+require 'network/helpers/classes/test_helper'
+
+class UsesTestHelper
+
+  include TestHelper
+
+  # The host variable
+  attr_accessor :host
+
+end

data/spec/network/helpers/helper_spec.rb

@@ -0,0 +1,25 @@
+require 'ronin/network/helpers/helper'
+
+require 'spec_helper'
+require 'network/helpers/classes/test_helper'
+require 'network/helpers/classes/uses_test_helper'
+
+describe Network::Helpers::Helper do
+  describe "require_variable" do
+    before(:each) do
+      @obj = UsesTestHelper.new
+      @obj.host = 'www.example.com'
+    end
+
+    it "should raise a RuntimeError exception if a variable is nil" do
+      lambda {
+        @obj.host = nil
+        @obj.connect
+      }.should raise_error(RuntimeError)
+    end
+
+    it "should do nothing if the variable is not nil" do
+      @obj.connect.should == @obj.host
+    end
+  end
+end

data/spec/network/http/proxy_spec.rb

@@ -0,0 +1,116 @@
+require 'ronin/network/http/proxy'
+
+require 'spec_helper'
+
+describe Network::HTTP::Proxy do
+  before(:each) do
+    @proxy = Network::HTTP::Proxy.new
+  end
+
+  it "should parse host-names" do
+    proxy = Network::HTTP::Proxy.parse('127.0.0.1')
+
+    proxy.host.should == '127.0.0.1'
+  end
+
+  it "should parse 'host:port' URLs" do
+    proxy = Network::HTTP::Proxy.parse('127.0.0.1:80')
+
+    proxy.host.should == '127.0.0.1'
+    proxy.port.should == 80
+  end
+
+  it "should parse 'user@host:port' URLs" do
+    proxy = Network::HTTP::Proxy.parse('joe@127.0.0.1:80')
+
+    proxy.user.should == 'joe'
+    proxy.host.should == '127.0.0.1'
+    proxy.port.should == 80
+  end
+
+  it "should prase 'user:password@host:port' URLs" do
+    proxy = Network::HTTP::Proxy.parse('joe:lol@127.0.0.1:80')
+
+    proxy.user.should == 'joe'
+    proxy.password.should == 'lol'
+    proxy.host.should == '127.0.0.1'
+    proxy.port.should == 80
+  end
+
+  it "should ignore http:// prefixes when parsing proxy URLs" do
+    proxy = Network::HTTP::Proxy.parse('http://joe:lol@127.0.0.1:80')
+
+    proxy.user.should == 'joe'
+    proxy.password.should == 'lol'
+    proxy.host.should == '127.0.0.1'
+    proxy.port.should == 80
+  end
+
+  it "should behave like a Hash" do
+    @proxy[:host] = 'example.com'
+    @proxy[:host].should == 'example.com'
+  end
+
+  it "should not have a host by default" do
+    @proxy.host.should be_nil
+  end
+
+  it "should have a default port" do
+    @proxy.port.should == 8080
+  end
+
+  it "should be disabled by default" do
+    @proxy.should_not be_enabled
+  end
+
+  it "should reset the host, port, user and password when disabled" do
+    @proxy[:host] = 'example.com'
+    @proxy[:port] = 9001
+    @proxy[:user] = 'joe'
+    @proxy[:password] = 'lol'
+
+    @proxy.disable!
+
+    @proxy[:host].should be_nil
+    @proxy[:port].should == 8080
+    @proxy[:user].should be_nil
+    @proxy[:password].should be_nil
+  end
+
+  it "should convert host-names to Strings" do
+    @proxy.host = :"example.com"
+    @proxy.host.should == 'example.com'
+  end
+
+  it "should convert ports to Integers" do
+    @proxy.port = '8001'
+    @proxy.port.should == 8001
+  end
+
+  it "should return a URI::HTTP representing the proxy" do
+    @proxy[:host] = 'example.com'
+    @proxy[:port] = 9001
+    @proxy[:user] = 'joe'
+    @proxy[:password] = 'lol'
+
+    url = @proxy.url
+
+    url.host.should == 'example.com'
+    url.port.should == 9001
+    url.user.should == 'joe'
+    url.password.should == 'lol'
+  end
+
+  it "should return nil when converting a disabled proxy to a URL" do
+    @proxy.url.should be_nil
+  end
+
+  it "should return the host-name when converted to a String" do
+    @proxy[:host] = 'example.com'
+    @proxy.to_s.should == 'example.com'
+  end
+
+  it "should return an empty String when there is no host-name" do
+    @proxy.to_s.should be_empty
+  end
+end

data/spec/os_spec.rb

@@ -4,11 +4,11 @@ require 'helpers/database'
 
 describe OS do
   it "should require a name" do
-    @os = OS.new
-    @os.should_not be_valid
+    os = OS.new
+    os.should_not be_valid
 
-    @os.name = 'test'
-    @os.should be_valid
+    os.name = 'test'
+    os.should be_valid
   end
 
   it "should provide methods for built-in OSes" do

data/spec/platform/extension_cache_spec.rb

@@ -6,7 +6,9 @@ require 'spec_helper'
 describe Platform::ExtensionCache do
   before(:all) do
     Platform.load_overlays(overlay_cache_path)
+  end
 
+  before(:each) do
     @cache = Platform::ExtensionCache.new
   end
 
@@ -17,6 +19,18 @@ describe Platform::ExtensionCache do
     ext.name.should == 'test'
   end
 
+  it "should determine if an extension was loaded" do
+    @cache['test']
+    @cache.has?('test').should == true
+  end
+
+  it "should select extensions with specific attributes" do
+    test = @cache['test']
+    random = @cache['random']
+
+    @cache.with { |ext| ext.name == 'test' }.should == [test]
+  end
+
   it "should provide transparent caching of extensions" do
     ext = @cache['test']
     ext.should_not be_nil

data/spec/platform/extension_spec.rb

@@ -5,6 +5,9 @@ require 'spec_helper'
 describe Platform::Extension do
   before(:each) do
     @ext = Platform::Extension.new('test') do
+      attr_reader :var
+      attr_writer :var
+
       setup do
         @var = :setup
       end
@@ -59,4 +62,12 @@ describe Platform::Extension do
       ext.run_method.should == :running
     end
   end
+
+  it "should allow the definition of reader and writer methods" do
+    @ext.run do |ext|
+      ext.var.should == :setup
+      ext.var = :random
+      ext.var.should == :random
+    end
+  end
 end