metasploit-model 0.24.1.pre.semantic.pre.versioning.pre.2.pre.0

data/spec/lib/metasploit/model/nilify_blanks_spec.rb

@@ -0,0 +1,156 @@
+require 'spec_helper'
+
+describe Metasploit::Model::NilifyBlanks do
+  let(:base_class) do
+    # capture for class_eval scope
+    described_class = self.described_class
+
+    Class.new do
+      extend ActiveModel::Callbacks
+      include ActiveModel::Validations
+      include ActiveModel::Validations::Callbacks
+
+      include described_class
+    end
+  end
+
+  context 'included' do
+    let(:base_class) do
+      Class.new do
+        extend ActiveModel::Callbacks
+        include ActiveModel::Validations
+        include ActiveModel::Validations::Callbacks
+      end
+    end
+
+    it 'should register #nilify_blanks as a before validation callback' do
+      base_class.should_receive(:before_validation).with(:nilify_blanks)
+
+      # capture for class_eval scope
+      described_class = self.described_class
+
+      base_class.class_eval do
+        include described_class
+      end
+    end
+  end
+
+  context 'nilify_blank' do
+    it 'should support adding multiple attributes' do
+      attributes = [:a, :b]
+
+      base_class.nilify_blank *attributes
+
+      attributes.each do |attribute|
+        base_class.nilify_blank_attribute_set.should include(attribute)
+      end
+    end
+
+    it 'should not add duplicate attributes' do
+      attribute = :a
+
+      base_class.nilify_blank attribute
+      base_class.nilify_blank attribute
+
+      base_class.nilify_blank_attribute_set.length.should == 1
+    end
+  end
+
+  context '#nilify_blanks' do
+    subject(:nilify_blanks) do
+      base_instance.nilify_blanks
+    end
+
+    let(:base_instance) do
+      base_class.new
+    end
+
+    let(:value) do
+      'value'
+    end
+
+    before(:each) do
+      base_class.class_eval do
+        #
+        # Attributes
+        #
+
+        # @!attribute [rw] blank
+        #   @return [String, nil]
+        attr_accessor :blank
+
+        #
+        # Callbacks
+        #
+
+        nilify_blank :blank
+      end
+
+      base_instance.blank = value
+    end
+
+    it 'should check if value responds to blank?' do
+      value.should_receive(:respond_to?).with(:blank?)
+
+      nilify_blanks
+    end
+
+    context 'with value responds to blank?' do
+      it 'should call blank?' do
+        value.should_receive(:blank?)
+
+        nilify_blanks
+      end
+
+      context 'with blank' do
+        let(:value) do
+          ''
+        end
+
+        it 'should set attribute to nil' do
+          nilify_blanks
+
+          base_instance.blank.should be_nil
+        end
+      end
+
+      context 'without blank' do
+        let(:value) do
+          'value'
+        end
+
+        it 'should not change attribute' do
+          expect {
+            nilify_blanks
+          }.to_not change(base_instance, :blank)
+        end
+      end
+    end
+
+    context 'without value responds to blank?' do
+      let(:value) do
+        double('Value')
+      end
+
+      before(:each) do
+        value.stub(:respond_to?).with(:blank?).and_return(false)
+      end
+
+      it 'should not call blank?' do
+        value.should_not_receive(:blank?)
+
+        nilify_blanks
+      end
+    end
+  end
+
+  context 'nilify_blank_attribute_set' do
+    subject(:nilify_blank_attribute_set) do
+      base_class.nilify_blank_attribute_set
+    end
+
+    it 'should default to an empty Set' do
+      nilify_blank_attribute_set.should == Set.new
+    end
+  end
+end

data/spec/lib/metasploit/model/platform_spec.rb

@@ -0,0 +1,100 @@
+require 'spec_helper'
+
+describe Metasploit::Model::Platform,
+         # setting the metadata type makes rspec-rails include RSpec::Rails::ModelExampleGroup, which includes a better
+         # be_valid matcher that will print full error messages
+         type: :model do
+  it_should_behave_like 'Metasploit::Model::Platform',
+                        namespace_name: 'Dummy' do
+    def attribute_type(attribute)
+      type_by_attribute = {
+          fully_qualified_name: :string
+      }
+
+      type_by_attribute.fetch(attribute)
+    end
+  end
+
+  # @note Not tested in 'Metasploit::Model::Platform' shared example because it is a module method and not a class
+  #   method because seeding should always refer back to {Metasploit::Model::Platform} and not the classes in which
+  #   it is included.
+  context 'fully_qualified_names' do
+    subject(:fully_qualified_name_set) do
+      described_class.fully_qualified_name_set
+    end
+
+    it { should include 'AIX' }
+    it { should include 'Android' }
+    it { should include 'BSD' }
+    it { should include 'BSDi' }
+    it { should include 'Cisco' }
+    it { should include 'Firefox' }
+    it { should include 'FreeBSD' }
+    it { should include 'HPUX' }
+    it { should include 'IRIX' }
+    it { should include 'Java' }
+    it { should include 'Javascript' }
+    it { should include 'NetBSD' }
+    it { should include 'Netware' }
+    it { should include 'NodeJS' }
+    it { should include 'OpenBSD' }
+    it { should include 'OSX' }
+    it { should include 'PHP' }
+    it { should include 'Python' }
+    it { should include 'Ruby' }
+
+    it { should include 'Solaris'}
+    it { should include 'Solaris 4' }
+    it { should include 'Solaris 5' }
+    it { should include 'Solaris 6' }
+    it { should include 'Solaris 7' }
+    it { should include 'Solaris 8' }
+    it { should include 'Solaris 9' }
+    it { should include 'Solaris 10' }
+
+    it { should include 'Windows' }
+
+    it { should include 'Windows 95' }
+
+    it { should include 'Windows 98' }
+    it { should include 'Windows 98 FE' }
+    it { should include 'Windows 98 SE' }
+
+    it { should include 'Windows ME' }
+
+    it { should include 'Windows NT' }
+    it { should include 'Windows NT SP0' }
+    it { should include 'Windows NT SP1' }
+    it { should include 'Windows NT SP2' }
+    it { should include 'Windows NT SP3' }
+    it { should include 'Windows NT SP4' }
+    it { should include 'Windows NT SP5' }
+    it { should include 'Windows NT SP6' }
+    it { should include 'Windows NT SP6a' }
+
+    it { should include 'Windows 2000' }
+    it { should include 'Windows 2000 SP0' }
+    it { should include 'Windows 2000 SP1' }
+    it { should include 'Windows 2000 SP2' }
+    it { should include 'Windows 2000 SP3' }
+    it { should include 'Windows 2000 SP4' }
+
+    it { should include 'Windows XP' }
+    it { should include 'Windows XP SP0' }
+    it { should include 'Windows XP SP1' }
+    it { should include 'Windows XP SP2' }
+    it { should include 'Windows XP SP3' }
+
+    it { should include 'Windows 2003' }
+    it { should include 'Windows 2003 SP0' }
+    it { should include 'Windows 2003 SP1' }
+
+    it { should include 'Windows Vista' }
+    it { should include 'Windows Vista SP0' }
+    it { should include 'Windows Vista SP1' }
+
+    it { should include 'Windows 7' }
+
+    it { should include 'UNIX' }
+  end
+end

data/spec/lib/metasploit/model/reference_spec.rb

@@ -0,0 +1,21 @@
+require 'spec_helper'
+
+describe Metasploit::Model::Reference do
+  it_should_behave_like 'Metasploit::Model::Reference',
+                        namespace_name: 'Dummy' do
+    def attribute_type(attribute)
+      type_by_attribute = {
+          :designation => :string,
+          :url => :text
+      }
+
+      type = type_by_attribute.fetch(attribute)
+
+      type
+    end
+
+    def authority_with_abbreviation(abbreviation)
+      Dummy::Authority.with_abbreviation(abbreviation)
+    end
+  end
+end

data/spec/lib/metasploit/model/search/association_spec.rb

@@ -0,0 +1,33 @@
+require 'spec_helper'
+
+describe Metasploit::Model::Search::Association do
+  subject(:base_class) do
+    described_class = self.described_class
+
+    Class.new do
+      include described_class
+    end
+  end
+
+  context 'search_association' do
+    it 'should add Symbol to search_association_set' do
+      association_string = 'associated_things'
+
+      base_class.class_eval do
+        search_association association_string
+      end
+
+      base_class.search_association_set.should include(association_string.to_sym)
+    end
+  end
+
+  context 'search_association_set' do
+    let(:search_association_set) do
+      base_class.search_association_set
+    end
+
+    it 'should default to an empty Set' do
+      search_association_set.should == Set.new
+    end
+  end
+end

data/spec/lib/metasploit/model/search/attribute_spec.rb

@@ -0,0 +1,99 @@
+require 'spec_helper'
+
+describe Metasploit::Model::Search::Attribute do
+  subject(:base_class) do
+    described_class = self.described_class
+
+    Class.new do
+      include described_class
+    end
+  end
+
+  context 'search_attribute' do
+    context 'with attribute' do
+      let(:attribute) do
+        FactoryGirl.generate :metasploit_model_search_operator_attribute_attribute
+      end
+
+      context 'with :type' do
+        let(:type) do
+          FactoryGirl.generate :metasploit_model_search_operator_attribute_type
+        end
+
+        context 'operator' do
+          subject(:operator) do
+            base_class.search_attribute attribute, :type => type
+          end
+
+          it 'should call search_with' do
+            base_class.should_receive(:search_with).with(
+                Metasploit::Model::Search::Operator::Attribute,
+                hash_including(
+                    :attribute => attribute,
+                    :type => type
+                )
+            )
+
+            operator
+          end
+
+          it 'should be in search_attribute_operator_by_attribute' do
+            # grab operator first since it calls search_attribute and populates search_attribute_operator_by_attribute
+            cached = operator
+            base_class.search_with_operator_by_name[attribute].should == cached
+          end
+
+          context 'attribute' do
+            subject(:operator_attribute) do
+              operator.attribute
+            end
+
+            it 'should be the attribute passed to search_attribute' do
+              operator_attribute.should == attribute
+            end
+          end
+
+          context 'klass' do
+            subject(:klass) do
+              operator.klass
+            end
+
+            it 'should be class on which search_attribute was called' do
+              klass.should == base_class
+            end
+          end
+
+          context 'type' do
+            subject(:operator_type) do
+              operator.type
+            end
+
+            it 'should be type passed to search_attribute' do
+              operator_type.should == type
+            end
+          end
+        end
+      end
+
+      context 'without :type' do
+        it 'should raise Metasploit::Model::Invalid' do
+          expect {
+            base_class.search_attribute(attribute)
+          }.to raise_error(Metasploit::Model::Invalid)
+        end
+      end
+    end
+
+    context 'without attribute' do
+      let(:attribute) do
+        ''
+      end
+
+      it 'should raise Metasploit::Model::Invalid' do
+        expect {
+          base_class.search_attribute attribute, :type => :string
+        }.to raise_error(Metasploit::Model::Invalid)
+      end
+    end
+  end
+end

data/spec/lib/metasploit/model/search/operation/integer/value_spec.rb

@@ -0,0 +1,20 @@
+require 'spec_helper'
+
+describe Metasploit::Model::Search::Operation::Integer::Value do
+  it_should_behave_like 'Metasploit::Model::Search::Operation::Integer::Value' do
+    let(:operation_class) do
+      described_class = self.described_class
+
+      Class.new(Metasploit::Model::Base) do
+        include described_class
+
+        #
+        # Attributes
+        #
+
+        # @!attribute [rw] value
+        attr_reader :value
+      end
+    end
+  end
+end

data/spec/lib/metasploit/model/search/operation/string/value_spec.rb

@@ -0,0 +1,20 @@
+require 'spec_helper'
+
+describe Metasploit::Model::Search::Operation::String::Value do
+  it_should_behave_like 'Metasploit::Model::Search::Operation::String::Value' do
+    let(:operation_class) do
+      described_class = self.described_class
+
+      Class.new(Metasploit::Model::Base) do
+        include described_class
+
+        #
+        # Attributes
+        #
+
+        # @!attribute [rw] value
+        attr_reader :value
+      end
+    end
+  end
+end