metasploit_data_models 0.17.1 → 0.17.2.pre.metasploit.pre.data.pre.models.pre.search

data/spec/support/shared/examples/metasploit_data_models/search/visitor/includes/visit/with_metasploit_model_search_operation_base.rb

@@ -0,0 +1,26 @@
+shared_examples_for 'MetasploitDataModels::Search::Visitor::Includes#visit with Metasploit::Model::Search::Operation::Base' do
+  let(:operator) do
+    double('Operation Operator')
+  end
+
+  let(:node) do
+    node_class.new(
+        :operator => operator
+    )
+  end
+
+  it 'should visit operator' do
+    visitor.should_receive(:visit).with(node).and_call_original
+    visitor.should_receive(:visit).with(operator).and_return([])
+
+    visit
+  end
+
+  it 'should return operator visit' do
+    operator_visit = ["Visited Operator"]
+    visitor.should_receive(:visit).with(node).and_call_original
+    visitor.stub(:visit).with(operator).and_return(operator_visit)
+
+    visit.should == operator_visit
+  end
+end

data/spec/support/shared/examples/metasploit_data_models/search/visitor/relation/visit/matching_record.rb

@@ -0,0 +1,37 @@
+shared_examples_for 'MetasploitDataModels::Search::Visitor::Relation#visit matching record' do |options={}|
+  options.assert_valid_keys(:attribute, :association)
+
+  attribute = options.fetch(:attribute)
+  association = options[:association]
+
+  if association
+    formatted_operator = "#{association}.#{attribute}"
+  else
+    formatted_operator = attribute.to_s
+  end
+
+  context "with #{formatted_operator}" do
+    let(:formatted) do
+      "#{formatted_operator}:\"#{value}\""
+    end
+
+    if association
+      let(:associated) do
+        # wrap in array so single and plural associations can be handled the same.
+        Array.wrap(matching_record.send(association)).first
+      end
+
+      let(:value) do
+        associated.send(attribute)
+      end
+    else
+      let(:value) do
+        matching_record.send(attribute)
+      end
+    end
+
+    it 'should find only matching record' do
+      expect(visit).to match_array([matching_record])
+    end
+  end
+end

data/spec/support/shared/examples/metasploit_data_models/search/visitor/relation/visit/matching_record/with_metasploit_model_search_opeator_deprecated_platform.rb

@@ -0,0 +1,31 @@
+shared_examples_for 'MetasploitDataModels::Search::Visitor::Relation#visit matching record with Metasploit::Model::Search::Operator::Deprecated::Platform' do |options={}|
+  options.assert_valid_keys(:name)
+
+  name = options.fetch(:name)
+
+  context "with #{name}" do
+    let(:formatted) do
+      "#{name}:\"#{value}\""
+    end
+
+    context 'with Mdm::Platform#fully_qualified_name' do
+      let(:value) do
+        matching_record.platforms.sample.fully_qualified_name
+      end
+
+      it 'should find only matching record' do
+        expect(visit).to match_array([matching_record])
+      end
+    end
+
+    context 'with Mdm::Module::Target#name' do
+      let(:value) do
+        matching_record.targets.sample.name
+      end
+
+      it 'should find only matching record' do
+        expect(visit).to match_array([matching_record])
+      end
+    end
+  end
+end

data/spec/support/shared/examples/metasploit_data_models/search/visitor/relation/visit/matching_record/with_metasploit_model_search_operator_deprecated_authority.rb

@@ -0,0 +1,78 @@
+shared_examples_for 'MetasploitDataModels::Search::Visitor::Relation#visit matching record with Metasploit::Model::Search::Operator::Deprecated::Authority' do |options={}|
+  options.assert_valid_keys(:abbreviation)
+
+  abbreviation = options.fetch(:abbreviation)
+
+  context "with #{abbreviation}" do
+    #
+    # lets
+    #
+
+    let(:formatted) do
+      "#{abbreviation}:#{value}"
+    end
+
+    let(:matching_authority) do
+      FactoryGirl.create(
+          :mdm_authority,
+          :abbreviation => abbreviation
+      )
+    end
+
+    let(:matching_record) do
+      FactoryGirl.build(
+          :mdm_module_instance,
+          module_class: matching_module_class,
+          # disable factory making references automatically so Mdm::Module::Reference#reference can be set to
+          # matching_reference
+          module_references_length: 0
+      ).tap { |module_instance|
+        module_instance.module_references << FactoryGirl.build(
+            :mdm_module_reference,
+            module_instance: module_instance,
+            reference: matching_reference
+        )
+      }
+    end
+
+    let(:matching_reference) do
+      FactoryGirl.create(
+          :mdm_reference,
+          :authority => matching_authority
+      )
+    end
+
+    let(:non_matching_authority) do
+      FactoryGirl.create(:mdm_authority)
+    end
+
+    let(:non_matching_record) do
+      FactoryGirl.build(
+          :mdm_module_instance,
+          module_class: non_matching_module_class,
+          module_references_length: 0
+      ).tap { |module_instance|
+        module_instance.module_references << FactoryGirl.build(
+            :mdm_module_reference,
+            module_instance: module_instance,
+            reference: non_matching_reference
+        )
+      }
+    end
+
+    let(:non_matching_reference) do
+      FactoryGirl.create(
+          :mdm_reference,
+          :authority => non_matching_authority
+      )
+    end
+
+    let(:value) do
+      matching_reference.designation
+    end
+
+    it 'should find only matching record' do
+      expect(visit).to match_array([matching_record])
+    end
+  end
+end

data/spec/support/shared/examples/metasploit_data_models/search/visitor/where/visit/with_equality.rb

@@ -0,0 +1,34 @@
+shared_examples_for 'MetasploitDataModels::Search::Visitor::Where#visit with equality operation' do
+  let(:node) do
+    node_class.new(
+        :operator => operator,
+        :value => value
+    )
+  end
+
+  let(:operator) do
+    Metasploit::Model::Search::Operator::Attribute.new(
+        # any class that responds to arel_table
+        :klass => Mdm::Host
+    )
+  end
+
+  let(:value) do
+    "value"
+  end
+
+  it 'should visit operation.operator with attribute_visitor' do
+    visitor.attribute_visitor.should_receive(:visit).with(operator).and_call_original
+
+    visit
+  end
+
+  it 'should call eq on Arel::Attributes::Attribute from attribute_visitor' do
+    attribute = double('Visited Operator')
+    visitor.attribute_visitor.stub(:visit).with(operator).and_return(attribute)
+
+    attribute.should_receive(:eq).with(value)
+
+    visit
+  end
+end

data/spec/support/shared/examples/metasploit_data_models/search/visitor/where/visit/with_metasploit_model_search_group_base.rb

@@ -0,0 +1,50 @@
+shared_examples_for 'MetasploitDataModels::Search::Visitor::Where#visit with Metasploit::Model::Search::Group::Base' do |options={}|
+  options.assert_valid_keys(:arel_class)
+
+  arel_class = options.fetch(:arel_class)
+
+  let(:children) do
+    2.times.collect { |n|
+      double("Group Child #{n}")
+    }
+  end
+
+  let(:node) do
+    node_class.new(
+        :children => children
+    )
+  end
+
+  it 'should visit each child' do
+    visitor.should_receive(:visit).with(node).and_call_original
+
+    children.each do |child|
+      visitor.should_receive(:visit).with(child).and_return(Arel::Nodes::Equality.new(1, 1))
+    end
+
+    visit
+  end
+
+  it "should combine children AREL with #{arel_class}" do
+    visitor.stub(:visit).with(node).and_call_original
+    child_visits = []
+
+    children.each_with_index do |child, i|
+      child_visit = Arel::Nodes::Equality.new(i, i)
+      visitor.stub(:visit).with(child).and_return(child_visit)
+      child_visits << child_visit
+    end
+
+    root = visit
+
+    if root.is_a? Arel::Nodes::Grouping
+      arel_pair = root.expr
+    else
+      arel_pair = root
+    end
+
+    arel_pair.should be_a arel_class
+
+    expect([arel_pair.left, arel_pair.right]).to match_array(child_visits)
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: metasploit_data_models
 version: !ruby/object:Gem::Version
-  version: 0.17.1
+  version: 0.17.2.pre.metasploit.pre.data.pre.models.pre.search
 platform: ruby
 authors:
 - Samuel Huckins
@@ -11,7 +11,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-05-09 00:00:00.000000000 Z
+date: 2014-06-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
@@ -89,6 +89,26 @@ dependencies:
     - - ! '>='
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: metasploit-model
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.24.1.pre.semantic.pre.versioning.pre.2.pre.0
+    - - <
+      - !ruby/object:Gem::Version
+        version: '0.25'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.24.1.pre.semantic.pre.versioning.pre.2.pre.0
+    - - <
+      - !ruby/object:Gem::Version
+        version: '0.25'
 - !ruby/object:Gem::Dependency
   name: redcarpet
   requirement: !ruby/object:Gem::Requirement
@@ -186,10 +206,17 @@ files:
 - app/models/mdm/wmap_request.rb
 - app/models/mdm/wmap_target.rb
 - app/models/mdm/workspace.rb
+- app/models/metasploit_data_models/search/visitor/attribute.rb
+- app/models/metasploit_data_models/search/visitor/includes.rb
+- app/models/metasploit_data_models/search/visitor/joins.rb
+- app/models/metasploit_data_models/search/visitor/method.rb
+- app/models/metasploit_data_models/search/visitor/relation.rb
+- app/models/metasploit_data_models/search/visitor/where.rb
 - app/validators/ip_format_validator.rb
 - app/validators/parameters_validator.rb
 - app/validators/password_is_strong_validator.rb
 - bin/mdm_console
+- config/locales/en.yml
 - console_db.yml
 - db/migrate/000_create_tables.rb
 - db/migrate/001_add_wmap_tables.rb
@@ -306,6 +333,8 @@ files:
 - lib/metasploit_data_models/change_required_columns_to_null_false.rb
 - lib/metasploit_data_models/engine.rb
 - lib/metasploit_data_models/models.rb
+- lib/metasploit_data_models/search.rb
+- lib/metasploit_data_models/search/visitor.rb
 - lib/metasploit_data_models/serialized_prefs.rb
 - lib/metasploit_data_models/validators.rb
 - lib/metasploit_data_models/version.rb
@@ -353,6 +382,12 @@ files:
 - spec/app/models/mdm/web_site_spec.rb
 - spec/app/models/mdm/web_vuln_spec.rb
 - spec/app/models/mdm/workspace_spec.rb
+- spec/app/models/metasploit_data_models/search/visitor/attribute_spec.rb
+- spec/app/models/metasploit_data_models/search/visitor/includes_spec.rb
+- spec/app/models/metasploit_data_models/search/visitor/joins_spec.rb
+- spec/app/models/metasploit_data_models/search/visitor/method_spec.rb
+- spec/app/models/metasploit_data_models/search/visitor/relation_spec.rb
+- spec/app/models/metasploit_data_models/search/visitor/where_spec.rb
 - spec/app/validators/parameters_validator_spec.rb
 - spec/app/validators/password_is_strong_validator_spec.rb
 - spec/dummy/Rakefile
@@ -372,6 +407,7 @@ files:
 - spec/dummy/config/environments/development.rb
 - spec/dummy/config/environments/production.rb
 - spec/dummy/config/environments/test.rb
+- spec/dummy/config/initializers/active_record_migrations.rb
 - spec/dummy/config/initializers/backtrace_silencers.rb
 - spec/dummy/config/initializers/inflections.rb
 - spec/dummy/config/initializers/mime_types.rb
@@ -438,6 +474,13 @@ files:
 - spec/spec_helper.rb
 - spec/support/shared/examples/mdm/module/detail/does_not_support_stance_with_mtype.rb
 - spec/support/shared/examples/mdm/module/detail/supports_stance_with_mtype.rb
+- spec/support/shared/examples/metasploit_data_models/search/visitor/includes/visit/with_children.rb
+- spec/support/shared/examples/metasploit_data_models/search/visitor/includes/visit/with_metasploit_model_search_operation_base.rb
+- spec/support/shared/examples/metasploit_data_models/search/visitor/relation/visit/matching_record.rb
+- spec/support/shared/examples/metasploit_data_models/search/visitor/relation/visit/matching_record/with_metasploit_model_search_opeator_deprecated_platform.rb
+- spec/support/shared/examples/metasploit_data_models/search/visitor/relation/visit/matching_record/with_metasploit_model_search_operator_deprecated_authority.rb
+- spec/support/shared/examples/metasploit_data_models/search/visitor/where/visit/with_equality.rb
+- spec/support/shared/examples/metasploit_data_models/search/visitor/where/visit/with_metasploit_model_search_group_base.rb
 homepage: ''
 licenses: []
 metadata: {}
@@ -452,9 +495,9 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
-  - - ! '>='
+  - - ! '>'
     - !ruby/object:Gem::Version
-      version: '0'
+      version: 1.3.1
 requirements: []
 rubyforge_project: 
 rubygems_version: 2.2.2
@@ -503,6 +546,12 @@ test_files:
 - spec/app/models/mdm/web_site_spec.rb
 - spec/app/models/mdm/web_vuln_spec.rb
 - spec/app/models/mdm/workspace_spec.rb
+- spec/app/models/metasploit_data_models/search/visitor/attribute_spec.rb
+- spec/app/models/metasploit_data_models/search/visitor/includes_spec.rb
+- spec/app/models/metasploit_data_models/search/visitor/joins_spec.rb
+- spec/app/models/metasploit_data_models/search/visitor/method_spec.rb
+- spec/app/models/metasploit_data_models/search/visitor/relation_spec.rb
+- spec/app/models/metasploit_data_models/search/visitor/where_spec.rb
 - spec/app/validators/parameters_validator_spec.rb
 - spec/app/validators/password_is_strong_validator_spec.rb
 - spec/dummy/Rakefile
@@ -522,6 +571,7 @@ test_files:
 - spec/dummy/config/environments/development.rb
 - spec/dummy/config/environments/production.rb
 - spec/dummy/config/environments/test.rb
+- spec/dummy/config/initializers/active_record_migrations.rb
 - spec/dummy/config/initializers/backtrace_silencers.rb
 - spec/dummy/config/initializers/inflections.rb
 - spec/dummy/config/initializers/mime_types.rb
@@ -588,4 +638,11 @@ test_files:
 - spec/spec_helper.rb
 - spec/support/shared/examples/mdm/module/detail/does_not_support_stance_with_mtype.rb
 - spec/support/shared/examples/mdm/module/detail/supports_stance_with_mtype.rb
+- spec/support/shared/examples/metasploit_data_models/search/visitor/includes/visit/with_children.rb
+- spec/support/shared/examples/metasploit_data_models/search/visitor/includes/visit/with_metasploit_model_search_operation_base.rb
+- spec/support/shared/examples/metasploit_data_models/search/visitor/relation/visit/matching_record.rb
+- spec/support/shared/examples/metasploit_data_models/search/visitor/relation/visit/matching_record/with_metasploit_model_search_opeator_deprecated_platform.rb
+- spec/support/shared/examples/metasploit_data_models/search/visitor/relation/visit/matching_record/with_metasploit_model_search_operator_deprecated_authority.rb
+- spec/support/shared/examples/metasploit_data_models/search/visitor/where/visit/with_equality.rb
+- spec/support/shared/examples/metasploit_data_models/search/visitor/where/visit/with_metasploit_model_search_group_base.rb
 has_rdoc: