ronin 0.2.2 → 0.2.3

data/spec/extensions/kernel_spec.rb

@@ -0,0 +1,19 @@
+require 'ronin/extensions/kernel'
+
+require 'spec_helper'
+
+describe Kernel do
+  it "should provide Kernel#try" do
+    Kernel.respond_to?('try').should == true
+  end
+
+  describe "try" do
+    it "should return the result of the block if nothing is raised" do
+      try { 2 + 2 }.should == 4
+    end
+
+    it "should return nil if an exception is raised" do
+      try { 2 + 'a' }.should be_nil
+    end
+  end
+end

data/spec/extensions/uri/query_params_spec.rb

@@ -35,4 +35,12 @@ describe URI::QueryParams do
     @uri.query = 'u=3'
     @uri.query_params['u'].should == '3'
   end
+
+  it "should properly escape query param values" do
+    @uri.query_params['x'] = '1&2'
+    @uri.query_params['y'] = 'one=two'
+    @uri.query_params['z'] = '?'
+
+    @uri.to_s.should == "http://www.test.com/page.php?x=1%262&y=one%3Dtwo&z=%3F"
+  end
 end

data/spec/formatting/binary/string_spec.rb

@@ -107,7 +107,7 @@ describe String do
     end
 
     it "should not contain the key used in the xor" do
-      @string.include?(@key).should_not == true
+      @string.include?(@key.chr).should_not == true
     end
 
     it "should not equal the original string" do

data/spec/formatting/digest/string_spec.rb

@@ -0,0 +1,84 @@
+require 'ronin/formatting/digest'
+
+require 'spec_helper'
+
+describe String do
+  before(:all) do
+    @string = "test"
+  end
+
+  it "should provide String#md5" do
+    @string.respond_to?('md5').should == true
+  end
+
+  it "should provide String#sha1" do
+    @string.respond_to?('sha1').should == true
+  end
+
+  it "should provide String#sha2" do
+    @string.respond_to?('sha2').should == true
+  end
+
+  it "should provide String#sha256" do
+    @string.respond_to?('sha256').should == true
+  end
+
+  it "should provide String#sha512" do
+    @string.respond_to?('sha512').should == true
+  end
+
+  describe "md5" do
+    before(:all) do
+      @digest_plain_text = "test"
+      @digest_md5 = "098f6bcd4621d373cade4e832627b4f6"
+    end
+
+    it "should return the MD5 digest of itself" do
+      @digest_plain_text.md5.should == @digest_md5
+    end
+  end
+
+  describe "sha1" do
+    before(:all) do
+      @digest_plain_text = "test"
+      @digest_sha1 = "a94a8fe5ccb19ba61c4c0873d391e987982fbbd3"
+    end
+
+    it "should return the SHA1 digest of itself" do
+      @digest_plain_text.sha1.should == @digest_sha1
+    end
+  end
+
+  describe "sha2" do
+    before(:all) do
+      @digest_plain_text = "test"
+      @digest_sha2 = "9f86d081884c7d659a2feaa0c55ad015a3bf4f1b2b0b822cd15d6c15b0f00a08"
+    end
+
+    it "should return the SHA2 digest of itself" do
+      @digest_plain_text.sha2.should == @digest_sha2
+    end
+  end
+
+  describe "sha256" do
+    before(:all) do
+      @digest_plain_text = "test"
+      @digest_sha256 = "9f86d081884c7d659a2feaa0c55ad015a3bf4f1b2b0b822cd15d6c15b0f00a08"
+    end
+
+    it "should return the SHA256 digest of itself" do
+      @digest_plain_text.sha256.should == @digest_sha256
+    end
+  end
+
+  describe "sha512" do
+    before(:all) do
+      @digest_plain_text = "test"
+      @digest_sha512 = "ee26b0dd4af7e749aa1a8ee3c10ae9923f618980772e473f8819a5d4940e0db27ac185f8a0e1d5f84f88bc887fd67b143732c304cc5fa9ad8e6f57f50028a8ff"
+    end
+
+    it "should return the SHA512 digest of itself" do
+      @digest_plain_text.sha512.should == @digest_sha512
+    end
+  end
+end

data/spec/formatting/http/string_spec.rb

@@ -0,0 +1,84 @@
+require 'ronin/formatting/http'
+
+require 'spec_helper'
+
+describe String do
+  before(:all) do
+    @string = "mod % 3"
+  end
+
+  it "should provide String#uri_encode" do
+    @string.respond_to?('uri_encode').should == true
+  end
+
+  it "should provide String#uri_decode" do
+    @string.respond_to?('uri_decode').should == true
+  end
+
+  it "should provide String#uri_escape" do
+    @string.respond_to?('uri_escape').should == true
+  end
+
+  it "should provide String#uri_unescape" do
+    @string.respond_to?('uri_unescape').should == true
+  end
+
+  it "should provide String#format_http" do
+    @string.respond_to?('format_http').should == true
+  end
+
+  describe "uri_encode" do
+    before(:all) do
+      @uri_unencoded = "mod % 3"
+      @uri_encoded = "mod%20%25%203"
+    end
+
+    it "should URI encode itself" do
+      @uri_unencoded.uri_encode.should == @uri_encoded
+    end
+  end
+
+  describe "uri_decode" do
+    before(:all) do
+      @uri_unencoded = "mod % 3"
+      @uri_encoded = "mod%20%25%203"
+    end
+
+    it "should URI decode itself" do
+      @uri_encoded.uri_decode.should == @uri_unencoded
+    end
+  end
+
+  describe "uri_escape" do
+    before(:all) do
+      @uri_unescaped = "x + y"
+      @uri_escaped = "x+%2B+y"
+    end
+
+    it "should URI escape itself" do
+      @uri_unescaped.uri_escape.should == @uri_escaped
+    end
+  end
+
+  describe "uri_unescape" do
+    before(:all) do
+      @uri_unescaped = "x + y"
+      @uri_escaped = "x+%2B+y"
+    end
+
+    it "should URI unescape itself" do
+      @uri_escaped.uri_unescape.should == @uri_unescaped
+    end
+  end
+
+  describe "format_http" do
+    before(:all) do
+      @uri_unencoded = "mod % 3"
+      @uri_http_encoded = "%6d%6f%64%20%25%20%33"
+    end
+
+    it "should format each byte of the String" do
+      @uri_unencoded.format_http.should == @uri_http_encoded
+    end
+  end
+end

data/spec/formatting/text/string_spec.rb

@@ -0,0 +1,51 @@
+require 'ronin/formatting/text'
+
+require 'spec_helper'
+
+describe String do
+  before(:all) do
+    @string = "hello"
+  end
+
+  it "should provide String#format_chars" do
+    @string.respond_to?('format_chars').should == true
+  end
+
+  it "should provide String#format_bytes" do
+    @string.respond_to?('format_bytes').should == true
+  end
+
+  it "should provide String#random_case" do
+    @string.respond_to?('random_case').should == true
+  end
+
+  describe "format_chars" do
+    it "should format each character in the String" do
+      @string.format_chars { |c|
+        "_#{c}"
+      }.should == "_h_e_l_l_o"
+    end
+  end
+
+  describe "format_bytes" do
+    it "should format each byte in the String" do
+      @string.format_bytes { |b|
+        sprintf("%%%x",b)
+      }.should == "%68%65%6c%6c%6f"
+    end
+  end
+
+  describe "random_case" do
+    it "should capitalize each character when :probability is 1.0" do
+      new_string = @string.random_case(:probability => 1.0)
+
+      @string.upcase.should == new_string
+    end
+
+    it "should not capitalize any characters when :probability is 0.0" do
+      new_string = @string.random_case(:probability => 0.0)
+
+      @string.should == new_string
+    end
+  end
+end

data/spec/has_license_spec.rb

@@ -0,0 +1,29 @@
+require 'ronin/has_license'
+
+require 'spec_helper'
+require 'classes/licensed_model'
+
+describe HasLicense do
+  before(:all) do
+    LicensedModel.auto_migrate!
+  end
+
+  it "should have a license" do
+    model = LicensedModel.new(:content => 'bla')
+    model.license = License.gpl_2
+    model.save!
+
+    model.license.should == License.gpl_2
+  end
+
+  it "should provide helper methods for querying licensed models" do
+    model = LicensedModel.new(:content => 'stuff here')
+    model.license = License.gpl_2
+    model.save!
+    model.reload
+
+    LicensedModel.all(
+      :content.like => '%stuff%'
+    ).licensed_under(:gpl_2).first.should == model
+  end
+end

data/spec/helpers/cacheable.rb

@@ -0,0 +1,7 @@
+require 'classes/cacheable_model'
+
+require 'tmpdir'
+require 'fileutils'
+
+CACHEABLE_FILE = File.expand_path(File.join(File.dirname(__FILE__),'contexts','ronin_cacheable_model.rb'))
+CACHEABLE_PATH = File.join(Dir.tmpdir,File.basename(CACHEABLE_FILE))

data/spec/helpers/contexts/ronin_cacheable_model.rb

@@ -0,0 +1,13 @@
+ronin_cacheable_model do
+
+  @config = true
+
+  cache do
+    self.content = 'this is a test'
+  end
+
+  def greeting
+    'hello'
+  end
+
+end

data/spec/helpers/database.rb

@@ -0,0 +1,5 @@
+require 'spec_helper'
+
+require 'ronin/database'
+
+Database.setup({ :adapter => 'sqlite3', :database => ':memory:' })

data/spec/platform/helpers/overlays/hello/lib/init.rb

@@ -0,0 +1 @@
+$hello_overlay_loaded = true

data/spec/platform/helpers/overlays/hello/lib/stuff/another_test.rb

@@ -0,0 +1,6 @@
+require 'stuff/test'
+
+module Stuff
+  class AnotherTest < Test
+  end
+end

data/spec/platform/helpers/overlays/hello/lib/stuff/test.rb

@@ -0,0 +1,4 @@
+module Stuff
+  class Test
+  end
+end

data/spec/platform/overlay_cache_spec.rb

@@ -53,7 +53,7 @@ describe Platform::OverlayCache do
   end
 
   it "should have extensions" do
-    @cache.extensions.should == ['test', 'hello', 'random']
+    @cache.extensions.should == ['hello', 'random', 'test']
   end
 
   it "should provide the paths to specific extensions" do

data/spec/platform/overlay_spec.rb

@@ -32,4 +32,32 @@ describe Platform::Overlay do
       @overlay.description.should == %{This is a test overlay used in Ronin's specs.}
     end
   end
+
+  describe "activate!" do
+    before(:all) do
+      @overlay = Platform::Overlay.new(File.join(OVERLAY_CACHE,'hello'))
+      @overlay.activate!
+    end
+
+    it "should load the init.rb file if present" do
+      $hello_overlay_loaded.should == true
+    end
+
+    it "should make the lib directory accessible to Kernel#require" do
+      require('stuff/test').should == true
+    end
+  end
+
+  describe "deactivate!" do
+    before(:all) do
+      @overlay = Platform::Overlay.new(File.join(OVERLAY_CACHE,'hello'))
+      @overlay.deactivate!
+    end
+
+    it "should make the lib directory unaccessible to Kernel#require" do
+      lambda {
+        require 'stuff/another_test'
+      }.should raise_error(LoadError)
+    end
+  end
 end

data/spec/product_spec.rb

@@ -13,4 +13,11 @@ describe Product do
     @product.version = '0.1.0'
     @product.should be_valid
   end
+
+  it "should default the vendor to the name of the product" do
+    @product = Product.new(:name => 'Adobe', :version => '0.1.0')
+    @product.vendor.should == @product.name
+
+    @product.should be_valid
+  end
 end

data/spec/sessions/classes/test_session.rb

@@ -0,0 +1,11 @@
+require 'ronin/sessions/session'
+
+module TestSession
+  include Sessions::Session
+
+  def connect
+    require_variable :host
+
+    return @host
+  end
+end

data/spec/sessions/classes/uses_test_session.rb

@@ -0,0 +1,10 @@
+require 'sessions/classes/test_session'
+
+class UsesTestSession
+
+  include TestSession
+
+  # The host variable
+  attr_accessor :host
+
+end

data/spec/sessions/session_spec.rb

@@ -1,56 +1,25 @@
 require 'ronin/sessions/session'
 
 require 'spec_helper'
+require 'sessions/classes/test_session'
+require 'sessions/classes/uses_test_session'
 
 describe Sessions::Session do
-  before(:all) do
-    module TestSession
-      include Sessions::Session
-
-      setup_session do
-        parameter :var, :default => :stuff, :description => 'Test parameter'
-      end
-
-      def test_one
-        'this_is_a_test'
-      end
-    end
-
-    class TestSessionClass
-      include TestSession
-    end
-
-    class TestSessionObject
-    end
-
-    @session_obj = TestSessionObject.new
-    @session_obj.extend TestSession
-  end
-
-  it "should add self.setup_session to a Module" do
-    TestSession.methods.include?('setup_session').should == true
-  end
-
-  describe "setup_session" do
-    it "should add self.included and self.extended methods once setup_session is called" do
-      TestSession.methods.include?('included').should == true
-      TestSession.methods.include?('extended').should == true
+  describe "require_variable" do
+    before(:each) do
+      @obj = UsesTestSession.new
+      @obj.host = 'www.example.com'
     end
 
-    it "should run the setup_session proc once included" do
-      TestSessionClass.var.should == :stuff
+    it "should raise a VariableMissing exception if a variable is nil" do
+      lambda {
+        @obj.host = nil
+        @obj.connect
+      }.should raise_error(Sessions::VariableMissing)
     end
 
-    it "should run the setup_session proc once extended" do
-      @session_obj.var.should == :stuff
+    it "should do nothing if the variable is not nil" do
+      @obj.connect.should == @obj.host
     end
   end
-
-  it "should include Parameters once included" do
-    TestSessionClass.include?(Parameters).should == true
-  end
-
-  it "should extend Parameters once extended" do
-    @session_obj.kind_of?(Parameters).should == true
-  end
 end