spec/payloads/payload_spec.rb in ronin-exploits-0.1.1 vs spec/payloads/payload_spec.rb in ronin-exploits-0.2.0

@@ -1,60 +1,95 @@
 require 'ronin/payloads/payload'
 
 require 'spec_helper'
+require 'helpers/objects'
 
 describe Payloads::Payload do
   before(:each) do
-    @payload = Payloads::Payload.new(:name => 'test') do
-      def builder
-        @payload = 'code'
-      end
-    end
+    @payload = load_payload('test')
   end
 
   it "should require a name attribute" do
-    payload = Payloads::Payload.new(:object_path => 'test.rb')
+    payload = Payloads::Payload.new
     payload.should_not be_valid
 
     payload.name = 'test'
     payload.should be_valid
   end
 
   it "should have a unique name and version" do
     first_payload = Payloads::Payload.create(
-      :object_path => 'test.rb',
       :name => 'test',
       :version => '0.0.1'
     )
     first_payload.should be_valid
 
     second_payload = Payloads::Payload.new(
-      :object_path => 'other.rb',
       :name => 'test',
       :version => '0.0.1'
     )
     second_payload.should_not be_valid
 
     third_payload = Payloads::Payload.new(
-      :object_path => 'other.rb',
       :name => 'test',
       :version => '0.0.2'
     )
     third_payload.should be_valid
   end
 
+  it "should not have any controls by default" do
+    @payload.controls.should be_empty
+  end
+
+  it "should specify what behaviors the payload controls" do
+    @payload.controlling :memory_read
+
+    @payload.behaviors.first.should == Vuln::Behavior[:memory_read]
+  end
+
+  it "should allow for the extending of Helper modules" do
+    @payload.instance_eval { helper :shell }.should == true
+  end
+
+  it "should raise an UnknownHelper when extending an unknown helper" do
+    lambda {
+      @payload.instance_eval { helper :obvious_not_there }
+    }.should raise_error(Payloads::UnknownHelper)
+  end
+
+  it "should not have an Arch by default" do
+    @payload.arch.should be_nil
+  end
+
+  it "should set the Arch when called with a name" do
+    @payload.arch :i686
+    @payload.arch.name.should == 'i686'
+    @payload.arch.endian == 'little'
+    @payload.arch.address_length == 4
+  end
+
+  it "should not have an OS by default" do
+    @payload.os.should be_nil
+  end
+
+  it "should set the OS when called with arguments" do
+    @payload.os(:name => 'FreeBSD', :version => '7.1')
+    @payload.os.name.should == 'FreeBSD'
+    @payload.os.version.should == '7.1'
+  end
+
   it "should have 'built' and 'unbiult' states" do
     @payload.should_not be_built
-    @payload.build
+    @payload.build!
     @payload.should be_built
   end
 
   it "should return the built payload when calling build" do
-    @payload.build.should == 'code'
+    @payload.build!.should == 'code'
   end
 
   it "should have a default deployer method" do
-    @payload.deploy do |payload|
-      @payload.should == payload
+    @payload.deploy! do |payload|
+      payload.should == @payload
     end
   end
 end